codemie-test-harness 0.1.130__py3-none-any.whl → 0.1.131__py3-none-any.whl

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflow_details_page.py

@@ -6,9 +6,11 @@ from playwright.sync_api import expect, Locator
 from reportportal_client import step
 from typing import List, Optional
 
-from tests.ui.pageobject.base_page import BasePage
-from tests.ui.pageobject.components import WorkflowSidebar
-from tests.ui.pageobject.components.execution_history_row import ExecutionHistoryRow
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.components import WorkflowSidebar
+from codemie_test_harness.tests.ui.pageobject.components.execution_history_row import (
+    ExecutionHistoryRow,
+)
 
 
 class WorkflowDetailsPage(BasePage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflow_executions_page.py

@@ -2,11 +2,11 @@ from typing import List, Optional
 from playwright.sync_api import expect, Locator
 from reportportal_client import step
 
-from tests.ui.pageobject.base_page import BasePage
-from tests.ui.pageobject.components.workflow_execution_history_item import (
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.components.workflow_execution_history_item import (
     WorkflowExecutionHistoryItem,
 )
-from tests.ui.pageobject.components.workflow_execution_state import (
+from codemie_test_harness.tests.ui.pageobject.components.workflow_execution_state import (
     WorkflowExecutionState,
 )
 

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflow_template_details.py

@@ -1,7 +1,7 @@
 from playwright.sync_api import expect
 from reportportal_client import step
 
-from tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
 
 
 class WorkflowTemplateDetailsPage(BasePage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflow_templates_page.py

@@ -1,9 +1,11 @@
 from reportportal_client import step
 
-from tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
 from playwright.sync_api import Page, expect
 
-from tests.ui.pageobject.workflows.workflows_page import WorkflowsPage
+from codemie_test_harness.tests.ui.pageobject.workflows.workflows_page import (
+    WorkflowsPage,
+)
 
 
 class WorkflowTemplatesPage(BasePage):

{tests → codemie_test_harness/tests}/ui/pageobject/workflows/workflows_page.py

@@ -6,9 +6,14 @@ from hamcrest import assert_that, has_length, greater_than
 from playwright.sync_api import expect
 from reportportal_client import step
 
-from tests.ui.pageobject.components import WorkflowCard, WorkflowSidebar
-from tests.ui.pageobject.base_page import BasePage
-from tests.ui.pageobject.workflows.create_workflow_page import CreateWorkflowPage
+from codemie_test_harness.tests.ui.pageobject.components import (
+    WorkflowCard,
+    WorkflowSidebar,
+)
+from codemie_test_harness.tests.ui.pageobject.base_page import BasePage
+from codemie_test_harness.tests.ui.pageobject.workflows.create_workflow_page import (
+    CreateWorkflowPage,
+)
 
 
 class WorkflowsPage(BasePage):

{tests → codemie_test_harness/tests}/ui/test_create_workflow.py

@@ -2,9 +2,13 @@ import os
 
 import pytest
 
-from tests.ui.pageobject.workflows.workflows_page import WorkflowsPage
-from tests.ui.pageobject.workflows.create_workflow_page import CreateWorkflowPage
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.ui.pageobject.workflows.workflows_page import (
+    WorkflowsPage,
+)
+from codemie_test_harness.tests.ui.pageobject.workflows.create_workflow_page import (
+    CreateWorkflowPage,
+)
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.ui

{tests → codemie_test_harness/tests}/ui/test_edit_workflow.py

@@ -3,10 +3,14 @@ import os
 import pytest
 
 from codemie_sdk.models.workflow import WorkflowMode
-from tests.ui.pageobject.workflows.workflows_page import WorkflowsPage
-from tests.ui.pageobject.workflows.edit_workflow_page import EditWorkflowPage
-from tests.utils.base_utils import get_random_name
-from tests.utils.yaml_utils import (
+from codemie_test_harness.tests.ui.pageobject.workflows.workflows_page import (
+    WorkflowsPage,
+)
+from codemie_test_harness.tests.ui.pageobject.workflows.edit_workflow_page import (
+    EditWorkflowPage,
+)
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.yaml_utils import (
     AssistantModel,
     StateModel,
     WorkflowYamlModel,

{tests → codemie_test_harness/tests}/ui/test_workflow_details.py

@@ -10,10 +10,10 @@ import os
 import pytest
 
 from codemie_sdk.models.workflow import WorkflowMode
-from tests import TEST_USER, PROJECT
-from tests.ui.pageobject.workflows import WorkflowDetailsPage
-from tests.utils.base_utils import get_random_name
-from tests.utils.yaml_utils import (
+from codemie_test_harness.tests import TEST_USER, PROJECT
+from codemie_test_harness.tests.ui.pageobject.workflows import WorkflowDetailsPage
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.yaml_utils import (
     AssistantModel,
     StateModel,
     WorkflowYamlModel,

{tests → codemie_test_harness/tests}/ui/test_workflow_executions_page.py

@@ -17,9 +17,9 @@ import pytz
 from tzlocal import get_localzone
 
 from codemie_sdk.models.workflow import WorkflowMode
-from tests.ui.pageobject.workflows import WorkflowExecutionsPage
-from tests.utils.base_utils import get_random_name
-from tests.utils.yaml_utils import (
+from codemie_test_harness.tests.ui.pageobject.workflows import WorkflowExecutionsPage
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.yaml_utils import (
     AssistantModel,
     StateModel,
     WorkflowYamlModel,

{tests → codemie_test_harness/tests}/ui/test_workflow_templates.py

@@ -1,13 +1,19 @@
 import pytest
 
-from tests import TEST_USER
-from tests.ui.pageobject.workflows.create_workflow_page import CreateWorkflowPage
-from tests.ui.pageobject.workflows.workflow_template_details import (
+from codemie_test_harness.tests import TEST_USER
+from codemie_test_harness.tests.ui.pageobject.workflows.create_workflow_page import (
+    CreateWorkflowPage,
+)
+from codemie_test_harness.tests.ui.pageobject.workflows.workflow_template_details import (
     WorkflowTemplateDetailsPage,
 )
-from tests.ui.pageobject.workflows.workflow_templates_page import WorkflowTemplatesPage
-from tests.ui.pageobject.workflows.workflows_page import WorkflowsPage
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.ui.pageobject.workflows.workflow_templates_page import (
+    WorkflowTemplatesPage,
+)
+from codemie_test_harness.tests.ui.pageobject.workflows.workflows_page import (
+    WorkflowsPage,
+)
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.fixture(scope="function")

{tests → codemie_test_harness/tests}/ui/test_workflows.py

@@ -1,5 +1,7 @@
 import pytest
-from tests.ui.pageobject.workflows.workflows_page import WorkflowsPage
+from codemie_test_harness.tests.ui.pageobject.workflows.workflows_page import (
+    WorkflowsPage,
+)
 
 
 @pytest.mark.ui

{tests → codemie_test_harness/tests}/utils/assistant_utils.py

@@ -7,8 +7,12 @@ from codemie_sdk.models.assistant import (
     EnvVars,
     ExportAssistantPayload,
 )
-from tests import PROJECT, LANGFUSE_TRACES_ENABLED
-from tests.utils.base_utils import BaseUtils, get_random_name, wait_for_entity
+from codemie_test_harness.tests import PROJECT, LANGFUSE_TRACES_ENABLED
+from codemie_test_harness.tests.utils.base_utils import (
+    BaseUtils,
+    get_random_name,
+    wait_for_entity,
+)
 
 
 class AssistantUtils(BaseUtils):

{tests → codemie_test_harness/tests}/utils/aws_parameters_store.py

@@ -10,7 +10,7 @@ import boto3
 from botocore.exceptions import ClientError
 
 from codemie_sdk.models.integration import CredentialValues
-from tests.enums.integrations import DataBaseDialect
+from codemie_test_harness.tests.enums.integrations import DataBaseDialect
 
 
 class AwsParameterStore:
@@ -543,9 +543,11 @@ class CredentialsUtil:
 
     @staticmethod
     def kubernetes_credentials() -> List[CredentialValues]:
-        kubernetes_creds = AwsParameterStore.get_cloud_provider_credentials(
-            "kubernetes"
-        )["sandbox"]["kubernetes"]
+        kubernetes_creds = (
+            AwsParameterStore.get_cloud_provider_credentials("kubernetes")
+            .get("sandbox", {})
+            .get("kubernetes", {})
+        )
         return [
             CredentialValues(key="url", value=CredentialsUtil.AUTO_GENERATED),
             CredentialValues(key="kubernetes_url", value=kubernetes_creds.get("url")),

{tests → codemie_test_harness/tests}/utils/base_utils.py

@@ -18,7 +18,7 @@ from codemie_sdk.models.workflow_state import (
 from codemie_sdk.services.workflow_execution_state import (
     WorkflowExecutionStateService,
 )
-from tests import autotest_entity_prefix
+from codemie_test_harness.tests import autotest_entity_prefix
 
 
 class BaseUtils:
@@ -50,7 +50,7 @@ def to_camel_case(input_string):
 def wait_for_completion(
     execution_state_service: WorkflowExecutionStateService,
     state_id: str,
-    timeout: int = int(os.getenv("DEFAULT_TIMEOUT")),
+    timeout: int = int(os.getenv("DEFAULT_TIMEOUT", "120")),
     pool_interval: int = 3,
 ) -> WorkflowExecutionState:
     start_time = time.time()

{tests → codemie_test_harness/tests}/utils/constants.py

@@ -1,7 +1,7 @@
 from enum import Enum
 from pathlib import Path
 
-from tests.enums.tools import VcsTool, NotificationTool
+from codemie_test_harness.tests.enums.tools import VcsTool, NotificationTool
 
 
 class ProjectManagementIntegrationType(Enum):

{tests → codemie_test_harness/tests}/utils/datasource_utils.py

@@ -19,8 +19,12 @@ from codemie_sdk.models.datasource import (
     UpdateJiraDataSourceRequest,
 )
 from codemie_sdk.models.datasource import DataSourceStatus
-from tests import PROJECT
-from tests.utils.base_utils import BaseUtils, get_random_name, wait_for_entity
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.utils.base_utils import (
+    BaseUtils,
+    get_random_name,
+    wait_for_entity,
+)
 
 
 class DataSourceUtils(BaseUtils):

{tests → codemie_test_harness/tests}/utils/integration_utils.py

@@ -7,8 +7,12 @@ from codemie_sdk.models.integration import (
     IntegrationType,
     IntegrationTestRequest,
 )
-from tests import PROJECT
-from tests.utils.base_utils import BaseUtils, wait_for_entity, get_random_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.utils.base_utils import (
+    BaseUtils,
+    wait_for_entity,
+    get_random_name,
+)
 
 
 class IntegrationUtils(BaseUtils):

{tests → codemie_test_harness/tests}/utils/notification_utils.py

@@ -3,7 +3,7 @@ from typing import Optional
 
 import requests
 
-from tests import AwsParameterStore
+from codemie_test_harness.tests import AwsParameterStore
 
 logger = logging.getLogger(__name__)
 

{tests → codemie_test_harness/tests}/utils/search_utils.py

@@ -1,5 +1,5 @@
 from codemie_sdk.models.integration import IntegrationType
-from tests.utils.base_utils import BaseUtils
+from codemie_test_harness.tests.utils.base_utils import BaseUtils
 
 
 class SearchUtils(BaseUtils):

{tests → codemie_test_harness/tests}/utils/similarity_check.py

@@ -1,7 +1,11 @@
 from codemie_sdk import CodeMieClient
 from hamcrest import assert_that, greater_than_or_equal_to
 
-from tests.utils.base_utils import wait_for_entity, wait_for_completion, clean_json
+from codemie_test_harness.tests.utils.base_utils import (
+    wait_for_entity,
+    wait_for_completion,
+    clean_json,
+)
 
 
 class SimilarityCheck:

{tests → codemie_test_harness/tests}/utils/workflow_utils.py

@@ -5,15 +5,15 @@ from codemie_sdk.models.workflow import (
     WorkflowUpdateRequest,
     WorkflowMode,
 )
-from tests import PROJECT
-from tests.utils import api_domain, verify_ssl
-from tests.utils.base_utils import (
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.utils import api_domain, verify_ssl
+from codemie_test_harness.tests.utils.base_utils import (
     BaseUtils,
     get_random_name,
     wait_for_entity,
     wait_for_completion,
 )
-from tests.utils.http_utils import RequestHandler
+from codemie_test_harness.tests.utils.http_utils import RequestHandler
 
 workflow_endpoint = "/v1/workflows"
 

{tests → codemie_test_harness/tests}/workflow/assistant_tools/ado/test_workflow_with_assistant_with_ado_test_plan_tools.py

@@ -1,7 +1,7 @@
 import pytest
 
-from tests.enums.tools import AzureDevOpsTestPlanTool, Toolkit
-from tests.test_data.ado_test_plan_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import AzureDevOpsTestPlanTool, Toolkit
+from codemie_test_harness.tests.test_data.ado_test_plan_tools_test_data import (
     ado_test_plan_get_test_data,
     ADO_TEST_PLAN_CREATE_TEST_PLAN,
     ADO_TEST_PLAN_CREATE_SUITE,
@@ -9,9 +9,9 @@ from tests.test_data.ado_test_plan_tools_test_data import (
     ADO_TEST_PLAN_DELETE_SUITE,
     ADO_TEST_PLAN_DELETE_PLAN,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import ID_PATTERN
-from tests.utils.json_utils import extract_id_from_ado_response
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import ID_PATTERN
+from codemie_test_harness.tests.utils.json_utils import extract_id_from_ado_response
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/ado/test_workflow_with_assistant_with_ado_wiki_tools.py

@@ -1,14 +1,14 @@
 import pytest
 
-from tests.enums.tools import AzureDevOpsWikiTool, Toolkit
-from tests.test_data.ado_wiki_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import AzureDevOpsWikiTool, Toolkit
+from codemie_test_harness.tests.test_data.ado_wiki_tools_test_data import (
     ado_wiki_get_test_data,
     ADO_WIKI_CREATE_PAGE,
     ADO_WIKI_RENAME_PAGE,
     ADO_WIKI_MODIFY_PAGE,
     ADO_WIKI_DELETE_PAGE,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/ado/test_workflow_with_assistant_with_ado_work_item_tools.py

@@ -1,15 +1,15 @@
 import pytest
 
-from tests.enums.tools import AzureDevOpsWorkItemTool, Toolkit
-from tests.test_data.ado_work_item_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import AzureDevOpsWorkItemTool, Toolkit
+from codemie_test_harness.tests.test_data.ado_work_item_tools_test_data import (
     ado_work_item_get_test_data,
     ADO_WORK_ITEM_CREATE,
     ADO_WORK_ITEM_UPDATE,
     ADO_WORK_ITEM_LINK,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import WORK_ITEM_ID_PATTERN
-from tests.utils.json_utils import extract_id_from_ado_response
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import WORK_ITEM_ID_PATTERN
+from codemie_test_harness.tests.utils.json_utils import extract_id_from_ado_response
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/cloud/test_workflow_with_assistant_cloud_tools.py

@@ -1,7 +1,7 @@
 import pytest
 
-from tests.enums.tools import Toolkit, CloudTool
-from tests.test_data.cloud_tools_test_data import cloud_test_data
+from codemie_test_harness.tests.enums.tools import Toolkit, CloudTool
+from codemie_test_harness.tests.test_data.cloud_tools_test_data import cloud_test_data
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/codebase/test_worfklow_with_assistant_codebase_tools.py

@@ -1,8 +1,8 @@
 import pytest
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.tools import CodeBaseTool
-from tests.test_data.codebase_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import CodeBaseTool
+from codemie_test_harness.tests.test_data.codebase_tools_test_data import (
     code_tools_test_data,
     sonar_tools_test_data,
 )

{tests → codemie_test_harness/tests}/workflow/assistant_tools/data_management/test_workflow_with_assistant_with_data_management_tools.py

@@ -3,8 +3,8 @@ import os
 import pytest
 
 from codemie_sdk.models.integration import CredentialTypes
-from tests.enums.tools import DataManagementTool, Toolkit
-from tests.test_data.data_management_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import DataManagementTool, Toolkit
+from codemie_test_harness.tests.test_data.data_management_tools_test_data import (
     ELASTIC_TOOL_TASK,
     RESPONSE_FOR_ELASTIC,
     sql_tools_test_data,
@@ -14,7 +14,7 @@ from tests.test_data.data_management_tools_test_data import (
     SQL_TOOL_QUERY_TABLE_TASK,
     RESPONSE_FOR_SQL,
 )
-from tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/default_integrations/test_default_integrations_for_tool.py

@@ -4,12 +4,15 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import Toolkit, ProjectManagementTool
-from tests.test_data.pm_tools_test_data import JIRA_TOOL_PROMPT, RESPONSE_FOR_JIRA_TOOL
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.integrations import integrations
+from codemie_test_harness.tests.enums.tools import Toolkit, ProjectManagementTool
+from codemie_test_harness.tests.test_data.pm_tools_test_data import (
+    JIRA_TOOL_PROMPT,
+    RESPONSE_FOR_JIRA_TOOL,
+)
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 
 @pytest.mark.not_for_parallel_run

{tests → codemie_test_harness/tests}/workflow/assistant_tools/default_integrations/test_default_integrations_for_tool_kit.py

@@ -4,12 +4,14 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import Toolkit, AzureDevOpsWikiTool
-from tests.test_data.ado_wiki_tools_test_data import ado_wiki_get_test_data
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.integrations import integrations
+from codemie_test_harness.tests.enums.tools import Toolkit, AzureDevOpsWikiTool
+from codemie_test_harness.tests.test_data.ado_wiki_tools_test_data import (
+    ado_wiki_get_test_data,
+)
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 ado_wiki_prompt = ado_wiki_get_test_data[0][2]
 ado_wiki_answer = ado_wiki_get_test_data[0][3]

{tests → codemie_test_harness/tests}/workflow/assistant_tools/default_integrations/test_default_integrations_for_tool_with_datasource.py

@@ -4,14 +4,14 @@ from codemie_sdk.models.integration import (
     CredentialTypes,
     IntegrationType,
 )
-from tests import PROJECT
-from tests.integrations import integrations
-from tests.enums.tools import Toolkit, GitTool
-from tests.test_data.git_tools_test_data import (
+from codemie_test_harness.tests import PROJECT
+from codemie_test_harness.tests.integrations import integrations
+from codemie_test_harness.tests.enums.tools import Toolkit, GitTool
+from codemie_test_harness.tests.test_data.git_tools_test_data import (
     list_branches_set_active_branch_test_data,
 )
-from tests.utils.aws_parameters_store import CredentialsUtil
-from tests.utils.constants import test_project_name
+from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.constants import test_project_name
 
 
 git_tool_prompt = list_branches_set_active_branch_test_data[0][2]

{tests → codemie_test_harness/tests}/workflow/assistant_tools/file_management/test_workflow_with_assistant_with_file_management_tools.py

@@ -3,8 +3,8 @@ import os
 import pytest
 from hamcrest import assert_that, contains_string, is_not, all_of
 
-from tests.enums.tools import FileManagementTool, Toolkit
-from tests.test_data.file_management_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import FileManagementTool, Toolkit
+from codemie_test_harness.tests.test_data.file_management_tools_test_data import (
     file_management_tools_test_data,
     GENERATE_IMAGE_TOOL_TASK,
     WRITE_FILE_TASK,
@@ -17,7 +17,7 @@ from tests.test_data.file_management_tools_test_data import (
     RESPONSE_FOR_DIFF_UPDATE,
     RESPONSE_FOR_FILE_EDITOR,
 )
-from tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import get_random_name
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/git/test_workflow_with_assistant_git_tools.py

@@ -1,7 +1,7 @@
 import pytest
 from hamcrest import assert_that, equal_to, not_none
-from tests.utils.base_utils import get_random_name, to_camel_case
-from tests.test_data.git_tools_test_data import (
+from codemie_test_harness.tests.utils.base_utils import get_random_name, to_camel_case
+from codemie_test_harness.tests.test_data.git_tools_test_data import (
     list_branches_set_active_branch_test_data,
     create_branch_test_data,
     create_file_test_data,

{tests → codemie_test_harness/tests}/workflow/assistant_tools/mcp/test_workflow_with_assistant_with_mcp_server.py

@@ -2,7 +2,7 @@ import os
 
 import pytest
 
-from tests.test_data.mcp_server_test_data import (
+from codemie_test_harness.tests.test_data.mcp_server_test_data import (
     FETCH_MCP_SERVER,
     fetch_expected_response,
     time_expected_response,
@@ -10,7 +10,7 @@ from tests.test_data.mcp_server_test_data import (
     time_server_prompt,
     fetch_server_prompt,
 )
-from tests.test_data.mcp_server_test_data import (
+from codemie_test_harness.tests.test_data.mcp_server_test_data import (
     cli_mcp_server_test_data,
     CLI_MCP_SERVER,
 )

{tests → codemie_test_harness/tests}/workflow/assistant_tools/notification/test_workflow_with_assistant_notification_tools.py

@@ -3,8 +3,8 @@ import os
 import pytest
 from hamcrest import assert_that, equal_to
 
-from tests.enums.tools import Toolkit, NotificationTool
-from tests.test_data.notification_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit, NotificationTool
+from codemie_test_harness.tests.test_data.notification_tools_test_data import (
     EMAIL_TOOL_PROMPT,
     EMAIL_RESPONSE,
     EMAIL_SUBJECT,

{tests → codemie_test_harness/tests}/workflow/assistant_tools/open_api/test_workflow_with_assistant_with_open_api_tools.py

@@ -2,8 +2,10 @@ import os
 
 import pytest
 
-from tests.enums.tools import Toolkit
-from tests.test_data.open_api_tools_test_data import open_api_tools_test_data
+from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.test_data.open_api_tools_test_data import (
+    open_api_tools_test_data,
+)
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/plugin/test_workflow_with_assistant_with_development_plugin.py

@@ -2,13 +2,13 @@ import os
 
 import pytest
 
-from tests.enums.tools import Toolkit
-from tests.test_data.plugin_tools_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.test_data.plugin_tools_test_data import (
     list_files_plugin_tools_test_data,
     CREATE_READ_DELETE_FILE_TEST_DATA,
 )
-from tests.utils.base_utils import get_random_name
-from tests.utils.constants import TESTS_PATH
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.constants import TESTS_PATH
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/plugin/test_workflow_with_assistant_with_plugin_and_mcp_servers.py

@@ -2,12 +2,12 @@ import os
 
 import pytest
 
-from tests.enums.tools import Toolkit
-from tests.test_data.mcp_server_test_data import (
+from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.test_data.mcp_server_test_data import (
     cli_mcp_server_with_plugin_test_data,
     filesystem_mcp_server_with_plugin_test_data,
 )
-from tests.utils.constants import TESTS_PATH
+from codemie_test_harness.tests.utils.constants import TESTS_PATH
 
 
 @pytest.mark.regression

{tests → codemie_test_harness/tests}/workflow/assistant_tools/project_management/test_workflow_with_assistant_pm_tools.py

@@ -1,8 +1,10 @@
 import pytest
 
-from tests.enums.tools import Toolkit
-from tests.test_data.project_management_test_data import pm_tools_test_data
-from tests.utils.constants import (
+from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.test_data.project_management_test_data import (
+    pm_tools_test_data,
+)
+from codemie_test_harness.tests.utils.constants import (
     project_management_integrations,
 )