Squash Odysseus development history

tests/test_mcp_email_index_search.py

@@ -0,0 +1,99 @@
+import sqlite3
+
+import mcp_servers.email_server as es
+
+
+def test_indexed_search_emails_uses_owner_visible_account_index(monkeypatch, tmp_path):
+    db_path = tmp_path / "scheduled_emails.db"
+    conn = sqlite3.connect(db_path)
+    conn.execute(
+        """
+        CREATE TABLE email_message_index (
+            owner TEXT,
+            account_key TEXT,
+            folder TEXT,
+            uid TEXT,
+            message_id TEXT,
+            subject TEXT,
+            from_name TEXT,
+            from_address TEXT,
+            to_text TEXT,
+            cc_text TEXT,
+            date_iso TEXT,
+            date_display TEXT,
+            date_epoch REAL
+        )
+        """
+    )
+    conn.execute(
+        """
+        INSERT INTO email_message_index VALUES (
+            'alice', 'acct-1', 'INBOX', '42', '<m42>', 'Your Runpod receipt',
+            'Runpod', 'support@runpod.io', 'alice@example.com', '',
+            '2026-08-20T06:28:39+00:00', 'Thu, 20 Aug 2026 06:28:39 +0000', 10
+        )
+        """
+    )
+    conn.execute(
+        """
+        INSERT INTO email_message_index VALUES (
+            'bob', 'acct-1', 'INBOX', '99', '<m99>', 'Your Runpod receipt',
+            'Runpod', 'support@runpod.io', 'bob@example.com', '',
+            '2026-08-20T06:28:39+00:00', 'Thu, 20 Aug 2026 06:28:39 +0000', 11
+        )
+        """
+    )
+    conn.commit()
+    conn.close()
+
+    owner_token = es._CURRENT_OWNER.set("alice")
+    monkeypatch.setattr(es, "SCHEDULED_EMAILS_DB", str(db_path))
+    monkeypatch.setattr(
+        es,
+        "_list_accounts_raw",
+        lambda: [{"id": "acct-1", "name": "Work", "imap_user": "alice@example.com"}],
+    )
+    monkeypatch.setattr(es, "_get_cached_summaries", lambda: {})
+    try:
+        hits = es._indexed_search_emails("Runpod", max_results=10)
+    finally:
+        es._CURRENT_OWNER.reset(owner_token)
+
+    assert hits is not None
+    assert [hit["uid"] for hit in hits] == ["42"]
+    assert hits[0]["_account"] == "Work"
+    assert hits[0]["_source"] == "index"
+
+
+def test_search_emails_falls_back_when_index_misses(monkeypatch):
+    monkeypatch.setattr(es, "_fixture_search_emails", lambda *args, **kwargs: None)
+    monkeypatch.setattr(es, "_indexed_search_emails", lambda *args, **kwargs: [])
+    monkeypatch.setattr(es, "_get_cached_summaries", lambda: {})
+
+    calls = []
+
+    class Conn:
+        def select(self, *_args, **_kwargs):
+            return "OK", []
+
+        def uid(self, *_args, **_kwargs):
+            calls.append(_args)
+            return "OK", [b""]
+
+        def logout(self):
+            pass
+
+    monkeypatch.setattr(es, "_imap_connect", lambda account=None: Conn())
+
+    assert es._search_emails("missing", folders=["INBOX"], max_results=1) == []
+    assert calls
+
+
+def test_fixture_account_selector_accepts_display_label_with_email(monkeypatch):
+    row = {
+        "account": "Research Mail",
+        "account_email": "alex.research@rowan.studio",
+        "account_id": "research-mail",
+    }
+
+    assert es._fixture_row_matches_account(row, "Research Mail (alex.research@rowan.studio)")