File size: 1,671 Bytes
8ede856 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | import os
import time
from pathlib import Path
from astrbot.core.utils.temp_dir_cleaner import TempDirCleaner, parse_size_to_bytes
def test_parse_size_to_bytes():
assert parse_size_to_bytes("1024") == 1024 * 1024**2
assert parse_size_to_bytes(2048) == 2048 * 1024**2
assert parse_size_to_bytes("0.5") == int(0.5 * 1024**2)
assert parse_size_to_bytes(0) == 0
assert parse_size_to_bytes("invalid") == 0
def _write_file(path: Path, size: int, mtime: float) -> None:
path.write_bytes(b"x" * size)
os.utime(path, (mtime, mtime))
def test_cleanup_once_releases_30_percent_and_prefers_old_files(tmp_path):
temp_dir = tmp_path / "temp"
temp_dir.mkdir(parents=True, exist_ok=True)
base_time = time.time() - 1000
file_old = temp_dir / "old.bin"
file_mid = temp_dir / "mid.bin"
file_new = temp_dir / "new.bin"
_write_file(file_old, 400, base_time)
_write_file(file_mid, 300, base_time + 10)
_write_file(file_new, 300, base_time + 20)
cleaner = TempDirCleaner(max_size_getter=lambda: "0.0008", temp_dir=temp_dir)
cleaner.cleanup_once()
remaining_size = sum(f.stat().st_size for f in temp_dir.rglob("*") if f.is_file())
assert remaining_size <= 600
assert not file_old.exists()
assert file_mid.exists()
assert file_new.exists()
def test_cleanup_once_noop_when_below_limit(tmp_path):
temp_dir = tmp_path / "temp"
temp_dir.mkdir(parents=True, exist_ok=True)
file_path = temp_dir / "a.bin"
_write_file(file_path, 100, time.time())
cleaner = TempDirCleaner(max_size_getter=lambda: "1", temp_dir=temp_dir)
cleaner.cleanup_once()
assert file_path.exists()
|