v0.1.0

tests/test_session_storage_runtime.py

@@ -0,0 +1,61 @@
+from pathlib import Path
+import sqlite3
+
+ROOT = Path(__file__).resolve().parents[1]
+
+
+def test_sqlite_session_migration_executes_and_survives_reopen(tmp_path):
+    db_path = tmp_path / "rustbox.sqlite"
+    migration = (ROOT / "migrations/sqlite/V047__sessions.sql").read_text()
+
+    conn = sqlite3.connect(db_path)
+    conn.executescript(migration)
+    conn.execute(
+        """INSERT INTO sessions (
+            token, account_id, browser_id, email, password_ciphertext, user_hash,
+            imap_host, imap_port, imap_tls, imap_tls_verify,
+            smtp_host, smtp_port, smtp_tls, smtp_tls_verify,
+            timeout_secs, created_at_unix, last_accessed_unix, user_agent
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
+        (
+            "token-1", "account-1", "browser-1", "user@example.com", "ciphertext", "hash-1",
+            "imap.example.com", 993, 1, 1, "smtp.example.com", 587, 1, 1,
+            3600, 1000, 1000, "pytest",
+        ),
+    )
+    conn.commit()
+    conn.close()
+
+    reopened = sqlite3.connect(db_path)
+    row = reopened.execute(
+        "SELECT token, account_id, browser_id, password_ciphertext FROM sessions WHERE token = ?",
+        ("token-1",),
+    ).fetchone()
+    reopened.close()
+    assert row == ("token-1", "account-1", "browser-1", "ciphertext")
+
+
+def test_sqlite_expiry_query_removes_only_expired_sessions(tmp_path):
+    db_path = tmp_path / "rustbox.sqlite"
+    conn = sqlite3.connect(db_path)
+    conn.executescript((ROOT / "migrations/sqlite/V047__sessions.sql").read_text())
+    base = (
+        "account", "browser", "user@example.com", "ciphertext", "hash",
+        "imap", 993, 1, 1, "smtp", 587, 1, 1, 100, 1000, "pytest",
+    )
+    for token, last_accessed in [("expired", 1000), ("live", 1950)]:
+        conn.execute(
+            """INSERT INTO sessions (
+                token, account_id, browser_id, email, password_ciphertext, user_hash,
+                imap_host, imap_port, imap_tls, imap_tls_verify,
+                smtp_host, smtp_port, smtp_tls, smtp_tls_verify,
+                timeout_secs, created_at_unix, last_accessed_unix, user_agent
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
+            (token, f"{base[0]}-{token}", *base[1:15], last_accessed, base[15]),
+        )
+    delete_sql = (ROOT / "src/queries/sqlite/queries.rs").read_text()
+    assert "last_accessed_unix + timeout_secs < ?" in delete_sql
+    conn.execute("DELETE FROM sessions WHERE last_accessed_unix + timeout_secs < ?", (2000,))
+    remaining = [row[0] for row in conn.execute("SELECT token FROM sessions ORDER BY token")]
+    conn.close()
+    assert remaining == ["live"]