testgenie-py 0.3.5__tar.gz → 0.3.7__tar.gz

{testgenie_py-0.3.5 → testgenie_py-0.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: testgenie-py
-Version: 0.3.5
+Version: 0.3.7
 Summary: Automated unit test generation tool for Python.
 Author: cjseitz
 Author-email: charlesjseitz@gmail.com

{testgenie_py-0.3.5 → testgenie_py-0.3.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "testgenie-py"
-version = "0.3.5"
+version = "0.3.7"
 description = "Automated unit test generation tool for Python."
 authors = ["cjseitz <charlesjseitz@gmail.com>"]
 readme = "README.md"

{testgenie_py-0.3.5 → testgenie_py-0.3.7}/testgen/controller/docker_controller.py

@@ -61,6 +61,8 @@ class DockerController:
 
             # Stream the logs to the console
             logs_output = self.get_logs(container)
+
+            self.clean_up(dest_path)
             self.debug(logs_output)
 
         except Exception as e:
@@ -221,6 +223,17 @@ class DockerController:
         print("Dockerfile not found in local project or package.")
         sys.exit(1)
 
+    def clean_up(self, dest_path: str) -> None:
+        # Remove copied file from project root
+        try:
+            if os.path.exists(dest_path):
+                os.remove(dest_path)
+                print(f"Deleted file from {dest_path}")
+                self.logger.debug(f"Deleted file from {dest_path}")
+        except Exception as e:
+            print(f"Failed to delete file: {e}")
+            self.logger.error(f"Failed to delete file: {e}")
+
     @staticmethod
     def is_inside_docker() -> bool:
         """Check if the current process is running inside a Docker container."""