[PBCKP-304] auto tests cleanup added to ptrack_helper.py

Author: Ivan Lazarev

tests/cfs_backup.py

@@ -4,7 +4,7 @@
 import shutil
 
 from .helpers.cfs_helpers import find_by_extensions, find_by_name, find_by_pattern, corrupt_file
-from .helpers.ptrack_helpers import ProbackupTest, ProbackupException, is_test_result_ok
+from .helpers.ptrack_helpers import ProbackupTest, ProbackupException
 
 tblspace_name = 'cfs_tblspace'
 
@@ -13,9 +13,6 @@ class CfsBackupNoEncTest(ProbackupTest, unittest.TestCase):
     # --- Begin --- #
     @unittest.skipUnless(ProbackupTest.enterprise, 'skip')
     def setUp(self):
-        self.module_name = self.id().split('.')[1]
-        self.fname = self.id().split('.')[3]
-
         self.backup_dir = os.path.join(
             self.tmp_path, self.module_name, self.fname, 'backup')
         self.node = self.make_simple_node(
@@ -1160,15 +1157,9 @@ def test_broken_file_pg_compression_into_tablespace_dir(self):
         )
 
 #    # --- End ---#
-    @unittest.skipUnless(ProbackupTest.enterprise, 'skip')
-    def tearDown(self):
-        module_name = self.id().split('.')[1]
-        fname = self.id().split('.')[3]
-        if is_test_result_ok(self):
-            self.del_test_dir(module_name, fname)
 
 
-#class CfsBackupEncTest(CfsBackupNoEncTest):
+# class CfsBackupEncTest(CfsBackupNoEncTest):
 #    # --- Begin --- #
 #    def setUp(self):
 #        os.environ["PG_CIPHER_KEY"] = "super_secret_cipher_key"

tests/cfs_restore.py

@@ -13,17 +13,14 @@
 import shutil
 
 from .helpers.cfs_helpers import find_by_name
-from .helpers.ptrack_helpers import ProbackupTest, ProbackupException, is_test_result_ok
+from .helpers.ptrack_helpers import ProbackupTest, ProbackupException
 
 tblspace_name = 'cfs_tblspace'
 tblspace_name_new = 'cfs_tblspace_new'
 
 
 class CfsRestoreBase(ProbackupTest, unittest.TestCase):
     def setUp(self):
-        self.module_name = self.id().split('.')[1]
-        self.fname = self.id().split('.')[3]
-
         self.backup_dir = os.path.join(self.tmp_path, self.module_name, self.fname, 'backup')
 
         self.node = self.make_simple_node(
@@ -59,13 +56,6 @@ def setUp(self):
     def add_data_in_cluster(self):
         pass
 
-    @unittest.skipUnless(ProbackupTest.enterprise, 'skip')
-    def tearDown(self):
-        module_name = self.id().split('.')[1]
-        fname = self.id().split('.')[3]
-        if is_test_result_ok(self):
-            self.del_test_dir(module_name, fname)
-
 
 class CfsRestoreNoencEmptyTablespaceTest(CfsRestoreBase):
     # @unittest.expectedFailure

tests/helpers/ptrack_helpers.py

@@ -201,9 +201,16 @@ class ProbackupTest(object):
     # Class attributes
     enterprise = is_enterprise()
     enable_nls = is_nls_enabled()
+    module_name = None
+    fname = None
 
     def __init__(self, *args, **kwargs):
         super(ProbackupTest, self).__init__(*args, **kwargs)
+
+        if isinstance(self, unittest.TestCase):
+            self.module_name = self.id().split('.')[1]
+            self.fname = self.id().split('.')[3]
+
         if '-v' in argv or '--verbose' in argv:
             self.verbose = True
         else:
@@ -367,6 +374,13 @@ def __init__(self, *args, **kwargs):
 
         os.environ["PGAPPNAME"] = "pg_probackup"
 
+    def tearDown(self):
+        if isinstance(self, unittest.TestCase):
+            module_name = self.id().split('.')[1]
+            fname = self.id().split('.')[3]
+            if is_test_result_ok(self):
+                self.del_test_dir(module_name, fname)
+
     @property
     def pg_config_version(self):
         return self.version_to_num(