testgenie-py 0.3.4__tar.gz → 0.3.6__tar.gz

{testgenie_py-0.3.4 → testgenie_py-0.3.6}/PKG-INFO

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

{testgenie_py-0.3.4 → testgenie_py-0.3.6}/pyproject.toml

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

{testgenie_py-0.3.4 → testgenie_py-0.3.6}/testgen/controller/docker_controller.py

@@ -52,13 +52,15 @@ class DockerController:
             sys.exit(1)
 
         docker_args = [os.path.basename(args.file_path)] + [arg for arg in sys.argv[2:] if arg != "--safe"]
-        docker_args[0] = f"/controller/testgen/{docker_args[0]}"
+        docker_args[0] = f"/controller/{docker_args[0]}"
 
         # Run the container with the same arguments
         try:
             self.debug(f"project_root: {project_root}")
             container = self.run_container(docker_client, image_name, docker_args, project_root)
 
+            self.clean_up(dest_path)
+
             # Stream the logs to the console
             logs_output = self.get_logs(container)
             self.debug(logs_output)
@@ -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."""