codemie-test-harness 0.1.164__tar.gz → 0.1.166__tar.gz

{codemie_test_harness-0.1.164 → codemie_test_harness-0.1.166}/PKG-INFO

@@ -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 → codemie_test_harness-0.1.166}/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-0.1.164 → codemie_test_harness-0.1.166}/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-0.1.164 → codemie_test_harness-0.1.166}/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 → codemie_test_harness-0.1.166}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-test-harness"
-version = "0.1.164"
+version = "0.1.166"
 description = "Autotest for CodeMie backend and UI"
 authors = [
     "Anton Yeromin <anton_yeromin@epam.com>",
@@ -12,7 +12,7 @@ packages = [
 
 [tool.poetry.dependencies]
 python = "^3.12"
-codemie-sdk-python = "0.1.164"
+codemie-sdk-python = "0.1.166"
 pytest = "^8.4.1"
 python-dotenv = "^1.1.0"
 PyHamcrest = "^2.1.0"