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

{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):

codemie_test_harness/tests/utils/http_utils.py

@@ -0,0 +1,141 @@
+import logging
+from typing import TypeVar, Type, Optional, Any, Union, Dict, List
+
+import requests
+from codemie_sdk.utils.http import ApiRequestHandler
+from codemie_sdk.utils.http import log_request
+from pydantic import BaseModel
+
+T = TypeVar("T", bound=Union[BaseModel, List[BaseModel], dict])
+
+logger = logging.getLogger(__name__)
+
+
+class RequestHandler(ApiRequestHandler):
+    """Handles HTTP requests with consistent error handling and response parsing."""
+
+    @log_request
+    def post(
+        self,
+        endpoint: str,
+        response_model: Type[T] = None,
+        json_data: Optional[Dict[str, Any]] = None,
+        stream: bool = False,
+        wrap_response: bool = True,
+    ) -> Union[T, requests.Response]:
+        """Makes a POST request and parses the response.
+
+        Args:
+            endpoint: API endpoint path
+            response_model: Pydantic model class or List[Model] for response
+            json_data: JSON request body
+            stream: Whether to return streaming response
+            wrap_response: Whether response is wrapped in 'data' field
+
+        Returns:
+            Parsed response object/list or streaming response
+        """
+        if json_data:
+            logger.debug(f"Request body: {json_data}")
+
+        response = requests.post(
+            url=f"{self._base_url}{endpoint}",
+            headers=self._get_headers(),
+            json=json_data,
+            verify=self._verify_ssl,
+            stream=stream,
+        )
+
+        if stream:
+            return response
+
+        return self._parse_response(response, response_model, wrap_response)
+
+    @log_request
+    def get(
+        self,
+        endpoint: str,
+        response_model: Type[T] = None,
+        params: Optional[Dict[str, Any]] = None,
+        wrap_response: bool = True,
+    ) -> Union[T, requests.Response]:
+        """Makes a GET request and parses the response.
+
+        Args:
+            endpoint: API endpoint path
+            response_model: Pydantic model class or List[Model] for response
+            params: Query parameters
+            wrap_response: Whether response is wrapped in 'data' field
+
+        Returns:
+            Parsed response object or list of objects
+        """
+        if params:
+            logger.debug(f"Request params: {params}")
+        response = requests.get(
+            url=f"{self._base_url}{endpoint}",
+            headers=self._get_headers(),
+            params=params,
+            verify=self._verify_ssl,
+        )
+
+        if response_model:
+            return self._parse_response(response, response_model, wrap_response)
+
+        return response
+
+    @log_request
+    def put(
+        self,
+        endpoint: str,
+        json_data: Dict[str, Any],
+        response_model: Type[T] = None,
+        params: Optional[Dict[str, Any]] = None,
+        wrap_response: bool = True,
+    ) -> Union[T, requests.Response]:
+        """Makes a PUT request and parses the response.
+
+        Args:
+            endpoint: API endpoint path
+            response_model: Pydantic model class or List[Model] for response
+            json_data: JSON request body
+            params: Query parameters
+            wrap_response: Whether response is wrapped in 'data' field
+
+        Returns:
+            Parsed response object or list of objects
+        """
+        logger.debug(f"Request body: {json_data}")
+        response = requests.put(
+            url=f"{self._base_url}{endpoint}",
+            headers=self._get_headers(),
+            json=json_data,
+            params=params,
+            verify=self._verify_ssl,
+        )
+
+        if response_model:
+            return self._parse_response(response, response_model, wrap_response)
+
+        return response
+
+    @log_request
+    def delete(
+        self,
+        endpoint: str,
+    ) -> requests.Response:
+        """Makes a DELETE request and parses the response.
+
+        Args:
+            endpoint: API endpoint path
+
+        Returns:
+            response object
+        """
+        response = requests.delete(
+            url=f"{self._base_url}{endpoint}",
+            headers=self._get_headers(),
+            verify=self._verify_ssl,
+        )
+
+        return response

{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__)
 

codemie_test_harness/tests/utils/provider_utils.py

