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

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

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

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

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

{testgenie_py-0.1.9 → 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=["poetry", "run", "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.1.9 → testgenie_py-0.2.1}/testgen/tests/test_decisions.py

@@ -3,174 +3,192 @@ import pytest
 import testgen.code_to_test.decisions as decisions
 
 def test_add_or_subtract_0():
-   args = (42, 74)
-   expected = 116
+   args = (12, 71)
+   expected = 83
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_1():
-   args = (11, 67)
-   expected = 78
+   args = (7, 81)
+   expected = 88
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_2():
-   args = (74, 73)
-   expected = 1
+   args = (97, 75)
+   expected = 22
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_3():
-   args = (93, 39)
-   expected = 54
+   args = (38, 40)
+   expected = 78
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_4():
-   args = (61, 77)
-   expected = 138
+   args = (66, 3)
+   expected = 63
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_5():
-   args = (73, 6)
-   expected = 67
+   args = (84, 10)
+   expected = 74
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_6():
-   args = (62, 37)
-   expected = 25
+   args = (95, 89)
+   expected = 6
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_7():
-   args = (42, 52)
-   expected = 94
+   args = (57, 12)
+   expected = 45
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_8():
-   args = (87, 32)
-   expected = 55
+   args = (64, 98)
+   expected = 162
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_9():
-   args = (93, 31)
-   expected = 62
+   args = (94, 80)
+   expected = 14
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_10():
-   args = (53, 42)
-   expected = 11
+   args = (26, 3)
+   expected = 23
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_11():
-   args = (58, 22)
-   expected = 36
+   args = (91, 39)
+   expected = 52
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_12():
-   args = (64, 41)
-   expected = 23
+   args = (53, 42)
+   expected = 11
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_13():
-   args = (76, 13)
-   expected = 63
+   args = (15, 42)
+   expected = 57
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_14():
-   args = (93, 9)
-   expected = 84
+   args = (13, 67)
+   expected = 80
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
 def test_add_or_subtract_15():
-   args = (65, 82)
-   expected = 147
+   args = (62, 92)
+   expected = 154
    result = decisions.add_or_subtract(*args)
    assert result == expected
 
-def test_email_type_16():
-   args = 'abc'
-   expected = 'invalid'
-   result = decisions.email_type(args)
+def test_add_or_subtract_16():
+   args = (91, 2)
+   expected = 89
+   result = decisions.add_or_subtract(*args)
    assert result == expected
 
-def test_http_code_17():
-   args = 66
+def test_email_type_17():
+   args = 'abc'
    expected = 'invalid'
-   result = decisions.http_code(args)
+   result = decisions.email_type(args)
    assert result == expected
 
 def test_http_code_18():
-   args = 58
+   args = 81
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_19():
-   args = 26
+   args = 60
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_20():
-   args = 44
+   args = 29
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_21():
-   args = 41
+   args = 11
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_22():
-   args = 13
+   args = 79
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_23():
-   args = 32
+   args = 70
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_24():
-   args = 34
+   args = 52
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_25():
-   args = 12
+   args = 41
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_26():
-   args = 91
+   args = 6
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
 def test_http_code_27():
-   args = 22
+   args = 63
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_28():
+   args = 47
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_29():
+   args = 58
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_30():
+   args = 87
    expected = 'invalid'
    result = decisions.http_code(args)
    assert result == expected
 
-def test_password_strength_28():
+def test_password_strength_31():
    args = 'abc'
    expected = 'weak'
    result = decisions.password_strength(args)

{testgenie_py-0.1.9 → 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.1.9/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"]