testgenie-py 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

testgen/controller/docker_controller.py

@@ -5,6 +5,9 @@ import docker
 from docker import DockerClient, client
 from docker import errors
 from docker.models.containers import Container
+import importlib.resources as pkg_resources
+import tempfile
+import shutil
 
 from testgen.service.logging_service import get_logger
 from testgen.service.service import Service
@@ -99,8 +102,7 @@ class DockerController:
             print(f"Building {image_name} Docker image...")
 
             # Look for Dockerfile in the project root
-            docker_dir = os.path.join(project_root, "testgen", "docker")
-            dockerfile_path = os.path.join(docker_dir, "Dockerfile")
+            dockerfile_path = self.get_dockerfile_path()
             if not os.path.exists(dockerfile_path):
                 print(f"Dockerfile not found at {dockerfile_path}")
                 sys.exit(1)
@@ -146,13 +148,17 @@ class DockerController:
     def build_docker_image(self, docker_client, image_name, dockerfile_path, project_root):
         try:
             print(f"Starting Docker build for image: {image_name}")
+            dockerfile_rel_path = os.path.relpath(dockerfile_path, project_root)
+            docker_dir = os.path.dirname(dockerfile_path)
+
             dockerfile_rel_path = os.path.relpath(dockerfile_path, project_root)
             self.debug(f"Project root {project_root}")
-            self.debug(f"Docker directory: {os.path.dirname(dockerfile_path)}")
+            self.debug(f"Docker directory: {docker_dir}")
             self.debug(f"Docker rel path: {dockerfile_rel_path}")
