We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c0bcc7b commit 901f0ecCopy full SHA for 901f0ec
testgres/testgres.py
@@ -417,6 +417,7 @@ def __init__(self,
417
self.port = port or reserve_port()
418
self.should_free_port = port is None
419
self.base_dir = base_dir or tempfile.mkdtemp()
420
+ self.should_rm_base_dir = base_dir is None
421
self.use_logging = use_logging
422
self.logger = None
423
@@ -749,8 +750,9 @@ def cleanup(self, max_attempts=3):
749
750
751
attempts += 1
752
- # remove data directory
753
- shutil.rmtree(self.data_dir, ignore_errors=True)
+ # remove data directory if necessary
754
+ if self.should_rm_base_dir:
755
+ shutil.rmtree(self.data_dir, ignore_errors=True)
756
757
return self
758
0 commit comments