tests: dash-chat hermetic suite (7/7) — лента БД портала, секрет команд, anchor-дедуп, коллапс повторов
This commit is contained in:
Simbiot Harness
parent
5c02bc7cdd
commit
b9d7fd9d23
1 changed files with 160 additions and 0 deletions
160
tests/test_dash_chat.py
Normal file
160
tests/test_dash_chat.py
Normal file
|
|
@ -0,0 +1,160 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""t-261001100740-99d55e: чат Simbiot в дашборде (hermetic, без сети/портала).
|
||||||
|
|
||||||
|
Запуск: ./venv/bin/python tests/test_dash_chat.py
|
||||||
|
|
||||||
|
Проверяет: build_chat собирает ленту из БД портала (agent_portal_chat) +
|
||||||
|
реплаев стора, команды канала (/set-tg-bot, cmd-settg-*) скрыты (секрет),
|
||||||
|
пара «вопрос + ответ» по anchor-дедупу (реплай стора vs push-строка БД,
|
||||||
|
«reply-<ms>-<id>»), подряд одинаковые agent-пуши схлопываются (repeat ≥ 2),
|
||||||
|
chat_send пишет sender='user' с consumed=0 и валидным msg_id (dash-*),
|
||||||
|
пустой текст отбрасывается.
|
||||||
|
"""
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import sqlite3
|
||||||
|
import sys
|
||||||
|
import tempfile
|
||||||
|
import time
|
||||||
|
import unittest
|
||||||
|
from unittest import mock
|
||||||
|
|
||||||
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||||
|
|
||||||
|
# изоляция: подменяем пути ДО импорта dashboard
|
||||||
|
_TMP = tempfile.mkdtemp(prefix="simbiot-chat-test-")
|
||||||
|
_COOP_DB = os.path.join(_TMP, "coop.db")
|
||||||
|
|
||||||
|
|
||||||
|
def _setup_coop_db():
|
||||||
|
db = sqlite3.connect(_COOP_DB)
|
||||||
|
db.execute("""CREATE TABLE agent_portal_chat (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
agent_id INTEGER NOT NULL,
|
||||||
|
msg_id TEXT UNIQUE,
|
||||||
|
sender TEXT NOT NULL,
|
||||||
|
text TEXT NOT NULL,
|
||||||
|
ts INTEGER NOT NULL,
|
||||||
|
consumed INTEGER DEFAULT 0)""")
|
||||||
|
db.execute("CREATE TABLE agents (id INTEGER PRIMARY KEY, agent_name TEXT, "
|
||||||
|
"harness_type TEXT)")
|
||||||
|
db.execute("INSERT INTO agents (id, agent_name, harness_type) VALUES (80, 'Simbiot', 'simbiot')")
|
||||||
|
db.commit()
|
||||||
|
db.close()
|
||||||
|
|
||||||
|
|
||||||
|
def _row(msg_id, sender, text, ts, consumed=0):
|
||||||
|
return (msg_id, sender, text, ts, consumed)
|
||||||
|
|
||||||
|
|
||||||
|
class DashChatTest(unittest.TestCase):
|
||||||
|
@classmethod
|
||||||
|
def setUpClass(cls):
|
||||||
|
_setup_coop_db()
|
||||||
|
import harness.dashboard as dash
|
||||||
|
dash._COOP_DB = _COOP_DB
|
||||||
|
dash._COOP_AGENT_CACHE.update({"id": 80, "name": "Simbiot", "ts": time.time() + 600})
|
||||||
|
cls.dash = dash
|
||||||
|
|
||||||
|
def _empty_store(self):
|
||||||
|
return {"inbox": {}}
|
||||||
|
|
||||||
|
# ── лента: БД портала ──
|
||||||
|
def test_feed_reads_portal_db(self):
|
||||||
|
d = self._seed([_row("p1", "user", "привет", 1_000),
|
||||||
|
_row("r1", "agent", "ответ", 2_000)])
|
||||||
|
# hermetic: реальный стор мастер-харнеса не участвует
|
||||||
|
with mock.patch.object(self.dash.es, "run_projections",
|
||||||
|
return_value={"inbox": {}}):
|
||||||
|
out = d.build_chat(limit=10)
|
||||||
|
self.assertTrue(out["ok"])
|
||||||
|
self.assertEqual(out["agent"], "Simbiot")
|
||||||
|
self.assertEqual([(m["from"], m["text"]) for m in out["messages"]],
|
||||||
|
[("user", "привет"), ("agent", "ответ")])
|
||||||
|
self.assertFalse(out["messages"][0]["pending"]) # consumed=0, но старше окна
|
||||||
|
|
||||||
|
# ── секрет: команды канала не попадают в ленту ──
|
||||||
|
def test_command_secret_hidden(self):
|
||||||
|
self._seed([_row("cmd-settg-80-1-abc", "user", "/set-tg-bot 7245:SECRET", 3_000, consumed=1),
|
||||||
|
_row("settg-1790932676601", "agent", "✅ Telegram подключён", 4_000),
|
||||||
|
_row("p1", "user", "обычное сообщение", 5_000)])
|
||||||
|
out = self.dash.build_chat(limit=10)
|
||||||
|
texts = " || ".join(m["text"] for m in out["messages"])
|
||||||
|
self.assertNotIn("SECRET", texts)
|
||||||
|
self.assertIn("обычное сообщение", texts)
|
||||||
|
self.assertIn("Telegram подключён", texts) # агентский отчёт виден
|
||||||
|
|
||||||
|
# ── anchor-дедуп: реплай стора не дублирует push-строку БД ──
|
||||||
|
def test_reply_anchor_dedup(self):
|
||||||
|
self._seed([_row("p1", "user", "вопрос к агенту", 6_000, consumed=0),
|
||||||
|
_row("reply-1790937000000-p1", "agent", "ответ агента", 6_500)])
|
||||||
|
store = {"inbox": {"m-p1": {
|
||||||
|
"update_id": "p1", "chat_id": "portal:Simbiot", "text": "вопрос к агенту",
|
||||||
|
"replied": True, "reply_text": "ответ агента", "ts": "2026-10-02T21:50:00+00:00"}}}
|
||||||
|
with mock.patch.object(self.dash.es, "run_projections", return_value=store):
|
||||||
|
out = self.dash.build_chat(limit=10)
|
||||||
|
msgs = out["messages"]
|
||||||
|
pairs = [(m["from"], m["text"]) for m in msgs]
|
||||||
|
# ровно одна пара вопрос+ответ: push-строка БД и реплай стора не задваивают
|
||||||
|
self.assertEqual(pairs.count(("agent", "ответ агента")), 1, "задвоили ответ (push vs стор)")
|
||||||
|
self.assertEqual(pairs.count(("user", "вопрос к агенту")), 1)
|
||||||
|
|
||||||
|
# ── реплай стора, пуш которого не доехал, — пара видна ──
|
||||||
|
def test_unpushed_store_reply_shown(self):
|
||||||
|
self._seed([_row("p2", "user", "вопрос два", 7_000, consumed=0)])
|
||||||
|
store = {"inbox": {"m-p2": {
|
||||||
|
"update_id": "p2", "chat_id": "portal:Simbiot", "text": "вопрос два",
|
||||||
|
"replied": True, "reply_text": "ответ два (не ушёл push)", "ts": "2026-10-02T21:55:00+00:00"}}}
|
||||||
|
with mock.patch.object(self.dash.es, "run_projections", return_value=store):
|
||||||
|
out = self.dash.build_chat(limit=10)
|
||||||
|
got = {(m["from"], m["text"]) for m in out["messages"]}
|
||||||
|
self.assertIn(("agent", "ответ два (не ушёл push)"), got)
|
||||||
|
self.assertIn(("user", "вопрос два"), got)
|
||||||
|
|
||||||
|
# ── коллапс подряд одинаковых agent-пушей ──
|
||||||
|
def test_repeat_collapse(self):
|
||||||
|
self._seed([_row("s1", "agent", "одно и то же", 8_000),
|
||||||
|
_row("s2", "agent", "одно и то же", 9_000),
|
||||||
|
_row("s3", "agent", "одно и то же", 10_000),
|
||||||
|
_row("s4", "agent", "другое", 11_000)])
|
||||||
|
out = self.dash.build_chat(limit=10)
|
||||||
|
by_text = {}
|
||||||
|
for m in out["messages"]:
|
||||||
|
by_text.setdefault(m["text"], []).append(m)
|
||||||
|
self.assertEqual(len(by_text["одно и то же"]), 1)
|
||||||
|
self.assertEqual(by_text["одно и то же"][0]["repeat"], 3)
|
||||||
|
self.assertEqual(len(by_text["другое"]), 1)
|
||||||
|
|
||||||
|
# ── chat_send: user-строка consumed=0 ──
|
||||||
|
def test_chat_send_inserts_user_row(self):
|
||||||
|
out = self.dash.chat_send("тест из теста")
|
||||||
|
self.assertTrue(out["ok"])
|
||||||
|
self.assertTrue(out["msg_id"].startswith("dash-80-"))
|
||||||
|
db = sqlite3.connect(_COOP_DB)
|
||||||
|
db.row_factory = sqlite3.Row
|
||||||
|
row = db.execute("SELECT sender, text, consumed FROM agent_portal_chat "
|
||||||
|
"WHERE msg_id=?", (out["msg_id"],)).fetchone()
|
||||||
|
db.close()
|
||||||
|
self.assertEqual(row["sender"], "user")
|
||||||
|
self.assertEqual(row["consumed"], 0)
|
||||||
|
self.assertEqual(row["text"], "тест из теста")
|
||||||
|
|
||||||
|
def test_chat_send_empty_rejected(self):
|
||||||
|
self.assertFalse(self.dash.chat_send(" ")["ok"])
|
||||||
|
|
||||||
|
# ── helpers ──
|
||||||
|
def _seed(self, rows):
|
||||||
|
d = self.dash
|
||||||
|
db = sqlite3.connect(_COOP_DB)
|
||||||
|
db.execute("DELETE FROM agent_portal_chat")
|
||||||
|
for mid, sender, text, ts, *rest in rows:
|
||||||
|
consumed = rest[0] if rest else 0
|
||||||
|
db.execute("INSERT INTO agent_portal_chat (msg_id, agent_id, sender, text, ts, consumed)"
|
||||||
|
" VALUES (?,?,?,?,?,?)", (mid, 80, sender, text, ts, consumed))
|
||||||
|
db.commit()
|
||||||
|
db.close()
|
||||||
|
return d
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main(verbosity=2)
|
||||||
Loading…
Reference in a new issue