Skip to content

Commit 901f0ec

Browse files
committed
don't remove base_dir if it was provided by user
1 parent c0bcc7b commit 901f0ec

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

Diff for: testgres/testgres.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -417,6 +417,7 @@ def __init__(self,
417417
self.port = port or reserve_port()
418418
self.should_free_port = port is None
419419
self.base_dir = base_dir or tempfile.mkdtemp()
420+
self.should_rm_base_dir = base_dir is None
420421
self.use_logging = use_logging
421422
self.logger = None
422423

@@ -749,8 +750,9 @@ def cleanup(self, max_attempts=3):
749750

750751
attempts += 1
751752

752-
# remove data directory
753-
shutil.rmtree(self.data_dir, ignore_errors=True)
753+
# remove data directory if necessary
754+
if self.should_rm_base_dir:
755+
shutil.rmtree(self.data_dir, ignore_errors=True)
754756

755757
return self
756758

0 commit comments

Comments
 (0)