codemie-test-harness 0.1.162__py3-none-any.whl → 0.1.164__py3-none-any.whl

codemie_test_harness/tests/test_data/direct_tools/cloud_tools_test_data.py

@@ -2,7 +2,7 @@ import pytest
 
 from codemie_test_harness.tests.enums.tools import Toolkit, CloudTool
 from codemie_sdk.models.integration import CredentialTypes
-from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 from codemie_test_harness.tests.utils.env_resolver import EnvironmentResolver
 
 cloud_test_data = [
@@ -10,7 +10,7 @@ cloud_test_data = [
         Toolkit.CLOUD,
         CloudTool.AWS,
         CredentialTypes.AWS,
-        CredentialsUtil.aws_credentials(),
+        CredentialsManager.aws_credentials(),
         {
             "query": {
                 "service": "s3",
@@ -105,7 +105,7 @@ cloud_test_data = [
         Toolkit.CLOUD,
         CloudTool.AZURE,
         CredentialTypes.AZURE,
-        CredentialsUtil.azure_credentials(),
+        CredentialsManager.azure_credentials(),
         {
             "method": "GET",
             "url": "https://management.azure.com/subscriptions/08679d2f-8945-4e08-8df8-b8e58626b13a/resourceGroups/krci-codemie-azure-env-rg?api-version=2021-04-01",
@@ -131,7 +131,7 @@ cloud_test_data = [
         Toolkit.CLOUD,
         CloudTool.GCP,
         CredentialTypes.GCP,
-        CredentialsUtil.gcp_credentials(),
+        CredentialsManager.gcp_credentials(),
         {
             "method": "GET",
             "scopes": ["https://www.googleapis.com/auth/cloud-platform"],
@@ -181,7 +181,7 @@ cloud_test_data = [
         Toolkit.CLOUD,
         CloudTool.KUBERNETES,
         CredentialTypes.KUBERNETES,
-        CredentialsUtil.kubernetes_credentials(),
+        CredentialsManager.kubernetes_credentials(),
         {"method": "GET", "suburl": "/api/v1/namespaces/argocd/services"},
         """
             {

codemie_test_harness/tests/test_data/direct_tools/data_management_tools_test_data.py

@@ -1,9 +1,14 @@
 import pytest
 
-from codemie_test_harness.tests.enums.tools import DataManagementTool, Toolkit
 from codemie_test_harness.tests.enums.integrations import DataBaseDialect
+from codemie_test_harness.tests.enums.tools import DataManagementTool, Toolkit
 from codemie_test_harness.tests.utils.env_resolver import EnvironmentResolver
 
+ELASTIC_TOOL_TASK = {
+    "index": "_all",
+    "query": '{"query": {"prefix": {"_index": "codemie"}}}',
+}
+
 sql_tools_test_data = [
     (
         Toolkit.DATA_MANAGEMENT,

codemie_test_harness/tests/test_data/direct_tools/file_management_tools_test_data.py

@@ -50,7 +50,7 @@ file_management_tools_test_data = [
         {"file_path": "/tmp/env.properties", "text": "env=preview"},
         "File written successfully to /tmp/env.properties",
     ),
-    (
+    pytest.param(
         FileManagementTool.RUN_COMMAND_LINE,
         {"command": "ls /usr"},
         """
@@ -65,5 +65,9 @@ file_management_tools_test_data = [
          share
          src
          ', '', 0, 1753875309836) """,
+        marks=pytest.mark.skipif(
+            EnvironmentResolver.is_localhost(),
+            reason="Skipping this test on local environment",
+        ),
     ),
 ]

codemie_test_harness/tests/test_data/integrations_test_data.py

@@ -2,13 +2,13 @@ import pytest
 
 from codemie_sdk.models.integration import CredentialTypes
 from codemie_test_harness.tests.enums.integrations import DataBaseDialect
-from codemie_test_harness.tests.utils.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 from codemie_test_harness.tests.utils.env_resolver import EnvironmentResolver
 
 valid_integrations = [
     pytest.param(
         CredentialTypes.AWS,
-        CredentialsUtil.aws_credentials(),
+        CredentialsManager.aws_credentials(),
         marks=[
             pytest.mark.aws,
             pytest.mark.cloud,
@@ -17,7 +17,7 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.AZURE,
-        CredentialsUtil.azure_credentials(),
+        CredentialsManager.azure_credentials(),
         marks=[
             pytest.mark.azure,
             pytest.mark.cloud,
@@ -26,7 +26,7 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.GCP,
-        CredentialsUtil.gcp_credentials(),
+        CredentialsManager.gcp_credentials(),
         marks=[
             pytest.mark.gcp,
             pytest.mark.cloud,
@@ -39,31 +39,31 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.SONAR,
-        CredentialsUtil.sonar_credentials(),
+        CredentialsManager.sonar_credentials(),
         marks=pytest.mark.sonar,
         id=f"{CredentialTypes.SONAR}_server",
     ),
     pytest.param(
         CredentialTypes.SONAR,
-        CredentialsUtil.sonar_cloud_credentials(),
+        CredentialsManager.sonar_cloud_credentials(),
         marks=pytest.mark.sonar,
         id=f"{CredentialTypes.SONAR}_cloud",
     ),
     pytest.param(
         CredentialTypes.GIT,
-        CredentialsUtil.gitlab_credentials(),
+        CredentialsManager.gitlab_credentials(),
         marks=pytest.mark.gitlab,
         id=f"{CredentialTypes.GIT}_gitlab",
     ),
     pytest.param(
         CredentialTypes.GIT,
-        CredentialsUtil.github_credentials(),
+        CredentialsManager.github_credentials(),
         marks=pytest.mark.github,
         id=f"{CredentialTypes.GIT}_github",
     ),
     pytest.param(
         CredentialTypes.CONFLUENCE,
-        CredentialsUtil.confluence_credentials(),
+        CredentialsManager.confluence_credentials(),
         marks=[
             pytest.mark.confluence,
             pytest.mark.project_management,
@@ -72,7 +72,7 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.CONFLUENCE,
-        CredentialsUtil.confluence_cloud_credentials(),
+        CredentialsManager.confluence_cloud_credentials(),
         marks=[
             pytest.mark.confluence,
             pytest.mark.confluence_cloud,
@@ -82,7 +82,7 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.JIRA,
-        CredentialsUtil.jira_credentials(),
+        CredentialsManager.jira_credentials(),
         marks=[
             pytest.mark.jira,
             pytest.mark.project_management,
@@ -91,7 +91,7 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.JIRA,
-        CredentialsUtil.jira_cloud_credentials(),
+        CredentialsManager.jira_cloud_credentials(),
         marks=[
             pytest.mark.jira,
             pytest.mark.jira_cloud,
@@ -101,43 +101,43 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.SQL,
-        CredentialsUtil.sql_credentials(DataBaseDialect.POSTGRES),
+        CredentialsManager.sql_credentials(DataBaseDialect.POSTGRES),
         marks=pytest.mark.sql,
         id=DataBaseDialect.POSTGRES,
     ),
     pytest.param(
         CredentialTypes.SQL,
-        CredentialsUtil.sql_credentials(DataBaseDialect.MY_SQL),
+        CredentialsManager.sql_credentials(DataBaseDialect.MY_SQL),
         marks=pytest.mark.sql,
         id=DataBaseDialect.MY_SQL,
     ),
     pytest.param(
         CredentialTypes.ELASTIC,
-        CredentialsUtil.elastic_credentials(),
+        CredentialsManager.elasticsearch_credentials(),
         marks=pytest.mark.elastic,
         id=CredentialTypes.ELASTIC,
     ),
     pytest.param(
         CredentialTypes.MCP,
-        CredentialsUtil.mcp_credentials(),
+        CredentialsManager.mcp_credentials(),
         marks=pytest.mark.mcp,
         id=CredentialTypes.MCP,
     ),
     pytest.param(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.azure_devops_credentials(),
+        CredentialsManager.azure_devops_credentials(),
         marks=pytest.mark.azure,
         id=CredentialTypes.AZURE_DEVOPS,
     ),
     pytest.param(
         CredentialTypes.FILESYSTEM,
-        CredentialsUtil.file_system_credentials(),
+        CredentialsManager.file_system_credentials(),
         marks=pytest.mark.file_system,
         id=CredentialTypes.FILESYSTEM,
     ),
     pytest.param(
         CredentialTypes.EMAIL,
-        CredentialsUtil.gmail_credentials(),
+        CredentialsManager.gmail_credentials(),
         marks=[
             pytest.mark.notification,
             pytest.mark.email,
@@ -146,7 +146,7 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.TELEGRAM,
-        CredentialsUtil.telegram_credentials(),
+        CredentialsManager.telegram_credentials(),
         marks=[
             pytest.mark.notification,
             pytest.mark.telegram,
@@ -155,19 +155,19 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.SERVICE_NOW,
-        CredentialsUtil.servicenow_credentials(),
+        CredentialsManager.servicenow_credentials(),
         marks=pytest.mark.servicenow,
         id=CredentialTypes.SERVICE_NOW,
     ),
     pytest.param(
         CredentialTypes.KEYCLOAK,
-        CredentialsUtil.keycloak_credentials(),
+        CredentialsManager.keycloak_credentials(),
         marks=pytest.mark.keycloak,
         id=CredentialTypes.KEYCLOAK,
     ),
     pytest.param(
         CredentialTypes.KUBERNETES,
-        CredentialsUtil.kubernetes_credentials(),
+        CredentialsManager.kubernetes_credentials(),
         marks=[
             pytest.mark.kubernetes,
             pytest.mark.cloud,
@@ -180,7 +180,7 @@ valid_integrations = [
     ),
     pytest.param(
         CredentialTypes.REPORT_PORTAL,
-        CredentialsUtil.report_portal_credentials(),
+        CredentialsManager.report_portal_credentials(),
         marks=pytest.mark.report_portal,
         id=CredentialTypes.REPORT_PORTAL,
     ),
@@ -189,7 +189,7 @@ valid_integrations = [
 testable_integrations = [
     pytest.param(
         CredentialTypes.AWS,
-        CredentialsUtil.aws_credentials(),
+        CredentialsManager.aws_credentials(),
         marks=[
             pytest.mark.aws,
             pytest.mark.cloud,
@@ -198,7 +198,7 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.AZURE,
-        CredentialsUtil.azure_credentials(),
+        CredentialsManager.azure_credentials(),
         marks=[
             pytest.mark.azure,
             pytest.mark.cloud,
@@ -207,7 +207,7 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.GCP,
-        CredentialsUtil.gcp_credentials(),
+        CredentialsManager.gcp_credentials(),
         marks=[
             pytest.mark.gcp,
             pytest.mark.cloud,
@@ -220,19 +220,19 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.SONAR,
-        CredentialsUtil.sonar_credentials(),
+        CredentialsManager.sonar_credentials(),
         marks=pytest.mark.sonar,
         id=f"{CredentialTypes.SONAR}_server",
     ),
     pytest.param(
         CredentialTypes.SONAR,
-        CredentialsUtil.sonar_cloud_credentials(),
+        CredentialsManager.sonar_cloud_credentials(),
         marks=pytest.mark.sonar,
         id=f"{CredentialTypes.SONAR}_cloud",
     ),
     pytest.param(
         CredentialTypes.CONFLUENCE,
-        CredentialsUtil.confluence_credentials(),
+        CredentialsManager.confluence_credentials(),
         marks=[
             pytest.mark.confluence,
             pytest.mark.project_management,
@@ -241,7 +241,7 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.CONFLUENCE,
-        CredentialsUtil.confluence_cloud_credentials(),
+        CredentialsManager.confluence_cloud_credentials(),
         marks=[
             pytest.mark.confluence,
             pytest.mark.confluence_cloud,
@@ -251,7 +251,7 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.JIRA,
-        CredentialsUtil.jira_credentials(),
+        CredentialsManager.jira_credentials(),
         marks=[
             pytest.mark.jira,
             pytest.mark.project_management,
@@ -260,7 +260,7 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.JIRA,
-        CredentialsUtil.jira_cloud_credentials(),
+        CredentialsManager.jira_cloud_credentials(),
         marks=[
             pytest.mark.jira,
             pytest.mark.jira_cloud,
@@ -270,7 +270,7 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.EMAIL,
-        CredentialsUtil.gmail_credentials(),
+        CredentialsManager.gmail_credentials(),
         marks=[
             pytest.mark.email,
             pytest.mark.notification,
@@ -283,13 +283,13 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.SERVICE_NOW,
-        CredentialsUtil.servicenow_credentials(),
+        CredentialsManager.servicenow_credentials(),
         marks=pytest.mark.servicenow,
         id=CredentialTypes.SERVICE_NOW,
     ),
     pytest.param(
         CredentialTypes.KUBERNETES,
-        CredentialsUtil.kubernetes_credentials(),
+        CredentialsManager.kubernetes_credentials(),
         marks=[
             pytest.mark.kubernetes,
             pytest.mark.cloud,
@@ -302,7 +302,7 @@ testable_integrations = [
     ),
     pytest.param(
         CredentialTypes.REPORT_PORTAL,
-        CredentialsUtil.report_portal_credentials(),
+        CredentialsManager.report_portal_credentials(),
         marks=pytest.mark.report_portal,
         id=CredentialTypes.REPORT_PORTAL,
     ),
@@ -311,7 +311,7 @@ testable_integrations = [
 invalid_integrations = [
     pytest.param(
         CredentialTypes.AWS,
-        CredentialsUtil.invalid_aws_credentials(),
+        CredentialsManager.invalid_aws_credentials(),
         "An error occurred (SignatureDoesNotMatch) when calling the GetUser operation: The request signature we calculated does not match the signature you provided. Check your AWS Secret Access Key and signing method. Consult the service documentation for details.",
         marks=[
             pytest.mark.aws,
@@ -321,7 +321,7 @@ invalid_integrations = [
     ),
     pytest.param(
         CredentialTypes.AZURE,
-        CredentialsUtil.invalid_azure_credentials(),
+        CredentialsManager.invalid_azure_credentials(),
         "Invalid client secret provided. Ensure the secret being sent in the request is the client secret value, not the client secret ID, for a secret added to app",
         marks=[
             pytest.mark.azure,
@@ -331,7 +331,7 @@ invalid_integrations = [
     ),
     pytest.param(
         CredentialTypes.GCP,
-        CredentialsUtil.invalid_gcp_credentials(),
+        CredentialsManager.invalid_gcp_credentials(),
         "Error: ('Could not deserialize key data. The data may be in an incorrect format, "
         + "the provided password may be incorrect, it may be encrypted with an unsupported algorithm, "
         + "or it may be an unsupported key type",
@@ -347,21 +347,21 @@ invalid_integrations = [
     ),
     pytest.param(
         CredentialTypes.SONAR,
-        CredentialsUtil.invalid_sonar_credentials(),
+        CredentialsManager.invalid_sonar_credentials(),
         "Invalid token",
         marks=pytest.mark.sonar,
         id=f"{CredentialTypes.SONAR}_server",
     ),
     pytest.param(
         CredentialTypes.SONAR,
-        CredentialsUtil.invalid_sonar_cloud_credentials(),
+        CredentialsManager.invalid_sonar_cloud_credentials(),
         "Invalid token",
         marks=pytest.mark.sonar,
         id=f"{CredentialTypes.SONAR}_cloud",
     ),
     pytest.param(
         CredentialTypes.EMAIL,
-        CredentialsUtil.invalid_gmail_credentials(),
+        CredentialsManager.invalid_gmail_credentials(),
         "SMTP Code: 535. SMTP error",
         marks=[
             pytest.mark.email,
@@ -375,28 +375,28 @@ invalid_integrations = [
     ),
     pytest.param(
         CredentialTypes.JIRA,
-        CredentialsUtil.invalid_jira_credentials(),
+        CredentialsManager.invalid_jira_credentials(),
         "Unauthorized (401)",
         marks=pytest.mark.jira,
         id=CredentialTypes.JIRA,
     ),
     pytest.param(
         CredentialTypes.CONFLUENCE,
-        CredentialsUtil.invalid_confluence_credentials(),
+        CredentialsManager.invalid_confluence_credentials(),
         "Access denied",
         marks=pytest.mark.confluence,
         id=CredentialTypes.CONFLUENCE,
     ),
     pytest.param(
         CredentialTypes.SERVICE_NOW,
-        CredentialsUtil.invalid_servicenow_credentials(),
+        CredentialsManager.invalid_servicenow_credentials(),
         'ServiceNow tool exception. Status: 401. Response: {"error":{"message":"User Not Authenticated","detail":"Required to provide Auth information"}',
         marks=pytest.mark.servicenow,
         id=CredentialTypes.SERVICE_NOW,
     ),
     pytest.param(
         CredentialTypes.KUBERNETES,
-        CredentialsUtil.invalid_kubernetes_credentials(),
+        CredentialsManager.invalid_kubernetes_credentials(),
         "Error: (401)\nReason: Unauthorized",
         marks=[
             pytest.mark.kubernetes,
@@ -406,7 +406,7 @@ invalid_integrations = [
     ),
     pytest.param(
         CredentialTypes.REPORT_PORTAL,
-        CredentialsUtil.invalid_report_portal_credentials(),
+        CredentialsManager.invalid_report_portal_credentials(),
         "401 Client Error:  for url: https://report-portal.core.kuberocketci.io/api/v1/epm-cdme/launch?page.page=1",
         marks=pytest.mark.report_portal,
         id=CredentialTypes.REPORT_PORTAL,

codemie_test_harness/tests/test_data/llm_test_data.py

@@ -53,7 +53,7 @@ MODEL_RESPONSES = [
     LlmResponseData(ModelTypes.CLAUDE_4_SONNET, AWS_ENVS),
     LlmResponseData(ModelTypes.CLAUDE_4_OPUS, AWS_ENVS),
     LlmResponseData(ModelTypes.CLAUDE_4_1_OPUS, AWS_ENVS),
-    LlmResponseData(ModelTypes.CLAUDE_4_SONNET_1M, [Environment.AWS]),
+    LlmResponseData(ModelTypes.CLAUDE_4_SONNET_1M, AWS_ENVS),
     # Other LLMs test data
     LlmResponseData(ModelTypes.RLAB_QWQ_32B, OTHER_ENVS),
     LlmResponseData(ModelTypes.DEEPSEEK_R1, OTHER_ENVS),

codemie_test_harness/tests/test_data/notification_tools_test_data.py

@@ -1,10 +1,6 @@
-from codemie_test_harness.tests.utils.aws_parameters_store import AwsParameterStore
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 
-CHAT_ID = (
-    AwsParameterStore.get_cloud_provider_credentials("telegram")
-    .get("telegram", {})
-    .get("chat_id")
-)
+CHAT_ID = CredentialsManager.get_parameter("TELEGRAM_CHAT_ID")
 
 EMAIL_SUBJECT = "Test email"