feat: complete client runtime foundation
This commit is contained in:
@@ -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()
|
||||
Reference in New Issue
Block a user