@@ -0,0 +1,178 @@
+import json
+import logging
+import os
+from typing import Optional, Dict, Any
+
+import requests
+from codemie_sdk import CodeMieClient
+
+from tests import autotest_entity_prefix
+from tests.utils import api_domain, verify_ssl
+from tests.utils.base_utils import (
+    BaseUtils,
+    get_random_name,
+)
+from tests.utils.http_utils import RequestHandler
+
+logger = logging.getLogger(__name__)
+
+providers_endpoint = "/v1/providers"
+
+
+class ProviderUtils(BaseUtils):
+    def __init__(self, client: CodeMieClient):
+        """Initialize the provides service."""
+
+        super().__init__(client)
+        self._api = RequestHandler(api_domain, self.client.token, verify_ssl)
+
+    @staticmethod
+    def provider_request_json():
+        """Load provider request JSON from file."""
+        payload_path = os.path.join(
+            os.path.dirname(__file__), "../test_data/files/provider_payload.json"
+        )
+        with open(payload_path, "r") as pyload_file:
+            request_json = json.load(pyload_file)
+
+        return request_json
+
+    def send_post_request_to_providers_endpoint(
+        self, request: Dict[str, Any]
+    ) -> requests.Response:
+        """
+        Send request to provider creation endpoint without raising error for response status codes.
+
+        Args:
+            request: The provider creation request
+
+        Returns:
+            Raw response from '/v1/providers' endpoint
+        """
+        return self._api.post(providers_endpoint, json_data=request, stream=True)
+
+    def send_get_request_to_providers_endpoint(
+        self,
+    ) -> requests.Response:
+        """
+        Send request to get providers endpoint.
+
+        Returns:
+            Raw response from '/v1/providers' endpoint
+        """
+        return self._api.get(providers_endpoint)
+
+    def send_get_request_datasource_schemas_endpoint(
+        self,
+    ) -> requests.Response:
+        """
+        Send request to get datasource_schemas endpoint.
+
+        Returns:
+            Raw response from '/v1/providers/datasource_schemas' endpoint
+        """
+        return self._api.get(f"{providers_endpoint}/datasource_schemas")
+
+    def create_provider(
+        self,
+    ) -> str:
+        """
+        Creates a provider for test.
+
+        Returns:
+            Created provider id
+        """
+        request_json = self.provider_request_json()
+        # Ensure unique provider name to avoid conflicts
+        request_json["name"] = get_random_name()
+
+        response = self.send_post_request_to_providers_endpoint(request_json)
+
+        return response.json()["id"]
+
+    def list_providers(
+        self,
+    ) -> requests.Response:
+        """
+        List all providers.
+
+        Returns:
+            List of provider objects
+        """
+        response = self.send_get_request_to_providers_endpoint()
+
+        return response
+
+    def get_provider_by_id(self, provider_id: str) -> requests.Response:
+        """
+        Get a specific provider by ID.
+
+        Args:
+            provider_id: The provider ID
+
+        Returns:
+            Provider object
+        """
+        response = self._api.get(f"{providers_endpoint}/{provider_id}")
+
+        return response
+
+    def update_provider(
+        self,
+        provider_id: str,
+        provider_update_payload: dict,
+    ) -> requests.Response:
+        """
+        Update an existing provider.
+
+        Args:
+            provider_id: The provider ID to update
+            provider_update_payload: request body
+
+        Returns:
+            Updated provider object
+        """
+        endpoint = f"{providers_endpoint}/{provider_id}"
+
+        return self._api.put(endpoint, json_data=provider_update_payload)
+
+    def send_delete_provider_request(self, provider_id: str) -> requests.Response:
+        """
+        Delete a provider by ID.
+
+        Args:
+            provider_id: The provider ID to delete
+
+        Returns:
+            Response from delete operation
+        """
+        return self._api.delete(f"{providers_endpoint}/{provider_id}")
+
+    def cleanup_test_providers(self, name_prefix: Optional[str] = None):
+        """
+        Clean up test providers (those with autotest prefix or specified prefix).
+
+        Args:
+            name_prefix: Prefix to match for cleanup (uses autotest prefix if None)
+        """
+
+        prefix = name_prefix if name_prefix else autotest_entity_prefix
+
+        try:
+            providers = self.list_providers()
+            test_providers = [
+                provider
+                for provider in providers.json()
+                if provider["name"].startswith(prefix)
+            ]
+
+            for provider in test_providers:
+                try:
+                    self.send_delete_provider_request(provider["id"])
+                except Exception as e:
+                    # Log but don't fail cleanup for individual provider
+                    logger.error(f"Failed to delete provider {provider['name']}: {e}")
+
+        except Exception as e:
+            # Log but don't fail if cleanup encounters issues
+            logger.error(f"Provider cleanup encountered an error: {e}")

{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