test: improve deterministic unit test coverage
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
"""Tests for Dream session key generation and rotation."""
|
||||
import time
|
||||
from datetime import datetime
|
||||
from datetime import datetime, timedelta
|
||||
from unittest.mock import patch
|
||||
|
||||
from nanobot.agent.memory import MemoryStore
|
||||
|
||||
@@ -13,9 +13,12 @@ class TestDreamSessionKey:
|
||||
datetime.strptime(ts_part, "%Y%m%d-%H%M%S")
|
||||
|
||||
def test_unique_across_calls(self):
|
||||
k1 = MemoryStore.dream_session_key()
|
||||
time.sleep(1.1)
|
||||
k2 = MemoryStore.dream_session_key()
|
||||
now = datetime(2026, 5, 28, 10, 0, 0)
|
||||
with patch("nanobot.agent.memory.datetime") as mock_dt:
|
||||
mock_dt.now.side_effect = [now, now + timedelta(seconds=1)]
|
||||
k1 = MemoryStore.dream_session_key()
|
||||
k2 = MemoryStore.dream_session_key()
|
||||
|
||||
assert k1 != k2
|
||||
|
||||
|
||||
@@ -62,3 +65,4 @@ class TestPruneDreamSessions:
|
||||
sessions_dir = tmp_path / "sessions"
|
||||
sessions_dir.mkdir()
|
||||
MemoryStore.prune_dream_sessions(sessions_dir, keep=10)
|
||||
assert list(sessions_dir.iterdir()) == []
|
||||
|
||||
Reference in New Issue
Block a user