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

codemie_test_harness/tests/utils/datasource_utils.py

@@ -1,8 +1,9 @@
-import os
 import time
 from time import sleep
 from typing import List, Union, Tuple
 
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
+
 from codemie_sdk.exceptions import ApiError
 from codemie_sdk.models.datasource import (
     DataSource,
@@ -83,7 +84,7 @@ class DataSourceUtils(BaseUtils):
     def create_gitlab_datasource(self, **kwargs):
         return self.create_code_datasource(
             name=kwargs.get("name", get_random_name()),
-            link=kwargs.get("link", os.getenv("GITLAB_PROJECT")),
+            link=kwargs.get("link", CredentialsManager.get_parameter("GITLAB_PROJECT")),
             branch=kwargs.get("branch", "main"),
             setting_id=kwargs.get("setting_id"),
             embeddings_model=kwargs.get("embeddings_model"),
@@ -93,7 +94,7 @@ class DataSourceUtils(BaseUtils):
     def create_github_datasource(self, **kwargs):
         return self.create_code_datasource(
             name=kwargs.get("name", get_random_name()),
-            link=kwargs.get("link", os.getenv("GITHUB_PROJECT")),
+            link=kwargs.get("link", CredentialsManager.get_parameter("GITHUB_PROJECT")),
             branch=kwargs.get("branch", "master"),
             setting_id=kwargs.get("setting_id"),
             embeddings_model=kwargs.get("embeddings_model"),
@@ -199,7 +200,7 @@ class DataSourceUtils(BaseUtils):
     def wait_for_indexing(
         self,
         datasource_id: str,
-        timeout: int = int(os.getenv("DEFAULT_TIMEOUT")),
+        timeout: int = int(CredentialsManager.get_parameter("DEFAULT_TIMEOUT", "120")),
         pool_interval: int = 3,
     ) -> DataSource:
         start_time = time.time()
@@ -221,7 +222,7 @@ class DataSourceUtils(BaseUtils):
             "name": name,
             "project_name": kwargs.get("project_name", PROJECT),
             "description": kwargs.get("description", get_random_name()),
-            "cql": kwargs.get("cql", os.getenv("CQL")),
+            "cql": kwargs.get("cql", CredentialsManager.confluence_cql()),
             "project_space_visible": kwargs.get("project_space_visible", False),
             "setting_id": kwargs.get("setting_id"),
         }
@@ -265,7 +266,7 @@ class DataSourceUtils(BaseUtils):
             "name": name,
             "project_name": kwargs.get("project_name", PROJECT),
             "description": kwargs.get("description", get_random_name()),
-            "jql": kwargs.get("jql", os.getenv("JQL")),
+            "jql": kwargs.get("jql", CredentialsManager.jira_jql()),
             "project_space_visible": kwargs.get("project_space_visible", False),
             "setting_id": kwargs.get("setting_id"),
         }

codemie_test_harness/tests/utils/gitbud_utils.py

@@ -1,9 +1,9 @@
-import os
 from time import sleep
 
 import gitlab
 from gitlab.exceptions import GitlabError
 import logging
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 
 logger = logging.getLogger(__name__)
 
@@ -15,9 +15,9 @@ class GitBudUtils:
 
     def __init__(
         self,
-        url: str = os.getenv("GITLAB_URL"),
-        token: str = os.getenv("GITLAB_TOKEN"),
-        project_id: str = os.getenv("GITLAB_PROJECT_ID"),
+        url: str = CredentialsManager.get_parameter("GITLAB_URL"),
+        token: str = CredentialsManager.get_parameter("GITLAB_TOKEN"),
+        project_id: str = CredentialsManager.get_parameter("GITLAB_PROJECT_ID"),
     ):
         """
         Initialize GitBudUtils with GitLab credentials

codemie_test_harness/tests/utils/notification_utils.py

@@ -3,7 +3,7 @@ from typing import Optional
 
 import requests
 
-from codemie_test_harness.tests.utils.aws_parameters_store import AwsParameterStore
+from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 
 logger = logging.getLogger(__name__)
 
@@ -20,21 +20,21 @@ class OAuthGmailTokenRetriever:
 
     @staticmethod
     def _fetch_access_token():
-        credentials = AwsParameterStore.get_cloud_provider_credentials("email")
+        credentials = CredentialsManager.oauth_credentials()
 
-        token_url = credentials["oauth"]["url"]
+        token_url = credentials[0].value
         if token_url is None:
             raise ValueError("Missing url")
 
-        client_id = credentials["oauth"]["client_id"]
+        client_id = credentials[1].value
         if client_id is None:
             raise ValueError("Missing client_id")
 
-        client_secret = credentials["oauth"]["client_secret"]
+        client_secret = credentials[2].value
         if client_secret is None:
             raise ValueError("Missing client_secret")
 
-        refresh_token = credentials["oauth"]["refresh_token"]
+        refresh_token = credentials[3].value
         if refresh_token is None:
             raise ValueError("Missing refresh_token")
 

codemie_test_harness/tests/workflow/assistant_tools/data_management/test_workflow_with_assistant_with_data_management_tools.py

@@ -12,7 +12,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 @@ def test_workflow_with_assistant_with_elastic_tools(
     similarity_check,
 ):
     """Test workflow execution with Elastic tools."""
-    credential_values = CredentialsUtil.elastic_credentials()
+    credential_values = CredentialsManager.elasticsearch_credentials()
     settings = integration_utils.create_integration(
         CredentialTypes.ELASTIC, credential_values
     )
@@ -65,7 +65,7 @@ def test_workflow_with_assistant_with_sql_tools(
     db_dialect,
 ):
     """Test workflow execution with SQL data management tools (various dialects)."""
-    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/tests/workflow/assistant_tools/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
 
 
@@ -52,20 +52,20 @@ def test_assistant_in_workflow_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,
         )
 
@@ -110,12 +110,12 @@ def test_assistant_in_workflow_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,
     )
 
@@ -157,7 +157,7 @@ def test_assistant_in_workflow_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(
@@ -199,13 +199,13 @@ def test_assistant_in_workflow_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,
     )
 
@@ -248,13 +248,13 @@ def test_assistant_in_workflow_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/tests/workflow/assistant_tools/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]
@@ -55,21 +55,21 @@ def test_assistant_in_workflow_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,
         )
 
@@ -115,12 +115,12 @@ def test_assistant_in_workflow_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,
     )
 
@@ -163,7 +163,7 @@ def test_assistant_in_workflow_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,
     )
 
@@ -207,14 +207,14 @@ def test_assistant_in_workflow_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,
     )
 
@@ -258,14 +258,14 @@ def test_assistant_in_workflow_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/tests/workflow/assistant_tools/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
 
 
@@ -60,7 +60,7 @@ def test_assistant_in_workflow_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
@@ -73,14 +73,14 @@ def test_assistant_in_workflow_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,
         )
 
@@ -124,7 +124,7 @@ def test_assistant_in_workflow_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,
@@ -141,12 +141,12 @@ def test_assistant_in_workflow_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,
     )
 
@@ -193,7 +193,7 @@ def test_assistant_in_workflow_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
@@ -242,7 +242,7 @@ def test_assistant_in_workflow_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,
@@ -258,13 +258,13 @@ def test_assistant_in_workflow_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,
     )
 
@@ -307,7 +307,7 @@ def test_assistant_in_workflow_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,
@@ -323,13 +323,13 @@ def test_assistant_in_workflow_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/tests/workflow/direct_tools_calling/default_integrations/test_default_integrations_for_tool.py

@@ -12,7 +12,7 @@ from codemie_test_harness.tests.enums.tools import ProjectManagementTool
 from codemie_test_harness.tests.test_data.direct_tools.project_management_tools_test_data import (
     project_management_tools_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 get_random_name
 from codemie_test_harness.tests.utils.constants import test_project_name
 
@@ -55,20 +55,20 @@ def test_tool_in_workflow_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,
         )
 
@@ -113,12 +113,12 @@ def test_tool_in_workflow_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,
     )
 
@@ -160,7 +160,7 @@ def test_tool_in_workflow_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
     )
 
     assistant_and_state_name = get_random_name()
@@ -202,13 +202,13 @@ def test_tool_in_workflow_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,
     )
 
@@ -251,13 +251,13 @@ def test_tool_in_workflow_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/tests/workflow/direct_tools_calling/default_integrations/test_default_integrations_for_tool_kit.py

@@ -14,7 +14,7 @@ from codemie_test_harness.tests.enums.tools import AzureDevOpsWikiTool
 from codemie_test_harness.tests.test_data.direct_tools.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.base_utils import get_random_name
 from codemie_test_harness.tests.utils.constants import test_project_name
 
@@ -57,21 +57,21 @@ def test_tool_in_workflow_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,
         )
 
@@ -116,12 +116,12 @@ def test_tool_in_workflow_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,
     )
 
@@ -163,7 +163,7 @@ def test_tool_in_workflow_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,
     )
 
@@ -206,14 +206,14 @@ def test_tool_in_workflow_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,
     )
 
@@ -256,14 +256,14 @@ def test_tool_in_workflow_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/tests/workflow/direct_tools_calling/test_workflow_with_codebase_tools.py

@@ -9,7 +9,7 @@ from codemie_test_harness.tests.enums.tools import CodeBaseTool
 from codemie_test_harness.tests.test_data.direct_tools.codebase_tools_test_data import (
     sonar_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 get_random_name
 
 
@@ -33,9 +33,9 @@ def test_workflow_with_sonar_tools_direct(
 ):
     assistant_and_state_name = get_random_name()
     credential_values = (
-        CredentialsUtil.sonar_credentials()
+        CredentialsManager.sonar_credentials()
         if tool_name == CodeBaseTool.SONAR
-        else CredentialsUtil.sonar_cloud_credentials()
+        else CredentialsManager.sonar_cloud_credentials()
     )
 
     integration = integration_utils.create_integration(
@@ -73,9 +73,9 @@ def test_workflow_with_sonar_tools_with_hardcoded_args(
 ):
     assistant_and_state_name = get_random_name()
     credential_values = (
-        CredentialsUtil.sonar_credentials()
+        CredentialsManager.sonar_credentials()
         if tool_name == CodeBaseTool.SONAR
-        else CredentialsUtil.sonar_cloud_credentials()
+        else CredentialsManager.sonar_cloud_credentials()
     )
 
     integration = integration_utils.create_integration(
@@ -114,9 +114,9 @@ def test_workflow_with_sonar_tools_with_overriding_args(
 ):
     assistant_and_state_name = get_random_name()
     credential_values = (
-        CredentialsUtil.sonar_credentials()
+        CredentialsManager.sonar_credentials()
         if tool_name == CodeBaseTool.SONAR
-        else CredentialsUtil.sonar_cloud_credentials()
+        else CredentialsManager.sonar_cloud_credentials()
     )
 
     integration = integration_utils.create_integration(