Handle missing calendar CLI relation (#1574)

This commit is contained in:
red person
2026-06-03 08:11:17 +03:00
committed by GitHub
parent 1453458519
commit 89b04675e2
2 changed files with 39 additions and 1 deletions

View File

@@ -69,11 +69,17 @@ def _parse_dt(s: str) -> datetime:
return datetime.fromisoformat(s.replace("Z", "+00:00")) return datetime.fromisoformat(s.replace("Z", "+00:00"))
def _calendar_name(ev: "CalendarEvent") -> str:
cal = getattr(ev, "calendar", None)
name = getattr(cal, "name", "") if cal else ""
return name if isinstance(name, str) else ""
def _serialize_event(ev: "CalendarEvent") -> dict: def _serialize_event(ev: "CalendarEvent") -> dict:
return { return {
"uid": ev.uid, "uid": ev.uid,
"calendar_id": ev.calendar_id, "calendar_id": ev.calendar_id,
"calendar_name": ev.calendar.name if ev.calendar else "", "calendar_name": _calendar_name(ev),
"summary": ev.summary, "summary": ev.summary,
"description": ev.description or "", "description": ev.description or "",
"location": ev.location or "", "location": ev.location or "",

View File

@@ -0,0 +1,32 @@
import importlib.machinery
import importlib.util
import sys
import types
from pathlib import Path
from types import SimpleNamespace
from unittest.mock import MagicMock
ROOT = Path(__file__).resolve().parents[1]
def _load_cli(monkeypatch):
db = types.ModuleType("core.database")
db.SessionLocal = MagicMock()
db.CalendarCal = MagicMock()
db.CalendarEvent = MagicMock()
monkeypatch.setitem(sys.modules, "core.database", db)
path = ROOT / "scripts" / "odysseus-calendar"
loader = importlib.machinery.SourceFileLoader("odysseus_calendar_cli", str(path))
spec = importlib.util.spec_from_loader(loader.name, loader)
module = importlib.util.module_from_spec(spec)
loader.exec_module(module)
return module
def test_calendar_name_handles_missing_relation(monkeypatch):
cli = _load_cli(monkeypatch)
assert cli._calendar_name(SimpleNamespace(calendar=None)) == ""
assert cli._calendar_name(SimpleNamespace(calendar=SimpleNamespace(name=123))) == ""
assert cli._calendar_name(SimpleNamespace(calendar=SimpleNamespace(name="Work"))) == "Work"