mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-07-08 11:56:59 +00:00
badgerbees
5c16d39e91
* fix(calendar): honor list_events date range aliases * fix(calendar): reject partially resolved loose range queries
107 lines
3.2 KiB
Python
107 lines
3.2 KiB
Python
"""manage_calendar list_events should honor common range aliases.
|
|
|
|
The agent prompt and schema prefer start/end, but model calls can emit
|
|
start_date/end_date or from/to. Those aliases used to be ignored, causing the
|
|
tool to fall back to its default 14-day window.
|
|
"""
|
|
|
|
import json
|
|
import sys
|
|
import tempfile
|
|
import uuid
|
|
|
|
import pytest
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.pool import NullPool
|
|
|
|
from tests.helpers.import_state import clear_fake_database_modules
|
|
|
|
clear_fake_database_modules()
|
|
|
|
import core.database as cdb
|
|
|
|
_TMPDB = tempfile.NamedTemporaryFile(suffix=".db", delete=False)
|
|
_ENGINE = create_engine(
|
|
f"sqlite:///{_TMPDB.name}",
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=NullPool,
|
|
)
|
|
cdb.Base.metadata.create_all(_ENGINE)
|
|
_TS = sessionmaker(bind=_ENGINE, autoflush=False, autocommit=False)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _bind_temp_db(monkeypatch):
|
|
monkeypatch.setitem(sys.modules, "core.database", cdb)
|
|
parent = sys.modules.get("core")
|
|
if parent is not None:
|
|
monkeypatch.setattr(parent, "database", cdb, raising=False)
|
|
monkeypatch.setattr(cdb, "SessionLocal", _TS)
|
|
yield
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("start_key", "end_key"),
|
|
[
|
|
("start_date", "end_date"),
|
|
("from", "to"),
|
|
("range_start", "range_end"),
|
|
],
|
|
)
|
|
async def test_list_events_honors_range_aliases(start_key, end_key):
|
|
from src.tool_implementations import do_manage_calendar
|
|
|
|
owner = "calendar-alias-" + uuid.uuid4().hex[:8]
|
|
|
|
inside = await do_manage_calendar(json.dumps({
|
|
"action": "create_event",
|
|
"summary": "Late June planning",
|
|
"dtstart": "2126-06-25T10:00:00Z",
|
|
}), owner=owner)
|
|
assert inside.get("exit_code", 0) == 0, inside
|
|
|
|
outside = await do_manage_calendar(json.dumps({
|
|
"action": "create_event",
|
|
"summary": "Outside July planning",
|
|
"dtstart": "2126-07-10T10:00:00Z",
|
|
}), owner=owner)
|
|
assert outside.get("exit_code", 0) == 0, outside
|
|
|
|
res = await do_manage_calendar(json.dumps({
|
|
"action": "list_events",
|
|
start_key: "2126-06-01T00:00:00Z",
|
|
end_key: "2126-07-01T00:00:00Z",
|
|
}), owner=owner)
|
|
|
|
assert res.get("exit_code", 0) == 0, res
|
|
summaries = [event["summary"] for event in res["events"]]
|
|
assert summaries == ["Late June planning"]
|
|
assert "between 2126-06-01 and 2126-07-01" in res["response"]
|
|
|
|
async def test_list_events_rejects_partial_loose_range():
|
|
from src.tool_implementations import do_manage_calendar
|
|
|
|
owner = "calendar-partial-" + uuid.uuid4().hex[:8]
|
|
|
|
# Partial: has query and start, but no end
|
|
res = await do_manage_calendar(json.dumps({
|
|
"action": "list_events",
|
|
"query": "July",
|
|
"start_time": "2126-07-01T00:00:00Z",
|
|
}), owner=owner)
|
|
|
|
assert res.get("exit_code", 1) == 1, res
|
|
assert "list_events needs explicit start/end" in res.get("error", "")
|
|
|
|
# Partial: has query and end, but no start
|
|
res2 = await do_manage_calendar(json.dumps({
|
|
"action": "list_events",
|
|
"query": "July",
|
|
"end_time": "2126-07-31T23:59:59Z",
|
|
}), owner=owner)
|
|
|
|
assert res2.get("exit_code", 1) == 1, res2
|
|
assert "list_events needs explicit start/end" in res2.get("error", "")
|
|
|