Apply retention_count as keep number

This commit is contained in:
Yusuke Kadowaki 2022-10-23 17:29:28 +09:00
parent 7f80cc15b3
commit 6bc424cb05
1 changed files with 5 additions and 5 deletions

View File

@ -31,14 +31,14 @@ class TempPathFactory:
_given_basetemp = attr.ib(type=Optional[Path]) _given_basetemp = attr.ib(type=Optional[Path])
_trace = attr.ib() _trace = attr.ib()
_basetemp = attr.ib(type=Optional[Path]) _basetemp = attr.ib(type=Optional[Path])
_retention_count = attr.ib(type=Optional[int]) _retention_count = attr.ib(type=int)
_retention_policy = attr.ib(type=Optional[str]) _retention_policy = attr.ib(type=str)
def __init__( def __init__(
self, self,
given_basetemp: Optional[Path], given_basetemp: Optional[Path],
retention_count: Optional[int], retention_count: int,
retention_policy: Optional[str], retention_policy: str,
trace, trace,
basetemp: Optional[Path] = None, basetemp: Optional[Path] = None,
*, *,
@ -157,7 +157,7 @@ class TempPathFactory:
basetemp = make_numbered_dir_with_cleanup( basetemp = make_numbered_dir_with_cleanup(
prefix="pytest-", prefix="pytest-",
root=rootdir, root=rootdir,
keep=3, keep=self._retention_count,
lock_timeout=LOCK_TIMEOUT, lock_timeout=LOCK_TIMEOUT,
mode=0o700, mode=0o700,
) )