codemie-test-harness 0.1.162__tar.gz → 0.1.164__tar.gz

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: codemie-test-harness
-Version: 0.1.162
+Version: 0.1.164
 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.162)
+Requires-Dist: codemie-sdk-python (==0.1.164)
 Requires-Dist: pytest (>=8.4.1,<9.0.0)
 Requires-Dist: pytest-playwright (>=0.7.0,<0.8.0)
 Requires-Dist: pytest-reportportal (>=5.5.2,<6.0.0)

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/__init__.py

@@ -6,6 +6,7 @@ from pathlib import Path
 from dotenv import load_dotenv
 
 from codemie_test_harness.cli.runner import resolve_tests_path_and_root
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 
 _, root_dir = resolve_tests_path_and_root()
 env_file_path = Path(root_dir) / ".env"
@@ -19,7 +20,7 @@ LANGFUSE_TRACES_ENABLED = (
 
 PROJECT = os.getenv("PROJECT_NAME", "codemie")
 TEST_USER = os.getenv("TEST_USER_FULL_NAME", "Test User")
-GITHUB_URL = os.getenv("GITHUB_URL", "https://github.com")
+GITHUB_URL = CredentialsManager.get_parameter("GITHUB_URL", "https://github.com")
 
 autotest_entity_prefix = (
     f"{''.join(random.choice(string.ascii_lowercase) for _ in range(3))}_"

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/datasource/test_confluence_datasource.py

@@ -12,7 +12,7 @@ from codemie_test_harness.tests.test_data.pm_tools_test_data import (
     CONFLUENCE_CLOUD_TOOL_PROMPT,
     RESPONSE_FOR_CONFLUENCE_CLOUD_TOOL,
 )
-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.base_utils import get_random_name, assert_response
 
 
@@ -20,7 +20,7 @@ from codemie_test_harness.tests.utils.base_utils import get_random_name, assert_
 def confluence_cloud_datasource(datasource_utils, confluence_cloud_integration):
     datasource = datasource_utils.create_confluence_datasource(
         setting_id=confluence_cloud_integration.id,
-        cql=CredentialsUtil.confluence_cloud_jql(),
+        cql=CredentialsManager.confluence_cloud_cql(),
     )
     yield datasource
     if datasource:

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/datasource/test_jira_datasource.py

@@ -11,14 +11,14 @@ from codemie_test_harness.tests.test_data.pm_tools_test_data import (
     JIRA_CLOUD_TOOL_PROMPT,
     RESPONSE_FOR_JIRA_CLOUD_TOOL,
 )
-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.base_utils import get_random_name, assert_response
 
 
 @pytest.fixture(scope="function")
 def jira_cloud_datasource(datasource_utils, jira_cloud_integration):
     datasource = datasource_utils.create_jira_datasource(
-        setting_id=jira_cloud_integration.id, jql=CredentialsUtil.jira_cloud_jql()
+        setting_id=jira_cloud_integration.id, jql=CredentialsManager.jira_cloud_jql()
     )
     yield datasource
     if datasource:

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/default_integrations/test_default_integrations_for_tool.py

@@ -11,7 +11,7 @@ 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.credentials_manager import CredentialsManager
 from codemie_test_harness.tests.utils.constants import test_project_name
 
 
@@ -48,20 +48,20 @@ def test_assistant_should_use_user_integration_by_default(
         )
 
     integration_utils.create_user_integration(
-        CredentialTypes.JIRA, CredentialsUtil.jira_credentials(), test_project_name
+        CredentialTypes.JIRA, CredentialsManager.jira_credentials(), test_project_name
     )
 
     if is_global:
         integration_utils.create_global_integration(
             CredentialTypes.JIRA,
-            CredentialsUtil.invalid_jira_credentials(),
+            CredentialsManager.invalid_jira_credentials(),
             test_project_name,
         )
 
     if project_integration:
         integration_utils.create_project_integration(
             CredentialTypes.JIRA,
-            CredentialsUtil.invalid_jira_credentials(),
+            CredentialsManager.invalid_jira_credentials(),
             test_project_name,
         )
 
@@ -100,12 +100,12 @@ def test_assistant_with_global_and_project_integration(
     # Create only global (valid) and project (invalid) integrations.
     # Global integration is created in project which is different from the project of the assistant.
     integration_utils.create_global_integration(
-        CredentialTypes.JIRA, CredentialsUtil.jira_credentials(), PROJECT
+        CredentialTypes.JIRA, CredentialsManager.jira_credentials(), PROJECT
     )
 
     integration_utils.create_project_integration(
         CredentialTypes.JIRA,
-        CredentialsUtil.invalid_jira_credentials(),
+        CredentialsManager.invalid_jira_credentials(),
         test_project_name,
     )
 
@@ -141,7 +141,7 @@ def test_assistant_with_project_integration_only(
 
     # Create project integration only
     integration_utils.create_project_integration(
-        CredentialTypes.JIRA, CredentialsUtil.jira_credentials(), test_project_name
+        CredentialTypes.JIRA, CredentialsManager.jira_credentials(), test_project_name
     )
 
     jira_assistant = assistant(
@@ -174,13 +174,13 @@ def test_assistant_with_global_valid_and_user_invalid_integration(
 
     # Create global valid integration.
     integration_utils.create_global_integration(
-        CredentialTypes.JIRA, CredentialsUtil.jira_credentials(), PROJECT
+        CredentialTypes.JIRA, CredentialsManager.jira_credentials(), PROJECT
     )
 
     # Create user invalid integration.
     integration_utils.create_user_integration(
         CredentialTypes.JIRA,
-        CredentialsUtil.invalid_jira_credentials(),
+        CredentialsManager.invalid_jira_credentials(),
         PROJECT,
     )
 
@@ -214,13 +214,13 @@ def test_assistant_with_project_valid_and_user_invalid_integration(
 
     # Create project valid integration.
     integration_utils.create_project_integration(
-        CredentialTypes.JIRA, CredentialsUtil.jira_credentials(), test_project_name
+        CredentialTypes.JIRA, CredentialsManager.jira_credentials(), test_project_name
     )
 
     # Create user invalid integration.
     integration_utils.create_user_integration(
         CredentialTypes.JIRA,
-        CredentialsUtil.invalid_jira_credentials(),
+        CredentialsManager.invalid_jira_credentials(),
         PROJECT,
     )
 

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/default_integrations/test_default_integrations_for_tool_kit.py

@@ -10,7 +10,7 @@ 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.credentials_manager import CredentialsManager
 from codemie_test_harness.tests.utils.constants import test_project_name
 
 ado_wiki_prompt = ado_wiki_get_test_data[0][2]
@@ -51,21 +51,21 @@ def test_assistant_should_use_user_integration_by_default(
 
     integration_utils.create_user_integration(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.azure_devops_credentials(),
+        CredentialsManager.azure_devops_credentials(),
         test_project_name,
     )
 
     if is_global:
         integration_utils.create_global_integration(
             CredentialTypes.AZURE_DEVOPS,
-            CredentialsUtil.invalid_ado_credentials(),
+            CredentialsManager.invalid_ado_credentials(),
             test_project_name,
         )
 
     if project_integration:
         integration_utils.create_project_integration(
             CredentialTypes.AZURE_DEVOPS,
-            CredentialsUtil.invalid_ado_credentials(),
+            CredentialsManager.invalid_ado_credentials(),
             test_project_name,
         )
 
@@ -105,12 +105,12 @@ def test_assistant_with_global_and_project_integration(
     # Global integration is created in project which is different from the project of the assistant.
     integration_utils.create_global_integration(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.azure_devops_credentials(),
+        CredentialsManager.azure_devops_credentials(),
         PROJECT,
     )
     integration_utils.create_project_integration(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.invalid_ado_credentials(),
+        CredentialsManager.invalid_ado_credentials(),
         test_project_name,
     )
 
@@ -147,7 +147,7 @@ def test_assistant_with_project_integration_only(
     # Create project integration only
     integration_utils.create_project_integration(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.azure_devops_credentials(),
+        CredentialsManager.azure_devops_credentials(),
         test_project_name,
     )
 
@@ -182,14 +182,14 @@ def test_assistant_with_global_valid_and_user_invalid_integration(
     # Create global valid integration.
     integration_utils.create_global_integration(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.azure_devops_credentials(),
+        CredentialsManager.azure_devops_credentials(),
         PROJECT,
     )
 
     # Create user invalid integration.
     integration_utils.create_user_integration(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.invalid_ado_credentials(),
+        CredentialsManager.invalid_ado_credentials(),
         PROJECT,
     )
 
@@ -224,14 +224,14 @@ def test_assistant_with_project_valid_and_user_invalid_integration(
     # Create project valid integration.
     integration_utils.create_project_integration(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.azure_devops_credentials(),
+        CredentialsManager.azure_devops_credentials(),
         test_project_name,
     )
 
     # Create user invalid integration.
     integration_utils.create_user_integration(
         CredentialTypes.AZURE_DEVOPS,
-        CredentialsUtil.invalid_ado_credentials(),
+        CredentialsManager.invalid_ado_credentials(),
         PROJECT,
     )
 

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/default_integrations/test_default_integrations_for_tool_with_datasource.py

@@ -10,7 +10,7 @@ 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 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.constants import test_project_name
 
 
@@ -55,7 +55,7 @@ def test_assistant_should_use_user_integration_by_default(
         )
 
     _git_integration = integration_utils.create_user_integration(
-        CredentialTypes.GIT, CredentialsUtil.gitlab_credentials(), test_project_name
+        CredentialTypes.GIT, CredentialsManager.gitlab_credentials(), test_project_name
     )
 
     # create a new datasource
@@ -68,14 +68,14 @@ def test_assistant_should_use_user_integration_by_default(
     if is_global:
         integration_utils.create_global_integration(
             CredentialTypes.GIT,
-            CredentialsUtil.invalid_git_credentials(),
+            CredentialsManager.invalid_git_credentials(),
             test_project_name,
         )
 
     if project_integration:
         integration_utils.create_project_integration(
             CredentialTypes.GIT,
-            CredentialsUtil.invalid_git_credentials(),
+            CredentialsManager.invalid_git_credentials(),
             test_project_name,
         )
 
@@ -112,7 +112,7 @@ def test_assistant_with_global_and_project_integration(
     """
     # Create datasource
     _git_integration = integration_utils.create_global_integration(
-        CredentialTypes.GIT, CredentialsUtil.gitlab_credentials(), test_project_name
+        CredentialTypes.GIT, CredentialsManager.gitlab_credentials(), test_project_name
     )
     git_datasource = datasource_utils.create_gitlab_datasource(
         setting_id=_git_integration.id,
@@ -129,12 +129,12 @@ def test_assistant_with_global_and_project_integration(
     # Create only global (valid) and project (invalid) integrations.
     # Global integration is created in project which is different from the project of the assistant.
     _git_integration = integration_utils.create_global_integration(
-        CredentialTypes.GIT, CredentialsUtil.gitlab_credentials(), PROJECT
+        CredentialTypes.GIT, CredentialsManager.gitlab_credentials(), PROJECT
     )
 
     integration_utils.create_project_integration(
         CredentialTypes.GIT,
-        CredentialsUtil.invalid_git_credentials(),
+        CredentialsManager.invalid_git_credentials(),
         test_project_name,
     )
 
@@ -174,7 +174,7 @@ def test_assistant_with_project_integration_only(
 
     # Create project integration only
     _git_integration = integration_utils.create_project_integration(
-        CredentialTypes.GIT, CredentialsUtil.gitlab_credentials(), test_project_name
+        CredentialTypes.GIT, CredentialsManager.gitlab_credentials(), test_project_name
     )
 
     # create a new datasource
@@ -216,7 +216,7 @@ def test_assistant_with_global_valid_and_user_invalid_integration(
     """
     # Create datasource
     _git_integration = integration_utils.create_global_integration(
-        CredentialTypes.GIT, CredentialsUtil.gitlab_credentials(), test_project_name
+        CredentialTypes.GIT, CredentialsManager.gitlab_credentials(), test_project_name
     )
     git_datasource = datasource_utils.create_gitlab_datasource(
         setting_id=_git_integration.id,
@@ -232,13 +232,13 @@ def test_assistant_with_global_valid_and_user_invalid_integration(
 
     # Create global valid integration
     _git_integration = integration_utils.create_global_integration(
-        CredentialTypes.GIT, CredentialsUtil.gitlab_credentials(), PROJECT
+        CredentialTypes.GIT, CredentialsManager.gitlab_credentials(), PROJECT
     )
 
     # Create user invalid integration
     integration_utils.create_user_integration(
         CredentialTypes.GIT,
-        CredentialsUtil.invalid_git_credentials(),
+        CredentialsManager.invalid_git_credentials(),
         PROJECT,
     )
 
@@ -274,7 +274,7 @@ def test_assistant_with_project_valid_and_user_invalid_integration(
     """
     # Create datasource
     _git_integration = integration_utils.create_global_integration(
-        CredentialTypes.GIT, CredentialsUtil.gitlab_credentials(), test_project_name
+        CredentialTypes.GIT, CredentialsManager.gitlab_credentials(), test_project_name
     )
     git_datasource = datasource_utils.create_gitlab_datasource(
         setting_id=_git_integration.id,
@@ -290,13 +290,13 @@ def test_assistant_with_project_valid_and_user_invalid_integration(
 
     # Create global valid integration
     _git_integration = integration_utils.create_project_integration(
-        CredentialTypes.GIT, CredentialsUtil.gitlab_credentials(), test_project_name
+        CredentialTypes.GIT, CredentialsManager.gitlab_credentials(), test_project_name
     )
 
     # Create user invalid integration
     integration_utils.create_user_integration(
         CredentialTypes.GIT,
-        CredentialsUtil.invalid_git_credentials(),
+        CredentialsManager.invalid_git_credentials(),
         PROJECT,
     )
 

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/tools/ado/test_assistant_for_ado_test_plan_tools.py

@@ -10,7 +10,7 @@ from codemie_test_harness.tests.test_data.ado_test_plan_tools_test_data import (
     ADO_TEST_PLAN_DELETE_SUITE,
     ADO_TEST_PLAN_DELETE_PLAN,
 )
-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.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
@@ -34,7 +34,7 @@ def test_assistant_with_ado_test_plan_get_tools(
     prompt,
     expected_response,
 ):
-    credential_values = CredentialsUtil.azure_devops_credentials()
+    credential_values = CredentialsManager.azure_devops_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.AZURE_DEVOPS, credential_values
     )
@@ -58,7 +58,7 @@ def test_assistant_with_ado_test_plan_tools(
     similarity_check,
 ):
     test_plan_title = f"Autotest Test Plan {get_random_name()}"
-    credential_values = CredentialsUtil.azure_devops_credentials()
+    credential_values = CredentialsManager.azure_devops_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.AZURE_DEVOPS, credential_values
     )

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/tools/ado/test_assistant_for_ado_wiki_tools.py

@@ -9,7 +9,7 @@ from codemie_test_harness.tests.test_data.ado_wiki_tools_test_data import (
     ADO_WIKI_MODIFY_PAGE,
     ADO_WIKI_DELETE_PAGE,
 )
-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.base_utils import get_random_name
 
 
@@ -31,7 +31,7 @@ def test_assistant_with_ado_wiki_get_tools(
     prompt,
     expected_response,
 ):
-    credential_values = CredentialsUtil.azure_devops_credentials()
+    credential_values = CredentialsManager.azure_devops_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.AZURE_DEVOPS, credential_values
     )
@@ -55,7 +55,7 @@ def test_assistant_with_ado_wiki_modify_tools(
     similarity_check,
 ):
     page_title = f"Autotest-Page-{get_random_name()}"
-    credential_values = CredentialsUtil.azure_devops_credentials()
+    credential_values = CredentialsManager.azure_devops_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.AZURE_DEVOPS, credential_values
     )

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/tools/ado/test_assistant_for_ado_work_item_tools.py

@@ -8,7 +8,7 @@ from codemie_test_harness.tests.test_data.ado_work_item_tools_test_data import (
     ADO_WORK_ITEM_UPDATE,
     ADO_WORK_ITEM_LINK,
 )
-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.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
@@ -32,7 +32,7 @@ def test_assistant_with_ado_work_item_get_tools(
     prompt,
     expected_response,
 ):
-    credential_values = CredentialsUtil.azure_devops_credentials()
+    credential_values = CredentialsManager.azure_devops_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.AZURE_DEVOPS, credential_values
     )
@@ -55,7 +55,7 @@ def test_assistant_with_ado_work_item_modify_tools(
     integration_utils,
     similarity_check,
 ):
-    credential_values = CredentialsUtil.azure_devops_credentials()
+    credential_values = CredentialsManager.azure_devops_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.AZURE_DEVOPS, credential_values
     )

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/tools/datamanagement/test_assistant_with_data_management_tools.py

@@ -17,7 +17,7 @@ from codemie_test_harness.tests.test_data.data_management_tools_test_data import
     SQL_TOOL_QUERY_TABLE_TASK,
     RESPONSE_FOR_SQL,
 )
-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
 
 pytestmark = pytest.mark.skipif(
@@ -33,7 +33,7 @@ pytestmark = pytest.mark.skipif(
 def test_create_assistant_with_elastic_tool(
     integration_utils, assistant, assistant_utils, similarity_check
 ):
-    credential_values = CredentialsUtil.elastic_credentials()
+    credential_values = CredentialsManager.elasticsearch_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.ELASTIC, credential_values
     )
@@ -57,13 +57,10 @@ def test_create_assistant_with_elastic_tool(
     sql_tools_test_data,
 )
 @pytest.mark.testcase("EPMCDME-6132")
-@pytest.mark.skipif(
-    EnvironmentResolver.is_localhost(), reason="Skipping this test on local environment"
-)
 def test_create_assistant_with_sql_tool(
     integration_utils, assistant_utils, assistant, similarity_check, db_dialect
 ):
-    credential_values = CredentialsUtil.sql_credentials(db_dialect)
+    credential_values = CredentialsManager.sql_credentials(db_dialect)
     settings = integration_utils.create_integration(
         CredentialTypes.SQL, credential_values
     )

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/tools/mcp/test_cli_mcp_server.py

@@ -6,7 +6,7 @@ from hamcrest import assert_that, contains_string
 
 from codemie_sdk.models.integration import CredentialTypes
 from codemie_test_harness.tests import autotest_entity_prefix
-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
 from codemie_test_harness.tests.utils.base_utils import get_random_name
 from codemie_test_harness.tests.test_data.mcp_server_test_data import (
@@ -45,7 +45,7 @@ def test_cli_mcp_server(
 def test_env_var_in_mcp_server(
     assistant_utils, assistant, similarity_check, integration_utils
 ):
-    credential_values = CredentialsUtil.mcp_credentials()
+    credential_values = CredentialsManager.mcp_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.MCP, credential_values
     )

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/tools/notification/test_assistant_notification_tools.py

@@ -3,7 +3,7 @@ from hamcrest import assert_that, equal_to
 
 from codemie_sdk.models.integration import CredentialTypes
 from codemie_test_harness.tests.enums.tools import Toolkit, NotificationTool
-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.enums.environment import Environment
 from codemie_test_harness.tests.utils.env_resolver import get_environment
 from codemie_test_harness.tests.test_data.notification_tools_test_data import (
@@ -31,7 +31,7 @@ def test_assistant_with_email_tool(
     integration_utils,
     gmail_message_operator,
 ):
-    credential_values = CredentialsUtil.gmail_credentials()
+    credential_values = CredentialsManager.gmail_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.EMAIL, credential_values
     )
@@ -69,7 +69,7 @@ def test_assistant_with_email_tool(
 def test_assistant_with_telegram_tool(
     assistant, assistant_utils, similarity_check, integration_utils
 ):
-    credential_values = CredentialsUtil.telegram_credentials()
+    credential_values = CredentialsManager.telegram_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.TELEGRAM, credential_values
     )

{codemie_test_harness-0.1.162 → codemie_test_harness-0.1.164}/codemie_test_harness/tests/assistant/tools/project_management/test_assistant_pm_tools.py

@@ -13,7 +13,7 @@ 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.aws_parameters_store import CredentialsUtil
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 from codemie_test_harness.tests.utils.base_utils import credentials_to_dict
 from codemie_test_harness.tests.utils.constants import (
     project_management_integrations,
@@ -60,7 +60,7 @@ def test_assistant_with_jira_tool_and_integration_id_in_chat(
     assistant, assistant_utils, integration_utils, similarity_check, jira_integration
 ):
     invalid_settings = integration_utils.create_integration(
-        CredentialTypes.JIRA, CredentialsUtil.invalid_jira_credentials()
+        CredentialTypes.JIRA, CredentialsManager.invalid_jira_credentials()
     )
     tool_config = ToolConfig(
         name=ProjectManagementTool.JIRA, integration_id=jira_integration.id
@@ -87,7 +87,7 @@ def test_assistant_with_jira_tool_and_credentials_in_chat(
 ):
     tool_config = ToolConfig(
         name=ProjectManagementTool.JIRA,
-        tool_creds=credentials_to_dict(CredentialsUtil.jira_credentials()),
+        tool_creds=credentials_to_dict(CredentialsManager.jira_credentials()),
     )
 
     assistant = assistant(
@@ -113,7 +113,7 @@ def test_assistant_with_confluence_tool_and_integration_id_in_chat(
     confluence_integration,
 ):
     invalid_settings = integration_utils.create_integration(
-        CredentialTypes.JIRA, CredentialsUtil.invalid_confluence_credentials()
+        CredentialTypes.JIRA, CredentialsManager.invalid_confluence_credentials()
     )
     tool_config = ToolConfig(
         name=ProjectManagementTool.CONFLUENCE, integration_id=confluence_integration.id
@@ -140,7 +140,7 @@ def test_assistant_with_confluence_tool_and_credentials_in_chat(
 ):
     tool_config = ToolConfig(
         name=ProjectManagementTool.CONFLUENCE,
-        tool_creds=credentials_to_dict(CredentialsUtil.confluence_credentials()),
+        tool_creds=credentials_to_dict(CredentialsManager.confluence_credentials()),
     )
 
     assistant = assistant(Toolkit.PROJECT_MANAGEMENT, ProjectManagementTool.CONFLUENCE)