testgenie-py 0.2.7__tar.gz → 0.2.9__tar.gz

{testgenie_py-0.2.7 → testgenie_py-0.2.9}/PKG-INFO

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

{testgenie_py-0.2.7 → testgenie_py-0.2.9}/pyproject.toml

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

{testgenie_py-0.2.7 → testgenie_py-0.2.9}/testgen/controller/docker_controller.py

@@ -1,4 +1,3 @@
-import importlib
 import os
 import sys
 from argparse import Namespace
@@ -6,7 +5,10 @@ import docker
 from docker import DockerClient, client
 from docker import errors
 from docker.models.containers import Container
-import importlib.resources
+import importlib.resources as pkg_resources
+import tempfile
+import shutil
+
 from testgen.service.logging_service import get_logger
 from testgen.service.service import Service
 
@@ -134,14 +136,9 @@ class DockerController:
         print(f"Docker args: {docker_args}")
         print(f"Project root: {project_root}")
         
-        # Debug command to find file and then run testgenie with args
-        debug_cmd = f'find /controller -type f -name "*.py" | grep -i boolean || echo "File not found"; ' \
-                    f'ls -la /controller; ' \
-                    f'testgenie /controller/testgen/code_to_test/boolean.py --test-mode=fuzz --test-format=pytest'
-        
         return docker_client.containers.run(
                 image=image_name,
-                command=["sh", "-c", debug_cmd],
+                command=["testgenie"] + docker_args,
                 volumes={
                 os.path.abspath(project_root): {
                 "bind": "/controller",
@@ -196,23 +193,23 @@ class DockerController:
             return False
 
     def get_dockerfile_path(self) -> str:
-        print("Get dockerfile path")
+        # First, try local development path
         local_dockerfile = os.path.join(os.path.dirname(__file__), "docker", "Dockerfile")
-        print(f"Checking local path: {local_dockerfile}")
         if os.path.exists(local_dockerfile):
+            self.debug(f"Found local Dockerfile at: {local_dockerfile}")
             return local_dockerfile
 
         # If not found locally, try inside installed package
         try:
-            print("With importlib.resources")
-            with importlib.resources.path('testgen.docker', 'Dockerfile') as dockerfile_path:
-                print(f"Checking package path: {dockerfile_path}")
-                if dockerfile_path.exists():
-                    return str(dockerfile_path)
+            dockerfile_resource = pkg_resources.files('testgen').joinpath('docker/Dockerfile')
+            if dockerfile_resource.is_file():
+                self.debug(f"Found package-installed Dockerfile at: {dockerfile_resource}")
+                return str(dockerfile_resource)
         except Exception as e:
             print(f"Error locating Dockerfile in package resources: {e}")
 
-        raise FileNotFoundError("Dockerfile not found in local project or package.")
+        print("Dockerfile not found in local project or package.")
+        sys.exit(1)
 
     @staticmethod
     def is_inside_docker() -> bool: