PBCKP-377 optionaly keep test logs

v.shepard · v.shepard · commit b3777b2c4bed · 2023-05-03T12:00:20.000+02:00
diff --git a/testgres/node.py b/testgres/node.py
@@ -801,13 +801,15 @@ def free_port(self):
             self._should_free_port = False
             release_port(self.port)
 
-    def cleanup(self, max_attempts=3):
+    def cleanup(self, max_attempts=3, full_clean=testgres_config.node_cleanup_full):
         """
         Stop node if needed and remove its data/logs directory.
         NOTE: take a look at TestgresConfig.node_cleanup_full.
 
         Args:
             max_attempts: how many times should we try to stop()?
+            full_clean: True - clean all node data
+                        False - clean only data dir, save logs
 
         Returns:
             This instance of :class:`.PostgresNode`.
@@ -816,7 +818,7 @@ def cleanup(self, max_attempts=3):
         self._try_shutdown(max_attempts)
 
         # choose directory to be removed
-        if testgres_config.node_cleanup_full:
+        if full_clean:
             rm_dir = self.base_dir    # everything
         else:
             rm_dir = self.data_dir    # just data, save logs
