Add test for retention count == 0

This commit is contained in:
Yusuke Kadowaki 2022-10-27 01:17:34 +09:00
parent ebf8761cb7
commit abf935b2c9
1 changed files with 7 additions and 2 deletions

View File

@ -277,12 +277,12 @@ class TestNumberedDir:
assert not lock.exists() assert not lock.exists()
def _do_cleanup(self, tmp_path: Path) -> None: def _do_cleanup(self, tmp_path: Path, keep: int = 2) -> None:
self.test_make(tmp_path) self.test_make(tmp_path)
cleanup_numbered_dir( cleanup_numbered_dir(
root=tmp_path, root=tmp_path,
prefix=self.PREFIX, prefix=self.PREFIX,
keep=2, keep=keep,
consider_lock_dead_if_created_before=0, consider_lock_dead_if_created_before=0,
) )
@ -291,6 +291,11 @@ class TestNumberedDir:
a, b = (x for x in tmp_path.iterdir() if not x.is_symlink()) a, b = (x for x in tmp_path.iterdir() if not x.is_symlink())
print(a, b) print(a, b)
def test_cleanup_keep_0(self, tmp_path: Path):
self._do_cleanup(tmp_path, 0)
dir_num = len(list(tmp_path.iterdir()))
assert dir_num == 0
def test_cleanup_locked(self, tmp_path): def test_cleanup_locked(self, tmp_path):
p = make_numbered_dir(root=tmp_path, prefix=self.PREFIX) p = make_numbered_dir(root=tmp_path, prefix=self.PREFIX)