poc2_worked

tests/test_store.py

@@ -83,7 +83,7 @@ class StoreTests(unittest.TestCase):
             row = store.recent(1)[0]
             self.assertEqual(row["hit_count"], 1)
             self.assertEqual(row["first_seen"], "2026-08-13T10:00:00+00:00")
-            self.assertEqual(store.database_info()["schema_version"], 4)
+            self.assertEqual(store.database_info()["schema_version"], 11)
             store.close()
 
     def test_normalizes_timezone_to_utc(self):
@@ -164,6 +164,30 @@ class StoreTests(unittest.TestCase):
             self.assertIsNone(store.find_recent_duplicate(later, 300))
             store.close()
 
+    def test_persists_traffic_snapshots(self):
+        with tempfile.TemporaryDirectory() as td:
+            path = os.path.join(td, "alerts.db")
+            store = AlertStore(path)
+            store.save_traffic_snapshot(900, {"events": 12, "timeline": [{"events": 12}]})
+            row = store.traffic_snapshot(900)
+            self.assertEqual(row["events"], 12)
+            self.assertTrue(row["persisted_snapshot"])
+            self.assertEqual(store.traffic_snapshot_status()["windows"][0]["window_seconds"], 900)
+            store.close()
+
+    def test_persists_and_expires_web_sessions(self):
+        from datetime import datetime, timedelta, timezone
+        with tempfile.TemporaryDirectory() as td:
+            path = os.path.join(td, "alerts.db")
+            store = AlertStore(path)
+            store.create_web_session("hash", "admin", "csrf", datetime.now(timezone.utc) + timedelta(hours=1))
+            self.assertEqual(store.get_web_session("hash")["username"], "admin")
+            store.delete_web_session("hash")
+            self.assertIsNone(store.get_web_session("hash"))
+            store.create_web_session("old", "admin", "csrf", datetime.now(timezone.utc) - timedelta(seconds=1))
+            self.assertIsNone(store.get_web_session("old"))
+            store.close()
+
 
 if __name__ == "__main__":
     unittest.main()