42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import os
|
|
import time
|
|
|
|
|
|
def test_cleanup_partial_backups_removes_only_stale_zero_byte_files(tmp_path, monkeypatch):
|
|
from services import db_backup_service as backup
|
|
|
|
stale_zero = tmp_path / "momo_analytics_20260611_010050.sql.gz"
|
|
fresh_zero = tmp_path / "momo_analytics_20260618_134900.sql.gz"
|
|
normal_backup = tmp_path / "momo_analytics_20260618_020001.sql.gz"
|
|
|
|
stale_zero.write_bytes(b"")
|
|
fresh_zero.write_bytes(b"")
|
|
normal_backup.write_bytes(b"ok")
|
|
|
|
old_ts = time.time() - 7200
|
|
os.utime(stale_zero, (old_ts, old_ts))
|
|
|
|
monkeypatch.setattr(backup, "BACKUP_DIR", str(tmp_path))
|
|
|
|
deleted = backup.cleanup_partial_backups(min_age_minutes=60)
|
|
|
|
assert deleted == 1
|
|
assert not stale_zero.exists()
|
|
assert fresh_zero.exists()
|
|
assert normal_backup.exists()
|
|
|
|
|
|
def test_cleanup_old_backups_counts_partial_cleanup(tmp_path, monkeypatch):
|
|
from services import db_backup_service as backup
|
|
|
|
stale_zero = tmp_path / "momo_analytics_20260611_010050.sql.gz"
|
|
stale_zero.write_bytes(b"")
|
|
old_ts = time.time() - 7200
|
|
os.utime(stale_zero, (old_ts, old_ts))
|
|
|
|
monkeypatch.setattr(backup, "BACKUP_DIR", str(tmp_path))
|
|
monkeypatch.setattr(backup, "RETENTION_DAYS", 7)
|
|
|
|
assert backup.cleanup_old_backups() == 1
|
|
assert not stale_zero.exists()
|