tests/test_store.py

@@ -0,0 +1,96 @@
+from __future__ import annotations
+
+import json
+
+from gibby.models import AccountRecord, StateFile, UsageSnapshot, UsageWindow
+from gibby.store import JsonStateStore
+
+
+def test_store_writes_canonical_usage_snapshot_shape(tmp_path) -> None:
+    store = JsonStateStore(tmp_path / "accounts.json")
+    snapshot = UsageSnapshot(
+        checked_at=1000,
+        used_percent=75,
+        remaining_percent=25,
+        exhausted=False,
+        primary_window=UsageWindow(75, 18000, 300, 1300),
+        secondary_window=UsageWindow(10, 604800, 3600, 4600),
+        limit_reached=False,
+        allowed=True,
+    )
+    store.save(
+        StateFile(
+            accounts=[
+                AccountRecord(
+                    id="acc@example.com",
+                    email="acc@example.com",
+                    access_token="tok",
+                    refresh_token="ref",
+                    expires_at=2000,
+                    last_known_usage=snapshot,
+                )
+            ]
+        )
+    )
+
+    payload = json.loads((tmp_path / "accounts.json").read_text())
+    saved_snapshot = payload["accounts"][0]["last_known_usage"]
+
+    assert set(saved_snapshot) == {
+        "checked_at",
+        "primary_window",
+        "secondary_window",
+        "limit_reached",
+        "allowed",
+    }
+
+
+def test_store_load_reconstructs_derived_usage_fields(tmp_path) -> None:
+    path = tmp_path / "accounts.json"
+    path.write_text(
+        json.dumps(
+            {
+                "version": 1,
+                "active_account_id": "acc@example.com",
+                "accounts": [
+                    {
+                        "id": "acc@example.com",
+                        "email": "acc@example.com",
+                        "account_id": "acc-1",
+                        "access_token": "tok",
+                        "refresh_token": "ref",
+                        "expires_at": 2000,
+                        "cooldown_until": None,
+                        "last_known_usage": {
+                            "checked_at": 1000,
+                            "primary_window": {
+                                "used_percent": 80,
+                                "limit_window_seconds": 18000,
+                                "reset_after_seconds": 300,
+                                "reset_at": 1300,
+                            },
+                            "secondary_window": {
+                                "used_percent": 100,
+                                "limit_window_seconds": 604800,
+                                "reset_after_seconds": 3600,
+                                "reset_at": 4600,
+                            },
+                            "limit_reached": False,
+                            "allowed": True,
+                        },
+                        "last_error": None,
+                    }
+                ],
+            }
+        )
+    )
+
+    state = JsonStateStore(path).load()
+    snapshot = state.accounts[0].last_known_usage
+
+    assert snapshot is not None
+    assert snapshot.used_percent == 100
+    assert snapshot.remaining_percent == 0
+    assert snapshot.exhausted is True
+    assert snapshot.limit_reached is False
+    assert snapshot.allowed is True