dekun
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from packages.db.backup import (
|
|
backup_db,
|
|
is_valid_backup_name,
|
|
list_backups,
|
|
resolve_backup_file,
|
|
restore_db,
|
|
)
|
|
|
|
|
|
def test_backup_roundtrip(tmp_path: Path):
|
|
db = tmp_path / "live.db"
|
|
backup_dir = tmp_path / "backups"
|
|
conn = __import__("sqlite3").connect(str(db))
|
|
conn.execute("CREATE TABLE t (id INTEGER PRIMARY KEY, v TEXT)")
|
|
conn.execute("INSERT INTO t(v) VALUES ('hello')")
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
dest = backup_db(db, backup_dir, tz="UTC")
|
|
assert dest.is_file()
|
|
assert is_valid_backup_name(dest.name)
|
|
|
|
conn = __import__("sqlite3").connect(str(db))
|
|
conn.execute("DELETE FROM t")
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
restore_db(dest, db)
|
|
conn = __import__("sqlite3").connect(str(db))
|
|
row = conn.execute("SELECT v FROM t").fetchone()
|
|
conn.close()
|
|
assert row[0] == "hello"
|
|
|
|
items = list_backups(backup_dir)
|
|
assert len(items) == 1
|
|
assert items[0]["name"] == dest.name
|
|
assert resolve_backup_file(backup_dir, dest.name) == dest.resolve()
|
|
|
|
|
|
def test_invalid_backup_name(tmp_path: Path):
|
|
with pytest.raises(ValueError):
|
|
resolve_backup_file(tmp_path, "../etc/passwd")
|