90 lines
3.4 KiB
Python
90 lines
3.4 KiB
Python
"""Channel lists render as one button menu, and each button opens that channel's card."""
|
|
import asyncio
|
|
import sys
|
|
|
|
sys.path.insert(0, "/app")
|
|
|
|
from db.database import init_db, close_db_pool
|
|
from db.repository import Repository
|
|
from services.admin_bot import AdminBotService
|
|
|
|
SRC = -1009999000030
|
|
TRG = -1009999000031
|
|
|
|
|
|
async def _cleanup(repo):
|
|
pool = await repo._get_pool()
|
|
async with pool.acquire() as conn:
|
|
await conn.execute("DELETE FROM posts WHERE source_channel_id = $1;", SRC)
|
|
await conn.execute("DELETE FROM sources WHERE channel_id = $1;", SRC)
|
|
await conn.execute("DELETE FROM targets WHERE channel_id = $1;", TRG)
|
|
|
|
|
|
def button_data(rows):
|
|
return [b.data.decode() for row in rows for b in row]
|
|
|
|
|
|
async def run_tests():
|
|
await init_db()
|
|
repo = Repository()
|
|
await _cleanup(repo)
|
|
try:
|
|
await _run_assertions(repo)
|
|
finally:
|
|
await _cleanup(repo)
|
|
await close_db_pool()
|
|
print("All admin menu tests passed successfully!")
|
|
|
|
|
|
async def _run_assertions(repo):
|
|
bot = AdminBotService(repo=repo, review_channel_id=0, admin_user_ids=[1])
|
|
|
|
src_id = await repo.add_source(SRC, "Menu Source", "menusrc")
|
|
target_id = await repo.add_target(TRG, "Menu Target", None)
|
|
await repo.create_raw_post(SRC, 1, "a post", content_hash="menu_1")
|
|
|
|
# Source list: one button per source, pointing at its detail view.
|
|
text, buttons = await bot._render_source_list()
|
|
assert "Menu Source" not in text, "the list message itself should stay a header, not a card dump"
|
|
data = button_data(buttons)
|
|
assert f"src_view:{src_id}" in data, f"source button missing: {data}"
|
|
|
|
# Target list likewise, and it flags auto-routing.
|
|
text, buttons = await bot._render_target_list()
|
|
data = button_data(buttons)
|
|
assert f"trg_view:{target_id}" in data, f"target button missing: {data}"
|
|
def label_for(rows, want):
|
|
for row in rows:
|
|
for b in row:
|
|
if b.data.decode() == want:
|
|
return b.text
|
|
raise AssertionError(f"button {want} not found")
|
|
|
|
assert "🤖" not in label_for(buttons, f"trg_view:{target_id}"), "no auto-route configured yet"
|
|
await repo.set_target_auto_sources(target_id, [SRC])
|
|
_, buttons = await bot._render_target_list()
|
|
assert "🤖" in label_for(buttons, f"trg_view:{target_id}"), "auto-routed target should be flagged"
|
|
|
|
# Source detail card carries fetch buttons and a way back.
|
|
card, buttons = await bot._render_source_config(src_id)
|
|
assert "Menu Source" in card and str(SRC) in card
|
|
assert "<b>1</b>" in card, f"collected count missing from card: {card}"
|
|
assert "Menu Target" in card, "auto-routed target should be listed on the source card"
|
|
data = button_data(buttons)
|
|
assert f"hist:{SRC}:20" in data and f"histn:{SRC}" in data, f"fetch buttons missing: {data}"
|
|
assert "list_src" in data, f"back button missing: {data}"
|
|
|
|
# Target detail card can get back to its list too.
|
|
card, buttons = await bot._render_target_config(target_id)
|
|
data = button_data(buttons)
|
|
assert f"auto_src:{target_id}" in data, f"auto-route button missing: {data}"
|
|
assert "list_trg" in data, f"back button missing: {data}"
|
|
|
|
# Missing ids degrade gracefully instead of raising.
|
|
card, buttons = await bot._render_source_config(999999)
|
|
assert "یافت نشد" in card and buttons == []
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(run_tests())
|