mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-29 18:40:38 +00:00
Jon Saad-FalconandClaude Opus 4.6
c5c420c1c7
Tests were storing digests with local time but get_today() filters by UTC date, causing failures when local date != UTC date. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
"""Tests for DigestStore and DigestArtifact."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
from openjarvis.agents.digest_store import DigestArtifact, DigestStore
|
|
|
|
|
|
def test_store_and_retrieve(tmp_path):
|
|
store = DigestStore(db_path=str(tmp_path / "digest.db"))
|
|
|
|
artifact = DigestArtifact(
|
|
text="Good morning sir.",
|
|
audio_path=Path("/tmp/digest.mp3"),
|
|
sections={"messages": "You have 3 emails.", "calendar": "2 meetings today."},
|
|
sources_used=["gmail", "gcalendar"],
|
|
generated_at=datetime(2026, 4, 1, 6, 0, 0),
|
|
model_used="claude-sonnet-4-6",
|
|
voice_used="jarvis-v1",
|
|
)
|
|
|
|
store.save(artifact)
|
|
retrieved = store.get_latest()
|
|
|
|
assert retrieved is not None
|
|
assert retrieved.text == "Good morning sir."
|
|
assert retrieved.sections["messages"] == "You have 3 emails."
|
|
assert retrieved.sources_used == ["gmail", "gcalendar"]
|
|
assert retrieved.voice_used == "jarvis-v1"
|
|
|
|
store.close()
|
|
|
|
|
|
def test_get_today(tmp_path):
|
|
store = DigestStore(db_path=str(tmp_path / "digest.db"))
|
|
artifact = DigestArtifact(
|
|
text="Today's digest",
|
|
audio_path=Path("/tmp/today.mp3"),
|
|
sections={"messages": "Nothing urgent."},
|
|
sources_used=["gmail"],
|
|
generated_at=datetime.now(tz=__import__("datetime").timezone.utc),
|
|
model_used="test-model",
|
|
voice_used="jarvis",
|
|
)
|
|
store.save(artifact)
|
|
today = store.get_today(timezone_name="UTC")
|
|
assert today is not None
|
|
assert today.text == "Today's digest"
|
|
store.close()
|
|
|
|
|
|
def test_get_today_returns_none_when_empty(tmp_path):
|
|
store = DigestStore(db_path=str(tmp_path / "digest.db"))
|
|
assert store.get_today() is None
|
|
store.close()
|
|
|
|
|
|
def test_history(tmp_path):
|
|
store = DigestStore(db_path=str(tmp_path / "digest.db"))
|
|
for i in range(3):
|
|
store.save(
|
|
DigestArtifact(
|
|
text=f"Digest {i}",
|
|
audio_path=Path(f"/tmp/d{i}.mp3"),
|
|
sections={},
|
|
sources_used=[],
|
|
generated_at=datetime(2026, 4, 1 + i, 6, 0, 0),
|
|
model_used="test",
|
|
voice_used="jarvis",
|
|
)
|
|
)
|
|
history = store.history(limit=2)
|
|
assert len(history) == 2
|
|
assert history[0].text == "Digest 2" # Most recent first
|
|
store.close()
|