Move deletion to static method

capstone-engine · Sep 9, 2024 · 19cfcea · 19cfcea
1 parent 4ae6215
commit 19cfcea
Showing 1 changed file with 16 additions and 15 deletions.
diff --git a/suite/auto-sync/src/autosync/Tests/test_mcupdater.py b/suite/auto-sync/src/autosync/Tests/test_mcupdater.py
@@ -24,6 +24,16 @@ def setUpClass(cls):
         )
         cls.mutex = Lock()
 
+    @staticmethod
+    def del_path(file: Path):
+        if not file.exists():
+            return
+        logging.debug(f"Delete old file: {file}")
+        if file.is_dir():
+            shutil.rmtree(file)
+        else:
+            os.remove(file)
+
     def test_test_case_gen(self):
         """
         To enforce sequential execution of the tests, we execute them in here.
@@ -48,8 +58,7 @@ def unified_test_cases(self):
         if not out_dir.exists():
             out_dir.mkdir(parents=True)
         for file in out_dir.iterdir():
-            logging.debug(f"Delete old file: {file}")
-            os.remove(file)
+            self.del_path(file)
         test_only_overwrite_path_var(
             "{MCUPDATER_OUT_DIR}",
             out_dir,
@@ -67,8 +76,7 @@ def separated_test_cases(self):
         if not out_dir.exists():
             out_dir.mkdir(parents=True)
         for file in out_dir.iterdir():
-            logging.debug(f"Delete old file: {file}")
-            os.remove(file)
+            self.del_path(file)
         test_only_overwrite_path_var(
             "{MCUPDATER_OUT_DIR}",
             out_dir,
@@ -86,8 +94,7 @@ def multi_mode_unified_test_cases(self):
         if not out_dir.exists():
             out_dir.mkdir(parents=True)
         for file in out_dir.iterdir():
-            logging.debug(f"Delete old file: {file}")
-            os.remove(file)
+            self.del_path(file)
         test_only_overwrite_path_var(
             "{MCUPDATER_OUT_DIR}",
             out_dir,
@@ -112,8 +119,7 @@ def multi_mode_separated_test_cases(self):
         if not out_dir.exists():
             out_dir.mkdir(parents=True)
         for file in out_dir.iterdir():
-            logging.debug(f"Delete old file: {file}")
-            os.remove(file)
+            self.del_path(file)
         test_only_overwrite_path_var(
             "{MCUPDATER_OUT_DIR}",
             out_dir,
@@ -137,8 +143,7 @@ def test_no_symbol_tests(self):
             if not out_dir.exists():
                 out_dir.mkdir(parents=True)
             for file in out_dir.iterdir():
-                logging.debug(f"Delete old file: {file}")
-                os.remove(file)
+                self.del_path(file)
             test_only_overwrite_path_var(
                 "{MCUPDATER_OUT_DIR}",
                 out_dir,
@@ -164,11 +169,7 @@ def test_systemz_mapping(self):
             if not out_dir.exists():
                 out_dir.mkdir(parents=True)
             for file in out_dir.iterdir():
-                logging.debug(f"Delete old file: {file}")
-                if file.is_dir():
-                    shutil.rmtree(file)
-                else:
-                    os.remove(file)
+                self.del_path(file)
             test_only_overwrite_path_var(
                 "{MCUPDATER_OUT_DIR}",
                 out_dir,