codemie-test-harness 0.1.175__py3-none-any.whl → 0.1.177__py3-none-any.whl

codemie_test_harness/tests/assistant/tools/mcp/test_single_assistant_dual_time_plugins.py

@@ -115,10 +115,16 @@ def dual_time_plugins(integration_utils):
         credential_type=CredentialTypes.PLUGIN, credential_values=credential_values
     )
 
-    # Download codemie-tools
-    download_process = subprocess.Popen(["uvx", "codemie-tools"])
+    # Download codemie-plugins and wait for completion
+    download_process = subprocess.Popen(["uvx", "codemie-plugins"])
     processes.append(download_process)
 
+    return_code = download_process.wait()
+    if return_code != 0:
+        raise RuntimeError(
+            f"Failed to download codemie-plugins, exit code: {return_code}"
+        )
+
     # Find paths and configure servers
     servers_json_path, codemie_plugins_path = find_cache_paths()
     configure_mcp_servers(servers_json_path)

codemie_test_harness/tests/conftest.py

@@ -253,7 +253,7 @@ def conversation_utils(client):
     return ConversationUtils(client)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def git_integration(integration_utils):
     integration = integration_utils.create_integration(
         credential_type=CredentialTypes.GIT,
@@ -264,7 +264,7 @@ def git_integration(integration_utils):
         integration_utils.delete_integration(integration)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def github_integration(integration_utils):
     integration = integration_utils.create_integration(
         credential_type=CredentialTypes.GIT,
@@ -275,7 +275,7 @@ def github_integration(integration_utils):
         integration_utils.delete_integration(integration)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def gitlab_integration(integration_utils):
     integration = integration_utils.create_integration(
         credential_type=CredentialTypes.GIT,
@@ -438,7 +438,7 @@ def general_integration(integration_utils):
         pass
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def code_datasource(
     datasource_utils, gitlab_integration, github_integration, default_embedding_llm
 ):
@@ -458,7 +458,7 @@ def code_datasource(
         datasource_utils.delete_datasource(datasource)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def gitlab_datasource(datasource_utils, gitlab_integration, default_embedding_llm):
     datasource = datasource_utils.create_gitlab_datasource(
         setting_id=gitlab_integration.id,
@@ -469,7 +469,7 @@ def gitlab_datasource(datasource_utils, gitlab_integration, default_embedding_ll
         datasource_utils.delete_datasource(datasource)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture(scope="session")
 def github_datasource(datasource_utils, github_integration, default_embedding_llm):
     datasource = datasource_utils.create_github_datasource(
         setting_id=github_integration.id,

codemie_test_harness/tests/llm/assistants/test_lite_llm.py

@@ -13,6 +13,11 @@ from codemie_test_harness.tests.utils.env_resolver import (
 )
 from codemie_test_harness.tests.utils.pytest_utils import check_mark
 
+pytestmark = pytest.mark.skipif(
+    EnvironmentResolver.is_sandbox() or EnvironmentResolver.is_localhost(),
+    reason="Skipping this tests on sandbox environments",
+)
+
 
 @pytest.fixture(scope="module")
 def lite_llm_integration(integration_utils):
@@ -66,10 +71,6 @@ def pytest_generate_tests(metafunc):
 @pytest.mark.assistant
 @pytest.mark.lite_llm
 @pytest.mark.regression
-@pytest.mark.skipif(
-    EnvironmentResolver.is_sandbox(),
-    reason="Skipping this tests on sandbox environments",
-)
 def test_assistant_with_different_models_in_lite_llm(
     llm_utils,
     lite_llm_integration,