codemie-test-harness 0.1.164__py3-none-any.whl → 0.1.166__py3-none-any.whl

codemie_test_harness/tests/conftest.py

@@ -60,6 +60,33 @@ from codemie_test_harness.tests.utils.env_resolver import (
 logger = setup_logger(__name__)
 
 
+def cleanup_plugin_process(process):
+    """Helper function to cleanly terminate a plugin process.
+
+    Args:
+        process: The subprocess.Popen process to terminate
+    """
+    if process and process.poll() is None:  # Process is still running
+        logger.info(f"Terminating plugin process with PID: {process.pid}")
+        try:
+            process.terminate()
+            process.wait(timeout=10)  # Wait up to 10 seconds for graceful shutdown
+            logger.info(
+                f"Successfully terminated plugin process with PID: {process.pid}"
+            )
+        except subprocess.TimeoutExpired:
+            logger.warning(
+                f"Plugin process {process.pid} did not terminate gracefully, forcing kill"
+            )
+            process.kill()
+            process.wait()
+            logger.info(f"Successfully killed plugin process with PID: {process.pid}")
+        except Exception as e:
+            logger.error(f"Error during plugin process cleanup: {e}")
+    else:
+        logger.info("Plugin process already terminated or not found")
+
+
 def pytest_configure(config):
     """Pytest hook that runs before test collection starts.
 
@@ -730,9 +757,11 @@ def filesystem_server(integration_utils):
         "-e",
         "filesystem=FILE_PATHS",
     ]
-    subprocess.Popen(command, env=env)
+    process = subprocess.Popen(command, env=env)
+    logger.info(f"Started filesystem server process with PID: {process.pid}")
     sleep(10)
-    return settings
+    yield settings
+    cleanup_plugin_process(process)
 
 
 @pytest.fixture(scope="session")
@@ -764,19 +793,23 @@ def cli_server(integration_utils):
         "-e",
         "cli-mcp-server=ALLOWED_DIR",
     ]
-    subprocess.Popen(command, env=env)
+    process = subprocess.Popen(command, env=env)
+    logger.info(f"Started CLI server process with PID: {process.pid}")
     sleep(10)
-    return settings
+    yield settings
+    cleanup_plugin_process(process)
 
 
 @pytest.fixture(scope="session")
 def development_plugin(integration_utils):
     key = str(uuid.uuid4())
+    logger.info(f"Plugin key: {key}")
 
     credential_values = CredentialsManager.plugin_credentials(key)
     settings = integration_utils.create_integration(
         CredentialTypes.PLUGIN, credential_values
     )
+    logger.info(f"Integration id: {settings.id}")
 
     env = os.environ.copy()
     env["PLUGIN_EXPERIMENTAL_PROTOCOL"] = "true"
@@ -792,9 +825,11 @@ def development_plugin(integration_utils):
         "--repo-path",
         str(TESTS_PATH),
     ]
-    subprocess.Popen(command, env=env)
+    process = subprocess.Popen(command, env=env)
+    logger.info(f"Started development plugin process with PID: {process.pid}")
     sleep(10)
-    return settings
+    yield settings
+    cleanup_plugin_process(process)
 
 
 @pytest.fixture(scope="module")

codemie_test_harness/tests/enums/model_types.py

@@ -24,6 +24,7 @@ class ModelTypes(str, Enum):
     CLAUDE_SONNET_V2_VERTEX = "claude-sonnet-v2-vertex"
     GEMINI_15_PRO = "gemini-1.5-pro"
     GEMINI_20_FLASH = "gemini-2.0-flash"
+    GEMINI_25_FLASH = "gemini-2.5-flash"
     GEMINI_25_PRO = "gemini-2.5-pro"
     CLAUDE_SONNET_37_VERTEX = "claude-sonnet-3-7-vertex"
 

codemie_test_harness/tests/test_data/llm_test_data.py

@@ -41,6 +41,7 @@ MODEL_RESPONSES = [
         GCP_ENVS,
     ),
     LlmResponseData(ModelTypes.GEMINI_20_FLASH, GCP_ENVS),
+    LlmResponseData(ModelTypes.GEMINI_25_FLASH, GCP_ENVS),
     LlmResponseData(ModelTypes.GEMINI_25_PRO, GCP_ENVS),
     LlmResponseData(
         ModelTypes.CLAUDE_SONNET_37_VERTEX,

{codemie_test_harness-0.1.164.dist-info → codemie_test_harness-0.1.166.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: codemie-test-harness
-Version: 0.1.164
+Version: 0.1.166
 Summary: Autotest for CodeMie backend and UI
 Author: Anton Yeromin
 Author-email: anton_yeromin@epam.com
@@ -13,7 +13,7 @@ Requires-Dist: aws-assume-role-lib (>=2.10.0,<3.0.0)
 Requires-Dist: boto3 (>=1.39.8,<2.0.0)
 Requires-Dist: click (>=8.1.7,<9.0.0)
 Requires-Dist: codemie-plugins (>=0.1.123,<0.2.0)
-Requires-Dist: codemie-sdk-python (==0.1.164)
+Requires-Dist: codemie-sdk-python (==0.1.166)
 Requires-Dist: pytest (>=8.4.1,<9.0.0)
 Requires-Dist: pytest-playwright (>=0.7.0,<0.8.0)
 Requires-Dist: pytest-reportportal (>=5.5.2,<6.0.0)

{codemie_test_harness-0.1.164.dist-info → codemie_test_harness-0.1.166.dist-info}/RECORD

@@ -59,7 +59,7 @@ codemie_test_harness/tests/assistant/tools/servicenow/__init__.py,sha256=47DEQpj
 codemie_test_harness/tests/assistant/tools/servicenow/test_servicenow_tools.py,sha256=Io-E5m8f1sA2EP6xQ1l4oJ4-4e9uKPqXHkmxwH0ApFM,691
 codemie_test_harness/tests/assistant/tools/vcs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/assistant/tools/vcs/test_assistant_with_vcs_tools.py,sha256=YjTa1iTU7RpG1y9PpkRkz8RkOqOZXQ8C15X92m2ZRhc,962
-codemie_test_harness/tests/conftest.py,sha256=fvauCdWikYJ392_qPPwNIDyVlKkjxpuvel9V7X3LkmI,28758
+codemie_test_harness/tests/conftest.py,sha256=AnxzI-2MYQg9W8f3Zw6FzzDQ1eZakLjT0s4AXfLBm1o,30281
 codemie_test_harness/tests/conversations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/conversations/test_conversations_endpoints.py,sha256=rLYMWLcOWuWFXSSfIjYSO4rjh_QZPZwQSfOQ8znhyr4,4163
 codemie_test_harness/tests/e2e/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -67,7 +67,7 @@ codemie_test_harness/tests/e2e/test_e2e.py,sha256=DavQToIg5SR38xw3AqhwNGzp5hli0n
 codemie_test_harness/tests/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/enums/environment.py,sha256=R3z7VetKeaJorUyrB5hKmlZVMyjHiAw4R5vREcBfdpU,3051
 codemie_test_harness/tests/enums/integrations.py,sha256=hG_cEKN0N5gtGHyrh3wuCpNAWhPyNOJaKsAw16iNshg,164
-codemie_test_harness/tests/enums/model_types.py,sha256=PgC3Zuu5Ml9C18Uz_nLVFRXd3IFJhNZ3TbfAUaMANoE,1231
+codemie_test_harness/tests/enums/model_types.py,sha256=XtLT0breZ1y6QWjmzc6Qw664ZY0lklGRZ_u1AxyXLN4,1272
 codemie_test_harness/tests/enums/tools.py,sha256=7vZeQFo9FYgNt2q7LVpAvzyKDLbqtxapIeZo-g3cE2s,5772
 codemie_test_harness/tests/integrations/__init__.py,sha256=5vnZbxvYQ1Y91O8DJG3QfBHWcdYsMii59cMBzsvHBCg,237
 codemie_test_harness/tests/integrations/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -144,7 +144,7 @@ codemie_test_harness/tests/test_data/google_datasource_test_data.py,sha256=fhMJV
 codemie_test_harness/tests/test_data/index_test_data.py,sha256=VZJC_Fmko32EHaybUwCy0mWMiwAeUAAmhRa0Xt4oTHQ,1115
 codemie_test_harness/tests/test_data/integrations_test_data.py,sha256=1YecCRnrBq5NiM9HEnyKFZ17nuFEnZUlNzKcwo09SDM,13050
 codemie_test_harness/tests/test_data/keycloak_tool_test_data.py,sha256=uF8YqHGgLpQVxKdpKXLe-7F-ipEGQiHHh28nZvvVGM8,1244
-codemie_test_harness/tests/test_data/llm_test_data.py,sha256=E1H1enkPqJz0tfenEOQC4noNZlm8oCWupPfVnlxkzDM,2415
+codemie_test_harness/tests/test_data/llm_test_data.py,sha256=akJYX36GfR1s5gHKOaWbTGB-4XBjCh1pc00quCjHFoQ,2474
 codemie_test_harness/tests/test_data/mcp_server_test_data.py,sha256=m6PImS_J2gPNY5ijf9MG_eOX_LxJjTZ23AXQDgaK_Oc,7151
 codemie_test_harness/tests/test_data/notification_tools_test_data.py,sha256=D7KLP-cyQljU2Io21CukphnzMadJ7VJ0FhXFdU3GfGk,794
 codemie_test_harness/tests/test_data/open_api_tools_test_data.py,sha256=XhVzelRXgLXHF2iIvhI70SP6qHQjT7QW70S06dF6XJI,2480
@@ -350,7 +350,7 @@ codemie_test_harness/tests/workflow/virtual_assistant_tools/servicenow/__init__.
 codemie_test_harness/tests/workflow/virtual_assistant_tools/servicenow/test_workflow_with_servicenow_tools.py,sha256=vq6tucNBxiNIQSmIj_pYiiPm0lipU9X3kzeCd6xEbRM,966
 codemie_test_harness/tests/workflow/virtual_assistant_tools/vcs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/workflow/virtual_assistant_tools/vcs/test_workflow_with_vcs_tools.py,sha256=uD2qs361j6Egp4UumfoQ4gC24-NioXfiW0IF53N9hVA,1175
-codemie_test_harness-0.1.164.dist-info/METADATA,sha256=_2t47U1CTgfwwUZdBDDuxWDV7fLzSOovs6xlaP9MMQI,8998
-codemie_test_harness-0.1.164.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-codemie_test_harness-0.1.164.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
-codemie_test_harness-0.1.164.dist-info/RECORD,,
+codemie_test_harness-0.1.166.dist-info/METADATA,sha256=PVBlFSNDf9gbwIdlbXPt9qHtl2u6lK749GZJJiJjC5g,8998
+codemie_test_harness-0.1.166.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+codemie_test_harness-0.1.166.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
+codemie_test_harness-0.1.166.dist-info/RECORD,,