feat: complete client runtime foundation

This commit is contained in:
2026-07-22 16:25:38 +00:00
parent c11a7b5b5b
commit 1219117403
20 changed files with 1201 additions and 48 deletions
+57
View File
@@ -0,0 +1,57 @@
import os
import tempfile
import unittest
from pathlib import Path
from archive_clients.backup import BackupError, SQLiteBackupManager
from archive_clients.config import BackupConfig
from archive_clients.locking import DatabaseLease, DatabaseLockedError
from archive_clients.state import ClientStore
class BackupTests(unittest.TestCase):
def test_create_verify_detect_corruption_and_restore(self):
with tempfile.TemporaryDirectory() as directory:
root = Path(directory)
database = root / "state" / "client.db"
backups = root / "backups"
store = ClientStore(database)
store.initialize()
manager = SQLiteBackupManager(database, backups, BackupConfig())
record = manager.create("test")
self.assertEqual(manager.verify(record.database).sha256, record.sha256)
database.write_bytes(b"broken")
preserved = manager.restore(record.database)
self.assertIsNotNone(preserved)
self.assertEqual(ClientStore(database).list_active_job_cursors(), [])
with record.database.open("ab") as target:
target.write(b"corruption")
with self.assertRaisesRegex(BackupError, "checksum"):
manager.verify(record.database)
def test_restore_refuses_a_live_database_lease(self):
with tempfile.TemporaryDirectory() as directory:
root = Path(directory)
database = root / "client.db"
ClientStore(database).initialize()
manager = SQLiteBackupManager(database, root / "backups", BackupConfig())
record = manager.create("test")
with DatabaseLease(database):
with self.assertRaises(DatabaseLockedError):
manager.restore(record.database)
def test_backup_database_has_private_permissions(self):
with tempfile.TemporaryDirectory() as directory:
root = Path(directory)
database = root / "client.db"
ClientStore(database).initialize()
record = SQLiteBackupManager(
database, root / "backups", BackupConfig()
).create("test")
self.assertEqual(os.stat(record.database).st_mode & 0o777, 0o600)
if __name__ == "__main__":
unittest.main()