From f03d932978b9d341d5b08d54f855afd1ce4c7770 Mon Sep 17 00:00:00 2001
From: mamad <mamad@fedora>
Date: Fri, 28 Aug 2026 22:21:42 +0330
Subject: [PATCH] test: update assertions in
 test_target_context_and_website_custom_needs.py

---
 ...est_target_context_and_website_custom_needs.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/tests/test_target_context_and_website_custom_needs.py b/tests/test_target_context_and_website_custom_needs.py
index 073bdae..b6d4b83 100644
--- a/tests/test_target_context_and_website_custom_needs.py
+++ b/tests/test_target_context_and_website_custom_needs.py
@@ -32,8 +32,9 @@ async def test_target_context_message_count_and_history():
     t1 = datetime(2026, 8, 28, 15, 0, 0, tzinfo=timezone.utc)
     t2 = datetime(2026, 8, 28, 16, 0, 0, tzinfo=timezone.utc)
 
+    src_id = -10099000 - int(time.time() % 100000)
     p0_id = await repo.create_raw_post(
-        source_channel_id=-1001,
+        source_channel_id=src_id,
         source_message_id=501,
         raw_text="پست منتشرشده ۱ در تارگت",
         source_created_at=t0
@@ -41,7 +42,7 @@ async def test_target_context_message_count_and_history():
     await repo.record_post_published_to_target(p0_id, target_id, "Target Context Test")
 
     p1_id = await repo.create_raw_post(
-        source_channel_id=-1001,
+        source_channel_id=src_id,
         source_message_id=502,
         raw_text="پست منتشرشده ۲ در تارگت",
         source_created_at=t1
@@ -49,7 +50,7 @@ async def test_target_context_message_count_and_history():
     await repo.record_post_published_to_target(p1_id, target_id, "Target Context Test")
 
     p2_id = await repo.create_raw_post(
-        source_channel_id=-1001,
+        source_channel_id=src_id,
         source_message_id=503,
         raw_text="پست در حال بازنویسی ۳",
         source_created_at=t2
@@ -112,10 +113,10 @@ async def test_website_custom_needs_and_prompt_injection():
 async def test_menu_layout():
     menu = get_persian_main_menu(is_paused=False)
     assert len(menu) == 5
-    # Row 1 has Home/Start
-    assert any("🏠 منوی اصلی" in btn.text for btn in menu[0])
-    # Row 4 has Unified Add
-    assert any("➕ افزودن مبدا / مقصد" in btn.text for btn in menu[3])
+    first_row_texts = [getattr(getattr(b, "button", None), "text", str(b)) for b in menu[0]]
+    assert any("🏠 منوی اصلی" in t for t in first_row_texts)
+    fourth_row_texts = [getattr(getattr(b, "button", None), "text", str(b)) for b in menu[3]]
+    assert any("➕ افزودن مبدا / مقصد" in t for t in fourth_row_texts)
 
 
 async def main():