+
             build_progress = docker_client.api.build(
-                path=os.path.join(project_root, "testgen", "docker"),
-                dockerfile=os.path.join(project_root, "testgen", "docker", "Dockerfile"),
+                path=docker_dir,          # now it knows what docker_dir is
+                dockerfile=dockerfile_path,
                 tag=image_name,
                 rm=True,
                 decode=True
@@ -178,6 +184,25 @@ class DockerController:
             print(f"Unexpected error during Docker build: {str(e)}")
             return False
 
+    def get_dockerfile_path(self) -> str:
+        # First, try local development path
+        local_dockerfile = os.path.join(os.path.dirname(__file__), "docker", "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:
+            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}")
+
+        print("Dockerfile not found in local project or package.")
+        sys.exit(1)
+
     def debug(self, message: str):
         """Log debug message"""
         if self.debug_mode:

testgen/tests/test_decisions.py

@@ -0,0 +1,177 @@
+import pytest
+
+import testgen.code_to_test.decisions as decisions
+
+def test_add_or_subtract_0():
+   args = (42, 74)
+   expected = 116
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_1():
+   args = (11, 67)
+   expected = 78
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_2():
+   args = (74, 73)
+   expected = 1
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_3():
+   args = (93, 39)
+   expected = 54
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_4():
+   args = (61, 77)
+   expected = 138
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_5():
+   args = (73, 6)
+   expected = 67
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_6():
+   args = (62, 37)
+   expected = 25
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_7():
+   args = (42, 52)
+   expected = 94
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_8():
+   args = (87, 32)
+   expected = 55
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_9():
+   args = (93, 31)
+   expected = 62
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_10():
+   args = (53, 42)
+   expected = 11
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_11():
+   args = (58, 22)
+   expected = 36
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_12():
+   args = (64, 41)
+   expected = 23
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_13():
+   args = (76, 13)
+   expected = 63
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_14():
+   args = (93, 9)
+   expected = 84
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_add_or_subtract_15():
+   args = (65, 82)
+   expected = 147
+   result = decisions.add_or_subtract(*args)
+   assert result == expected
+
+def test_email_type_16():
+   args = 'abc'
+   expected = 'invalid'
+   result = decisions.email_type(args)
+   assert result == expected
+
+def test_http_code_17():
+   args = 66
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_18():
+   args = 58
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_19():
+   args = 26
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_20():
+   args = 44
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_21():
+   args = 41
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_22():
+   args = 13
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_23():
+   args = 32
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_24():
+   args = 34
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_25():
+   args = 12
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_26():
+   args = 91
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_http_code_27():
+   args = 22
+   expected = 'invalid'
+   result = decisions.http_code(args)
+   assert result == expected
+
+def test_password_strength_28():
+   args = 'abc'
+   expected = 'weak'
+   result = decisions.password_strength(args)
+   assert result == expected

{testgenie_py-0.1.7.dist-info → testgenie_py-0.1.9.dist-info}/METADATA

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

{testgenie_py-0.1.7.dist-info → testgenie_py-0.1.9.dist-info}/RECORD

@@ -1,4 +1,4 @@
-testgen/.coverage,sha256=LBC31i6ROQvL24jd3a-mAlGfailO1bmEWlrTCTbxC8s,53248
+testgen/.coverage,sha256=_X1vh4t_SSQH7gwvZk6A8xGjBlVYhPYfhXwe0Z8Iahs,69632
 testgen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/analyzer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/analyzer/ast_analyzer.py,sha256=JmqiosxgZJm2fhNDr5VUBpKKHmf8S2HP9Jm0Xl7NnnI,6783
@@ -22,7 +22,7 @@ testgen/code_to_test/no_types.py,sha256=kD8CGjVAxVeCjQC45abO773sRNIjiZ4jvhwPRV8-
 testgen/code_to_test/sample_code_bin.py,sha256=kMvV_-oFeAkxUmDx41Pw0iTTmMG1pvX31TezXPWMOkE,4225
 testgen/controller/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/controller/cli_controller.py,sha256=jSwT6Cy3JQhxFPRXaODdZ1md5uXO83Q0IrYh86sIdiE,8832
-testgen/controller/docker_controller.py,sha256=Jz1Nc7LW-njvCLDk3svPCVz5UDeQ7gAEFLHXrU1Yrws,7237
+testgen/controller/docker_controller.py,sha256=AF6RFhRMJS9rhE-TGpm8Df48LKCdZFhyDp6jDbukvmk,8186
 testgen/docker/Dockerfile,sha256=OLaWW4i6uUvVQinzYOmazGDAILyM8czcfl-ENkHLKQw,534
 testgen/docker/poetry.lock,sha256=3ZPF2N4KtpMRc9qfamUXumMw7qGmQHaZDaxJ9_bMo7U,51684
 testgen/docker/pyproject.toml,sha256=JWhMupWM7aQFO7-teR9sfFahKB3aM4NPO8on5P3TRxw,563
@@ -59,6 +59,8 @@ testgen/sqlite/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/sqlite/db.py,sha256=RUt88ndT4CQRo9aZzLGXPHRK1iw0LyDEGnoWlOLzKGM,2390
 testgen/sqlite/db_service.py,sha256=i87YkhMP0VNyug5p1tWUvnzHB8isFcSIRe9vJJTmFLA,8954
 testgen/testgen.db,sha256=iyQiDqur-JoS70QD-VYkJFkgjb_YcK2jBAP2aZvvAhY,106496
+testgen/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+testgen/tests/test_decisions.py,sha256=KEbmGbN8IjZsAEv_UOKzdP1a7M7x_U2IHku5XUvpRJE,4039
 testgen/tree/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 testgen/tree/node.py,sha256=ONJtbACShN4yNj1X-UslFRgLyBP6mrbg7qZr3c6dWyA,165
 testgen/tree/tree_utils.py,sha256=gT7jucky6_GWVOlDI6jpv6RMeWCkntGOHIYLvHxD85k,2122
@@ -74,7 +76,7 @@ testgen/util/z3_utils/branch_condition.py,sha256=N9FNR-iJmxIC62NpDQNVZ1OP14rXXqY
 testgen/util/z3_utils/constraint_extractor.py,sha256=RXJLpmk6dAvHZ27839VXKXNtdy9St1F-17-pSEFu4bM,1285
 testgen/util/z3_utils/variable_finder.py,sha256=dUh3F9_L_BDMz1ybiGss09LLcM_egbitgj0FT5Nh9u4,245
 testgen/util/z3_utils/z3_test_case.py,sha256=yF4oJOrXMLzOwDUqXdoeg83MOTl3pvc_lYaZcS01CuQ,4983
-testgenie_py-0.1.7.dist-info/METADATA,sha256=NEqEmswegGYpbG08DuV50BxSw3yvYbXDQJdgujnH5I8,843
-testgenie_py-0.1.7.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-testgenie_py-0.1.7.dist-info/entry_points.txt,sha256=OUN4GqB4zHlHWwWGjwIPbur4E_ZqQgkeeqaCLhzRZgg,47
-testgenie_py-0.1.7.dist-info/RECORD,,
+testgenie_py-0.1.9.dist-info/METADATA,sha256=tr7Wa212lhnjykHl1qKbMXUy14unK_t7HN45KxgOVKg,843
+testgenie_py-0.1.9.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+testgenie_py-0.1.9.dist-info/entry_points.txt,sha256=OUN4GqB4zHlHWwWGjwIPbur4E_ZqQgkeeqaCLhzRZgg,47
+testgenie_py-0.1.9.dist-info/RECORD,,