testgenie-py 0.2.0__tar.gz → 0.2.1__tar.gz

{testgenie_py-0.2.0 → testgenie_py-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: testgenie-py
-Version: 0.2.0
+Version: 0.2.1
 Summary: 
 Author: cjseitz
 Author-email: charlesjseitz@gmail.com

{testgenie_py-0.2.0 → testgenie_py-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "testgenie-py"
-version = "0.2.0"
+version = "0.2.1"
 description = ""
 authors = ["cjseitz <charlesjseitz@gmail.com>"]
 readme = "README.md"

{testgenie_py-0.2.0 → testgenie_py-0.2.1}/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:

testgenie_py-0.2.1/testgen/docker/Dockerfile

@@ -0,0 +1,31 @@
+# Use Python 3.10 image
+FROM python:3.10
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    graphviz \
+    libgraphviz-dev \
+    pkg-config \
+    curl
+
+# 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
+
+# Install your package from PyPI
+RUN python3 -m pip install --no-cache-dir testgenie-py
+
+# Set a working directory (where your code will be mounted at runtime)
+WORKDIR /controller
+
+# Set up PYTHONPATH
+ENV PYTHONPATH=/controller:/controller/testgen
+
+# Default command
+CMD ["testgenie"]

testgenie_py-0.2.1/testgen/tests/test_boolean.py

@@ -0,0 +1,69 @@
+import pytest
+
+import testgen.code_to_test.boolean as boolean
+
+def test_bin_and_0():
+   args = (False, False)
+   expected = False
+   result = boolean.bin_and(*args)
+   assert result == expected
+
+def test_bin_and_1():
+   args = (True, True)
+   expected = True
+   result = boolean.bin_and(*args)
+   assert result == expected
+
+def test_bin_and_2():
+   args = (True, False)
+   expected = False
+   result = boolean.bin_and(*args)
+   assert result == expected
+
+def test_bin_xor_3():
+   args = (True, True)
+   expected = False
+   result = boolean.bin_xor(*args)
+   assert result == expected
+
+def test_bin_xor_4():
+   args = (True, False)
+   expected = True
+   result = boolean.bin_xor(*args)
+   assert result == expected
+
+def test_bin_xor_5():
+   args = (False, False)
+   expected = False
+   result = boolean.bin_xor(*args)
+   assert result == expected
+
+def test_status_flags_6():
+   args = (True, False, True)
+   expected = 'admin-unverified'
+   result = boolean.status_flags(*args)
+   assert result == expected
+
+def test_status_flags_7():
+   args = (False, False, False)
+   expected = 'inactive'
+   result = boolean.status_flags(*args)
+   assert result == expected
+
+def test_status_flags_8():
+   args = (True, False, False)
+   expected = 'user-unverified'
+   result = boolean.status_flags(*args)
+   assert result == expected
+
+def test_status_flags_9():
+   args = (False, True, True)
+   expected = 'admin-verified'
+   result = boolean.status_flags(*args)
+   assert result == expected
+
+def test_status_flags_10():
+   args = (True, True, False)
+   expected = 'user-verified'
+   result = boolean.status_flags(*args)
+   assert result == expected

{testgenie_py-0.2.0 → testgenie_py-0.2.1}/testgen/util/file_utils.py

@@ -109,8 +109,23 @@ def load_and_parse_file_for_tree(file) -> Module:
     tree = ast.parse(code)
     return tree
 
-def adjust_file_path_for_docker(file_path) -> str:
-        """Adjust file path for Docker environment to handle subdirectories and relative paths."""
+def adjust_file_path_for_docker(file_path: str) -> str:
+    """Adjust the file path to be valid inside the Docker container."""
+    print(f"Docker - adjusting path: {file_path}")
+
+    # If already absolute to /controller, return as-is
+    if file_path.startswith("/controller/"):
+        print(f"Docker - already adjusted: {file_path}")
+        return file_path
+
+    # If relative, make absolute
+    adjusted_path = f"/controller/{file_path.lstrip('/')}"
+    print(f"Docker - adjusted to: {adjusted_path}")
+    return adjusted_path
+
+
+"""def adjust_file_path_for_docker(file_path) -> str:
+        #Adjust file path for Docker environment to handle subdirectories and relative paths.
         print(f"Docker - adjusting path: {file_path}")
         
         # Try direct path first (maybe it's correct already)
@@ -144,13 +159,28 @@ def adjust_file_path_for_docker(file_path) -> str:
         
         # Return original path if we couldn't find a better match
         print(f"Docker - couldn't find file, returning original: {file_path}")
-        return file_path
+        return file_path"""
+
+
 
 def get_project_root_in_docker(script_path) -> str:
-        script_path = os.path.abspath(sys.argv[0])
-        print(f"Script path: {script_path}")
-        script_dir = os.path.dirname(script_path)
-        print(f"Script directory: {script_dir}")
-        project_root = os.path.dirname(script_dir)
-        print(f"Project root directory: {project_root}")
-        return project_root
+    # Don't use sys.argv[0] as it points to the virtualenv
+    # Instead, find the actual project root
+    if os.path.exists('/mnt/c/Users/cjsei/thesis/dev/testgen'):
+        # Hard-coded path for now
+        project_root = '/mnt/c/Users/cjsei/thesis/dev/testgen'
+    else:
+        # Try to find project root by looking for pyproject.toml or similar
+        current_dir = os.path.dirname(os.path.abspath(script_path))
+        while current_dir != '/':
+            if os.path.exists(os.path.join(current_dir, 'pyproject.toml')) or \
+               os.path.exists(os.path.join(current_dir, 'setup.py')):
+                project_root = current_dir
+                break
+            current_dir = os.path.dirname(current_dir)
+        else:
+            # Fallback - use parent of script dir
+            project_root = os.path.dirname(os.path.dirname(os.path.abspath(script_path)))
+    
+    print(f"Project root directory: {project_root}")
+    return project_root

testgenie_py-0.2.0/testgen/docker/Dockerfile

@@ -1,22 +0,0 @@
-FROM python:3.10
-
-RUN apt-get update && apt-get install -y curl build-essential
-
-RUN curl -sSL https://install.python-poetry.org | python3 - \
-    && ln -s /root/.local/bin/poetry /usr/local/bin/poetry
-
-ENV POETRY_VIRTUALENVS_CREATE=false \
-    PYTHONUNBUFFERED=1 \
-    RUNNING_IN_DOCKER=true
-
-WORKDIR /controller
-
-# Copy poetry files
-COPY . .
-
-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"]