testgenie-py 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

testgen/controller/docker_controller.py

@@ -30,6 +30,7 @@ class DockerController:
     def run_in_docker(self, project_root: str, docker_client: DockerClient, args: Namespace) -> bool:
         self.args = args
         self.debug_mode = True if args.debug else False
+    
         os.environ["RUNNING_IN_DOCKER"] = "1"
 
         # Check if Docker image exists, build it if not
@@ -41,6 +42,7 @@ class DockerController:
             return False
 
         docker_args = [args.file_path] + [arg for arg in sys.argv[2:] if arg != "--safe"]
+        docker_args[0] = f"/controller/testgen/code_to_test/{os.path.basename(docker_args[0])}"
 
         # Run the container with the same arguments
         try:
@@ -51,48 +53,44 @@ class DockerController:
             logs_output = self.get_logs(container)
             self.debug(logs_output)
 
-            try:
-                # Create the target directory if it doesn't exist
-                if args.output is None:
-                    target_path = os.path.join(os.getcwd(), 'tests')
-                else:
-                    target_path = args.output
-                os.makedirs(target_path, exist_ok=True)
+        except Exception as e:
+            print(f"Error running container: {e}")
+            sys.exit(1)
 
-                self.debug(f"SERVICE target path after logs: {target_path}")
+        # Create the target directory if it doesn't exist
+        if args.output is None:
+            target_path = os.path.join(os.getcwd(), 'tests')
+        else:
+            target_path = args.output
+        os.makedirs(target_path, exist_ok=True)
 
-                test_cases = self.service.parse_test_cases_from_logs(logs_output)
+        self.debug(f"SERVICE target path after logs: {target_path}")
 
-                print(f"Extracted {len(test_cases)} test cases from container.")
+        test_cases = self.service.parse_test_cases_from_logs(logs_output)
 
-                file_path = os.path.abspath(args.file_path)
-                self.debug(f"Filepath in CLI CONTROLLER: {file_path}")
-                self.service.set_file_path(file_path)
+        print(f"Extracted {len(test_cases)} test cases from container.")
 
-                if args.test_format == "pytest":
-                    self.service.set_test_generator_format(PYTEST_FORMAT)
-                elif args.test_format == "doctest":
-                    self.service.set_test_generator_format(DOCTEST_FORMAT)
-                else:
-                    self.service.set_test_generator_format(UNITTEST_FORMAT)
+        file_path = os.path.abspath(args.file_path)
+        self.debug(f"Filepath in CLI CONTROLLER: {file_path}")
+        self.service.set_file_path(file_path)
 
-                test_file = self.service.generate_test_file(test_cases, target_path)
-                print(f"Unit tests saved to: {test_file}")
+        if args.test_format == "pytest":
+            self.service.set_test_generator_format(PYTEST_FORMAT)
+        elif args.test_format == "doctest":
+            self.service.set_test_generator_format(DOCTEST_FORMAT)
+        else:
+            self.service.set_test_generator_format(UNITTEST_FORMAT)
 
-                if not args.generate_only:
-                    print("Running coverage...")
-                    self.service.run_coverage(test_file)
-                    
-                # Add explicit return True here
-                return True
+        test_file = self.service.generate_test_file(test_cases, target_path)
+        print(f"Unit tests saved to: {test_file}")
 
-            except Exception as e:
-                print(f"Error running container: {e}")
-                sys.exit(1)
+        if not args.generate_only:
+            print("Running coverage...")
+            self.service.run_coverage(test_file)
+            
+        # Add explicit return True here
+        return True
 
-        except Exception as e:
-            print(f"Error running container: {e}")
-            sys.exit(1)
 
     def get_image(self, docker_client: DockerClient, image_name: str, project_root: str):
         try:
@@ -129,15 +127,30 @@ class DockerController:
         # Create Docker-specific environment variables
         docker_env = {
             "RUNNING_IN_DOCKER": "1",
-            "PYTHONPATH": "/controller",
-            "COVERAGE_FILE": "/tmp/.coverage",  # Move coverage file to /tmp
-            "DB_PATH": "/tmp/testgen.db"        # Move DB to /tmp
+            "PYTHONPATH": "/controller"
         }
         
+        # Join arguments with proper escaping
+        args_str = ' '.join(f'"{arg}"' for arg in docker_args)
+        
+        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=["python", "-m", "testgen.main"] + docker_args,
-                volumes={project_root: {"bind": "/controller", "mode": "rw"}},
+                command=["sh", "-c", debug_cmd],
+                volumes={
+                os.path.abspath(project_root): {
+                "bind": "/controller",
+                "mode": "rw"
+                    }
+                },
+                working_dir="/controller",
                 environment=docker_env,
                 detach=True,
                 remove=True,
@@ -203,6 +216,11 @@ class DockerController:
         print("Dockerfile not found in local project or package.")
         sys.exit(1)
 
+    @staticmethod
+    def is_inside_docker() -> bool:
+        """Check if the current process is running inside a Docker container."""
+        return os.environ.get("RUNNING_IN_DOCKER") in ("1", "true", "True")
+
     def debug(self, message: str):
         """Log debug message"""
         if self.debug_mode:

testgen/docker/Dockerfile

@@ -1,22 +1,31 @@
+# Use Python 3.10 image
 FROM python:3.10
 
-RUN apt-get update && apt-get install -y curl build-essential
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    graphviz \
+    libgraphviz-dev \
+    pkg-config \
+    curl
 
-RUN curl -sSL https://install.python-poetry.org | python3 - \
-    && ln -s /root/.local/bin/poetry /usr/local/bin/poetry
+# Install Poetry (optional, in case you ever need it)
+RUN curl -sSL https://install.python-poetry.org | python3 - && \
+    ln -s /root/.local/bin/poetry /usr/local/bin/poetry
 
+# Set environment variables
 ENV POETRY_VIRTUALENVS_CREATE=false \
     PYTHONUNBUFFERED=1 \
     RUNNING_IN_DOCKER=true
 
-WORKDIR /controller
+# Install your package from PyPI
+RUN python3 -m pip install --no-cache-dir testgenie-py
 
-# Copy poetry files
-COPY . .
+# Set a working directory (where your code will be mounted at runtime)
+WORKDIR /controller
 
+# Set up PYTHONPATH
 ENV PYTHONPATH=/controller:/controller/testgen
 
-RUN poetry install --no-root
-
-# Entrypoint (Can be overridden by Docker SDK in Python)
-CMD ["poetry", "run", "python", "-m", "testgen.main"]
+# Default command
+CMD ["testgenie"]