mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-18 21:02:20 +02:00
Squash Odysseus development history
This commit is contained in:
pewdiepie-archdaemon
@@ -0,0 +1,260 @@
|
||||
"""Regression tests for owner-scoped model resolution in scheduled actions."""
|
||||
|
||||
import sqlite3
|
||||
from datetime import datetime
|
||||
from types import SimpleNamespace
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
class _Column:
|
||||
def __eq__(self, _other):
|
||||
return True
|
||||
|
||||
def __ne__(self, _other):
|
||||
return True
|
||||
|
||||
def __ge__(self, _other):
|
||||
return True
|
||||
|
||||
def __le__(self, _other):
|
||||
return True
|
||||
|
||||
|
||||
class _Query:
|
||||
def __init__(self, rows):
|
||||
self._rows = rows
|
||||
|
||||
def filter(self, *_args, **_kwargs):
|
||||
return self
|
||||
|
||||
def limit(self, _limit):
|
||||
return self
|
||||
|
||||
def all(self):
|
||||
return list(self._rows)
|
||||
|
||||
|
||||
class _Db:
|
||||
def __init__(self, rows_by_model):
|
||||
self._rows_by_model = rows_by_model
|
||||
self.commits = 0
|
||||
self.closed = False
|
||||
|
||||
def query(self, model):
|
||||
return _Query(self._rows_by_model.get(model, []))
|
||||
|
||||
def commit(self):
|
||||
self.commits += 1
|
||||
|
||||
def close(self):
|
||||
self.closed = True
|
||||
|
||||
|
||||
def _resolver_spy(monkeypatch, candidates=None):
|
||||
from src import task_endpoint
|
||||
|
||||
calls = []
|
||||
|
||||
def fake_candidates(*args, **kwargs):
|
||||
calls.append(kwargs.get("owner"))
|
||||
if candidates is None:
|
||||
return [("http://llm", "model", {})]
|
||||
return list(candidates)
|
||||
|
||||
monkeypatch.setattr(task_endpoint, "resolve_task_candidates", fake_candidates)
|
||||
return calls
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_classify_events_resolves_llm_for_task_owner(monkeypatch):
|
||||
from core import database
|
||||
from src.builtin_actions import action_classify_events
|
||||
|
||||
class FakeCalendarEvent:
|
||||
dtstart = _Column()
|
||||
status = _Column()
|
||||
|
||||
event = SimpleNamespace(
|
||||
summary="Demo presentation",
|
||||
event_type="work",
|
||||
importance="high",
|
||||
color=None,
|
||||
dtstart=datetime(2026, 1, 1, 9, 0, 0),
|
||||
location="",
|
||||
)
|
||||
db = _Db({FakeCalendarEvent: [event]})
|
||||
calls = _resolver_spy(monkeypatch)
|
||||
|
||||
monkeypatch.setattr(database, "CalendarEvent", FakeCalendarEvent)
|
||||
monkeypatch.setattr(database, "SessionLocal", lambda: db)
|
||||
|
||||
message, ok = await action_classify_events("alice")
|
||||
|
||||
assert ok is True
|
||||
assert "Scanned 1 upcoming event" in message
|
||||
assert calls == ["alice"]
|
||||
assert db.closed is True
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_learn_sender_signatures_resolves_llm_for_task_owner(monkeypatch):
|
||||
from routes import email_helpers
|
||||
from src.builtin_actions import action_learn_sender_signatures
|
||||
|
||||
class FakeImap:
|
||||
def __init__(self, owner=""):
|
||||
self.owner = owner
|
||||
|
||||
def select(self, *_args, **_kwargs):
|
||||
return "OK", []
|
||||
|
||||
def uid(self, command, *_args):
|
||||
if command == "SEARCH":
|
||||
return "OK", [b"1 2 3"]
|
||||
return "OK", [(None, b"From: Writer <writer@example.com>\r\n\r\n")]
|
||||
|
||||
def logout(self):
|
||||
return None
|
||||
|
||||
calls = _resolver_spy(monkeypatch, candidates=[])
|
||||
imap_owners = []
|
||||
|
||||
def fake_imap_connect(_account_id=None, owner=""):
|
||||
imap_owners.append(owner)
|
||||
return FakeImap(owner)
|
||||
|
||||
monkeypatch.setattr(email_helpers, "_imap_connect", fake_imap_connect)
|
||||
|
||||
message, ok = await action_learn_sender_signatures("alice")
|
||||
|
||||
assert ok is False
|
||||
assert message == "No LLM endpoint available"
|
||||
assert calls == ["alice"]
|
||||
assert imap_owners == ["alice"]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_learn_sender_signatures_writes_owner_scoped_cache(monkeypatch, tmp_path):
|
||||
from routes import email_helpers
|
||||
from src import llm_core, task_endpoint
|
||||
from src.builtin_actions import action_learn_sender_signatures
|
||||
|
||||
db_path = tmp_path / "scheduled_emails.db"
|
||||
monkeypatch.setattr(email_helpers, "SCHEDULED_DB", db_path)
|
||||
email_helpers._init_scheduled_db()
|
||||
|
||||
conn = sqlite3.connect(db_path)
|
||||
try:
|
||||
conn.execute(
|
||||
"""
|
||||
INSERT INTO sender_signatures
|
||||
(from_address, owner, signature_text, sample_count, last_built_at, model_used, source)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?)
|
||||
""",
|
||||
(
|
||||
"writer@example.com",
|
||||
"bob",
|
||||
"bob cached signature",
|
||||
3,
|
||||
"2999-01-01T00:00:00",
|
||||
"old-model",
|
||||
"llm",
|
||||
),
|
||||
)
|
||||
conn.commit()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
class FakeImap:
|
||||
def select(self, *_args, **_kwargs):
|
||||
return "OK", []
|
||||
|
||||
def uid(self, command, uid=None, query=None):
|
||||
if command == "SEARCH":
|
||||
return "OK", [b"1 2 3"]
|
||||
if query and "HEADER.FIELDS" in query:
|
||||
return "OK", [(None, b"From: Writer <writer@example.com>\r\n\r\n")]
|
||||
return "OK", [
|
||||
(
|
||||
None,
|
||||
(
|
||||
b"Thanks for the update.\r\n\r\n"
|
||||
b"Regards,\r\n"
|
||||
b"Writer Example\r\n"
|
||||
b"Example Co.\r\n"
|
||||
+ str(uid).encode()
|
||||
),
|
||||
)
|
||||
]
|
||||
|
||||
def logout(self):
|
||||
return None
|
||||
|
||||
imap_owners = []
|
||||
|
||||
def fake_imap_connect(_account_id=None, owner=""):
|
||||
imap_owners.append(owner)
|
||||
return FakeImap()
|
||||
|
||||
monkeypatch.setattr(email_helpers, "_imap_connect", fake_imap_connect)
|
||||
monkeypatch.setattr(
|
||||
task_endpoint,
|
||||
"resolve_task_candidates",
|
||||
lambda *args, **kwargs: [("http://llm", "alice-model", {})],
|
||||
)
|
||||
|
||||
async def fake_llm_call_async(_candidates, **_kwargs):
|
||||
return "Writer Example\nExample Co.\nwriter@example.com"
|
||||
|
||||
monkeypatch.setattr(llm_core, "llm_call_async_with_fallback", fake_llm_call_async)
|
||||
|
||||
message, ok = await action_learn_sender_signatures("alice")
|
||||
|
||||
assert ok is True
|
||||
assert message.startswith("Learned sigs: 1 found")
|
||||
assert imap_owners == ["alice", "alice"]
|
||||
|
||||
conn = sqlite3.connect(db_path)
|
||||
try:
|
||||
rows = conn.execute(
|
||||
"""
|
||||
SELECT owner, signature_text, model_used
|
||||
FROM sender_signatures
|
||||
WHERE from_address = ?
|
||||
ORDER BY owner
|
||||
""",
|
||||
("writer@example.com",),
|
||||
).fetchall()
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
assert rows == [
|
||||
("alice", "Writer Example\nExample Co.\nwriter@example.com", "alice-model"),
|
||||
("bob", "bob cached signature", "old-model"),
|
||||
]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_check_email_urgency_resolves_llm_candidates_for_task_owner(monkeypatch, tmp_path):
|
||||
from core import database
|
||||
from src.builtin_actions import TaskNoop, action_check_email_urgency
|
||||
|
||||
class FakeEmailAccount:
|
||||
enabled = _Column()
|
||||
owner = _Column()
|
||||
imap_user = _Column()
|
||||
from_address = _Column()
|
||||
|
||||
db = _Db({FakeEmailAccount: []})
|
||||
calls = _resolver_spy(monkeypatch)
|
||||
|
||||
monkeypatch.chdir(tmp_path)
|
||||
monkeypatch.setattr(database, "EmailAccount", FakeEmailAccount)
|
||||
monkeypatch.setattr(database, "SessionLocal", lambda: db)
|
||||
|
||||
with pytest.raises(TaskNoop, match="no email accounts configured"):
|
||||
await action_check_email_urgency("alice")
|
||||
|
||||
assert calls == ["alice"]
|
||||
assert db.closed is True
|
||||
Reference in New Issue
Block a user