codemie-test-harness 0.1.212__tar.gz → 0.1.213__tar.gz

{codemie_test_harness-0.1.212 → codemie_test_harness-0.1.213}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: codemie-test-harness
-Version: 0.1.212
+Version: 0.1.213
 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.212)
+Requires-Dist: codemie-sdk-python (==0.1.213)
 Requires-Dist: pytest (>=8.4.1,<9.0.0)
 Requires-Dist: pytest-playwright (>=0.7.0,<0.8.0)
 Requires-Dist: pytest-repeat (>=0.9.3,<0.10.0)

{codemie_test_harness-0.1.212 → codemie_test_harness-0.1.213}/codemie_test_harness/tests/assistant/tools/project_management/test_assistant_pm_tools.py

@@ -1,7 +1,7 @@
 import pytest
-
 from codemie_sdk.models.assistant import ToolConfig
 from codemie_sdk.models.integration import CredentialTypes
+
 from codemie_test_harness.tests.enums.tools import Toolkit, ProjectManagementTool
 from codemie_test_harness.tests.test_data.pm_tools_test_data import (
     CONFLUENCE_TOOL_PROMPT,
@@ -13,7 +13,6 @@ from codemie_test_harness.tests.test_data.pm_tools_test_data import (
 from codemie_test_harness.tests.test_data.project_management_test_data import (
     pm_tools_test_data,
 )
-from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 from codemie_test_harness.tests.utils.base_utils import (
     credentials_to_dict,
     assert_tool_triggered,
@@ -21,6 +20,7 @@ from codemie_test_harness.tests.utils.base_utils import (
 from codemie_test_harness.tests.utils.constants import (
     project_management_integrations,
 )
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 
 
 @pytest.mark.assistant
@@ -41,14 +41,14 @@ def test_assistant_with_project_management_tools(
     prompt,
     expected_response,
 ):
-    integration = request.getfixturevalue(
+    _integration = request.getfixturevalue(
         project_management_integrations[integration_type]
     )
 
     assistant = assistant(
         Toolkit.PROJECT_MANAGEMENT,
         tool_name,
-        settings=integration,
+        settings=_integration,
     )
 
     response, triggered_tools = assistant_utils.ask_assistant(

{codemie_test_harness-0.1.212 → codemie_test_harness-0.1.213}/codemie_test_harness/tests/conftest.py

@@ -31,6 +31,7 @@ from codemie_test_harness.tests.test_data.google_datasource_test_data import (
 from codemie_test_harness.tests.utils.assistant_utils import AssistantUtils
 from codemie_test_harness.tests.utils.base_utils import get_random_name, wait_for_entity
 from codemie_test_harness.tests.utils.client_factory import get_client
+from codemie_test_harness.tests.utils.confluence_utils import ConfluenceUtils
 from codemie_test_harness.tests.utils.constants import TESTS_PATH, FILES_PATH
 from codemie_test_harness.tests.utils.conversation_utils import ConversationUtils
 from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
@@ -41,6 +42,7 @@ from codemie_test_harness.tests.utils.env_resolver import (
 )
 from codemie_test_harness.tests.utils.gitbud_utils import GitBudUtils
 from codemie_test_harness.tests.utils.integration_utils import IntegrationUtils
+from codemie_test_harness.tests.utils.jira_utils import JiraUtils
 from codemie_test_harness.tests.utils.llm_utils import LLMUtils
 from codemie_test_harness.tests.utils.logger_util import setup_logger
 from codemie_test_harness.tests.utils.notification_utils import GmailUtils
@@ -967,3 +969,15 @@ def pytest_sessionfinish(session):
 
         providers_utils = ProviderUtils(client)
         providers_utils.cleanup_test_providers()
+
+        # Clean up Jira test issues
+        jira_utils = JiraUtils(is_cloud=False)
+        jira_utils.cleanup_jira_project(summary_prefix=prefix)
+        jira_utils = JiraUtils(is_cloud=True)
+        jira_utils.cleanup_jira_project(summary_prefix=prefix)
+
+        # Clean up Confluence test pages
+        confluence_utils = ConfluenceUtils(is_cloud=False)
+        confluence_utils.cleanup_confluence_space(title_prefix=prefix)
+        confluence_utils = ConfluenceUtils(is_cloud=True)
+        confluence_utils.cleanup_confluence_space(title_prefix=prefix)

{codemie_test_harness-0.1.212 → codemie_test_harness-0.1.213}/codemie_test_harness/tests/test_data/pm_tools_test_data.py

@@ -1,3 +1,7 @@
+from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.confluence_utils import CONFLUENCE_SPACE_KEYS
+from codemie_test_harness.tests.utils.jira_utils import JIRA_PROJECT_KEYS
+
 JIRA_TOOL_PROMPT = (
     "Get a title for ticket EPMCDME-222 ticket."
     "For generic jira tool use exactly the same parameters:"
@@ -89,3 +93,44 @@ RESPONSE_FOR_CONFLUENCE_CLOUD_TOOL = """
     Description: This space includes a section for describing the purpose of the space, a search feature, and a filter by label functionality.
     If you need more detailed information about a specific page, please let me know!
 """
+
+
+def get_jira_tool_create_prompt(jira_type: str) -> str:
+    return (
+        "Create a new issue in Jira with the following details. "
+        "For generic jira tool use exactly the same parameters: "
+        "relative_url=/rest/api/2/issue "
+        "method=POST "
+        f"params={{'fields': {{'project': {{'key': '{JIRA_PROJECT_KEYS[jira_type]}'}}, 'summary': '{get_random_name()} Test issue created by automation', "
+        "'description': 'This is a test issue created for testing purposes', 'issuetype': {'name': 'Task'}}}}}"
+    )
+
+
+def get_response_for_jira_tool_create(jira_type: str) -> str:
+    return f"""
+    The issue has been successfully created in Jira. Here are the details:
+      
+       - **ID**: 13411639
+       - **Key**: {JIRA_PROJECT_KEYS[jira_type]}-.....
+       - **Link**: [{JIRA_PROJECT_KEYS[jira_type]}-.....](https://jira.....)
+"""
+
+
+def get_confluence_tool_create_prompt(confluence_type: str) -> str:
+    return f"""
+    Create a new page in Confluence with the following details using Generic Confluence Tool with the params:
+    relative_url='/rest/api/content'
+    method='POST'
+    params={{'type': 'page', 'title': '{get_random_name()} Test Page Created by Automation', 'space': {{'key': '{CONFLUENCE_SPACE_KEYS[confluence_type]}'}},
+    'body': {{'storage': {{'value': '<p>This is a test page created for testing purposes.</p>', 'representation': 'storage'}}}}}}
+"""
+
+
+def get_response_for_confluence_tool_create(confluence_type: str) -> str:
+    return f"""
+    Successfully created a new page in Confluence.
+    The page has been created with the following details:
+    - Title: Test Page Created by Automation
+    - Space: {CONFLUENCE_SPACE_KEYS[confluence_type]}
+    - Content: This is a test page created for testing purposes.
+"""

{codemie_test_harness-0.1.212 → codemie_test_harness-0.1.213}/codemie_test_harness/tests/test_data/project_management_test_data.py

@@ -2,14 +2,18 @@ import pytest
 
 from codemie_test_harness.tests.enums.tools import ProjectManagementTool
 from codemie_test_harness.tests.test_data.pm_tools_test_data import (
-    JIRA_TOOL_PROMPT,
-    RESPONSE_FOR_JIRA_TOOL,
+    get_confluence_tool_create_prompt,
+    get_response_for_confluence_tool_create,
+    CONFLUENCE_CLOUD_TOOL_PROMPT,
+    RESPONSE_FOR_CONFLUENCE_CLOUD_TOOL,
     CONFLUENCE_TOOL_PROMPT,
     RESPONSE_FOR_CONFLUENCE_TOOL,
-    RESPONSE_FOR_JIRA_CLOUD_TOOL,
+    get_jira_tool_create_prompt,
+    get_response_for_jira_tool_create,
+    JIRA_TOOL_PROMPT,
+    RESPONSE_FOR_JIRA_TOOL,
     JIRA_CLOUD_TOOL_PROMPT,
-    CONFLUENCE_CLOUD_TOOL_PROMPT,
-    RESPONSE_FOR_CONFLUENCE_CLOUD_TOOL,
+    RESPONSE_FOR_JIRA_CLOUD_TOOL,
 )
 from codemie_test_harness.tests.utils.constants import ProjectManagementIntegrationType
 
@@ -46,4 +50,36 @@ pm_tools_test_data = [
         marks=[pytest.mark.confluence, pytest.mark.confluence_cloud],
         id=ProjectManagementIntegrationType.CONFLUENCE_CLOUD,
     ),
+    pytest.param(
+        ProjectManagementTool.JIRA,
+        ProjectManagementIntegrationType.JIRA,
+        get_jira_tool_create_prompt("server"),
+        get_response_for_jira_tool_create("server"),
+        marks=[pytest.mark.jira],
+        id=f"{ProjectManagementIntegrationType.JIRA.value} - create",
+    ),
+    pytest.param(
+        ProjectManagementTool.JIRA,
+        ProjectManagementIntegrationType.JIRA_CLOUD,
+        get_jira_tool_create_prompt("cloud"),
+        get_response_for_jira_tool_create("cloud"),
+        marks=[pytest.mark.jira, pytest.mark.jira_cloud],
+        id=f"{ProjectManagementIntegrationType.JIRA_CLOUD.value} - create",
+    ),
+    pytest.param(
+        ProjectManagementTool.CONFLUENCE,
+        ProjectManagementIntegrationType.CONFLUENCE,
+        get_confluence_tool_create_prompt("server"),
+        get_response_for_confluence_tool_create("server"),
+        marks=[pytest.mark.confluence],
+        id=f"{ProjectManagementIntegrationType.CONFLUENCE.value} - create",
+    ),
+    pytest.param(
+        ProjectManagementTool.CONFLUENCE,
+        ProjectManagementIntegrationType.CONFLUENCE_CLOUD,
+        get_confluence_tool_create_prompt("cloud"),
+        get_response_for_confluence_tool_create("cloud"),
+        marks=[pytest.mark.confluence, pytest.mark.confluence_cloud],
+        id=f"{ProjectManagementIntegrationType.CONFLUENCE_CLOUD.value} - create",
+    ),
 ]

codemie_test_harness-0.1.213/codemie_test_harness/tests/utils/confluence_utils.py

@@ -0,0 +1,128 @@
+import logging
+
+import requests
+
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
+
+logger = logging.getLogger(__name__)
+
+CONFLUENCE_SPACE_KEYS = {"server": "EPMCDMETST", "cloud": "CODEMIE"}
+
+
+class ConfluenceUtils:
+    """Utility class for interacting with Confluence API."""
+
+    def __init__(self, is_cloud: bool = False):
+        """
+        Initialize Confluence utilities with credentials from CredentialsManager.
+
+        Args:
+            is_cloud: Whether to use Confluence Cloud or Confluence Server credentials
+        """
+        self.is_cloud = is_cloud
+
+        if is_cloud:
+            self.confluence_url = CredentialsManager.get_parameter(
+                "CONFLUENCE_CLOUD_URL"
+            )
+            self.token = CredentialsManager.get_parameter("CONFLUENCE_CLOUD_TOKEN")
+            self.email = CredentialsManager.get_parameter("CONFLUENCE_CLOUD_EMAIL")
+            self.auth = (self.email, self.token)
+            self.headers = {
+                "Accept": "application/json",
+                "Content-Type": "application/json",
+            }
+            self.confluence_api = "/wiki/api/v2"
+            self.pages_endpoint = "/pages"
+            self.space_key = CONFLUENCE_SPACE_KEYS["cloud"]
+        else:
+            self.confluence_url = CredentialsManager.get_parameter("CONFLUENCE_URL")
+            self.token = CredentialsManager.get_parameter("CONFLUENCE_TOKEN")
+            self.email = None
+            self.auth = None
+            self.headers = {
+                "Authorization": f"Bearer {self.token}",
+                "Accept": "application/json",
+                "Content-Type": "application/json",
+            }
+            self.confluence_api = "/rest/api"
+            self.pages_endpoint = "/content"
+            self.space_key = CONFLUENCE_SPACE_KEYS["server"]
+
+    def delete_confluence_page(self, page_id: str):
+        """
+        Delete a Confluence page via Confluence API.
+
+        Args:
+            page_id: The Confluence page ID
+        """
+        # Construct API endpoint
+        endpoint = (
+            f"{self.confluence_url}{self.confluence_api}{self.pages_endpoint}/{page_id}"
+        )
+
+        try:
+            logger.info(f"Deleting Confluence page: {page_id}")
+            requests.delete(
+                url=endpoint,
+                headers=self.headers,
+                auth=self.auth,
+                verify=True,
+            )
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Error deleting Confluence page {page_id}: {str(e)}")
+
+    def cleanup_confluence_space(self, title_prefix: str, space_key: str = None):
+        """
+        Clean up Confluence pages in a space with title starting with specified prefix.
+
+        Args:
+            title_prefix: The prefix to filter pages by title (e.g., autotest_entity_prefix)
+            space_key: The Confluence space key (e.g., 'TEST')
+        """
+        space_key = self.space_key if not space_key else space_key
+
+        try:
+            # Search for pages in the space
+            search_endpoint = (
+                f"{self.confluence_url}{self.confluence_api}{self.pages_endpoint}"
+            )
+            params = {
+                "spaceKey": space_key,
+                "type": "page",
+                "limit": 100,
+            }
+
+            logger.info(
+                f"Searching for Confluence pages in space {space_key} with title starting with '{title_prefix}'"
+            )
+            response = requests.get(
+                url=search_endpoint,
+                headers=self.headers,
+                auth=self.auth,
+                params=params,
+                verify=True,
+            )
+
+            if response.status_code == 200:
+                all_pages = response.json().get("results", [])
+
+                # Filter pages by title prefix
+                pages_to_delete = [
+                    page
+                    for page in all_pages
+                    if page.get("title", "").startswith(title_prefix)
+                ]
+
+                # Delete each page
+                for page in pages_to_delete:
+                    self.delete_confluence_page(page["id"])
+            else:
+                logger.error(
+                    f"Failed to search for pages. Status: {response.status_code}, Response: {response.text}"
+                )
+
+        except requests.exceptions.RequestException as e:
+            logger.error(
+                f"Error during Confluence cleanup for space {space_key}: {str(e)}"
+            )

codemie_test_harness-0.1.213/codemie_test_harness/tests/utils/jira_utils.py

@@ -0,0 +1,128 @@
+import logging
+
+import requests
+
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
+
+logger = logging.getLogger(__name__)
+
+JIRA_PROJECT_KEYS = {"server": "EPMCDMETST", "cloud": "SCRUM"}
+
+
+class JiraUtils:
+    """Utility class for interacting with Jira API."""
+
+    def __init__(self, is_cloud: bool = False):
+        """
+        Initialize Jira utilities with credentials from CredentialsManager.
+
+        Args:
+            is_cloud: Whether to use Jira Cloud or Jira Server credentials
+        """
+        self.is_cloud = is_cloud
+
+        if is_cloud:
+            self.jira_url = CredentialsManager.get_parameter("JIRA_CLOUD_URL")
+            self.token = CredentialsManager.get_parameter("JIRA_CLOUD_TOKEN")
+            self.email = CredentialsManager.get_parameter("JIRA_CLOUD_EMAIL")
+            self.auth = (self.email, self.token)
+            self.headers = {
+                "Accept": "application/json",
+                "Content-Type": "application/json",
+            }
+            self.jira_api = "/rest/api/3"
+            self.search_endpoint = "/search/jql"
+            self.project_key = JIRA_PROJECT_KEYS["cloud"]
+        else:
+            self.jira_url = CredentialsManager.get_parameter("JIRA_URL")
+            self.token = CredentialsManager.get_parameter("JIRA_TOKEN")
+            self.email = None
+            self.auth = None
+            self.headers = {
+                "Authorization": f"Bearer {self.token}",
+                "Accept": "application/json",
+                "Content-Type": "application/json",
+            }
+            self.jira_api = "/rest/api/2"
+            self.search_endpoint = "/search"
+            self.project_key = JIRA_PROJECT_KEYS["server"]
+
+    def delete_jira_item(self, issue_key: str):
+        """
+        Delete a Jira issue via Jira API.
+
+        Args:
+            issue_key: The Jira issue key (e.g., 'PROJ-123')
+        """
+        # Construct API endpoint
+        endpoint = f"{self.jira_url}{self.jira_api}/issue/{issue_key}"
+
+        try:
+            logger.info(f"Deleting Jira issue: {issue_key}")
+            requests.delete(
+                url=endpoint,
+                headers=self.headers,
+                auth=self.auth,
+                verify=True,
+            )
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Error deleting Jira issue {issue_key}: {str(e)}")
+
+    def cleanup_jira_project(self, summary_prefix: str, project_key: str = None):
+        """
+        Clean up Jira issues in a project with summary starting with specified prefix.
+
+        Args:
+            project_key: The Jira project key (e.g., 'EPMCDMETST')
+            summary_prefix: The prefix to filter issues by summary (e.g., autotest_entity_prefix)
+        """
+        # Construct JQL query to find all issues in the project
+        # We'll filter by summary prefix in Python since JQL text search has limitations
+        project_key = self.project_key if not project_key else project_key
+
+        jql = f'project = "{project_key}"'
+
+        try:
+            # Search for issues matching the JQL
+            search_endpoint = f"{self.jira_url}{self.jira_api}{self.search_endpoint}"
+            params = {
+                "jql": jql,
+                "maxResults": 100,
+                "fields": "key,summary",
+            }
+
+            logger.info(
+                f"Searching for Jira issues in project {project_key} with summary starting with '{summary_prefix}'"
+            )
+            response = requests.get(
+                url=search_endpoint,
+                headers=self.headers,
+                auth=self.auth,
+                params=params,
+                verify=True,
+            )
+
+            if response.status_code == 200:
+                all_issues = response.json().get("issues", [])
+
+                # Filter issues by summary prefix in Python
+                issues_to_delete = [
+                    issue
+                    for issue in all_issues
+                    if issue.get("fields", {})
+                    .get("summary", "")
+                    .startswith(summary_prefix)
+                ]
+
+                # Delete each issue
+                for issue in issues_to_delete:
+                    self.delete_jira_item(issue["key"])
+            else:
+                logger.error(
+                    f"Failed to search for issues. Status: {response.status_code}, Response: {response.text}"
+                )
+
+        except requests.exceptions.RequestException as e:
+            logger.error(
+                f"Error during Jira cleanup for project {project_key}: {str(e)}"
+            )

{codemie_test_harness-0.1.212 → codemie_test_harness-0.1.213}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-test-harness"
-version = "0.1.212"
+version = "0.1.213"
 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.212"
+codemie-sdk-python = "0.1.213"
 pytest = "^8.4.1"
 python-dotenv = "^1.1.0"
 PyHamcrest = "^2.1.0"