codemie-test-harness 0.1.184__py3-none-any.whl → 0.1.198__py3-none-any.whl

codemie_test_harness/tests/workflow/assistant_tools/plugin/test_workflow_with_assistant_with_plugin_and_mcp_servers.py

@@ -2,11 +2,12 @@ import os
 
 import pytest
 
-from codemie_test_harness.tests.enums.tools import Toolkit
+from codemie_test_harness.tests.enums.tools import Toolkit, CliMcpServer, PluginTool
 from codemie_test_harness.tests.test_data.mcp_server_test_data import (
     cli_mcp_server_with_plugin_test_data,
     filesystem_mcp_server_with_plugin_test_data,
 )
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 from codemie_test_harness.tests.utils.constants import TESTS_PATH
 
 
@@ -36,6 +37,11 @@ def test_workflow_with_assistant_with_plugin_and_cli_mcp_server(
         workflow_with_assistant.id, assistant.name, user_input=prompt
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(CliMcpServer.RUN_COMMAND, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -69,6 +75,13 @@ def test_workflow_with_assistant_with_plugin_and_filesystem_mcp_server(
             workflow_with_assistant.id, assistant.name, user_input=prompt
         )
 
+        triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+            workflow_with_assistant
+        )
+        # Match the pattern from assistant test - conditionally assert tools
+        if tool_name != PluginTool.READ_FILE:
+            assert_tool_triggered(tool_name, triggered_tools)
+
         similarity_check.check_similarity(response, expected_response)
     finally:
         file_to_remove = f"{str(TESTS_PATH / 'sdk_tests')}.properties"

codemie_test_harness/tests/workflow/assistant_tools/project_management/test_workflow_with_assistant_pm_tools.py

@@ -4,6 +4,7 @@ from codemie_test_harness.tests.enums.tools import Toolkit
 from codemie_test_harness.tests.test_data.project_management_test_data import (
     pm_tools_test_data,
 )
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 from codemie_test_harness.tests.utils.constants import (
     project_management_integrations,
 )
@@ -45,4 +46,10 @@ def test_workflow_with_assistant_with_project_management_tools(
     response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name
     )
+
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)

codemie_test_harness/tests/workflow/assistant_tools/report_portal/test_workflow_with_assistant_with_report_portal_tools.py

@@ -3,6 +3,7 @@ import pytest
 from codemie_test_harness.tests.test_data.report_portal_tools_test_data import (
     rp_test_data,
 )
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 
 
 @pytest.mark.workflow
@@ -35,4 +36,10 @@ def test_workflow_with_assistant_with_report_portal_tools(
     response = workflow_utils.execute_workflow(
         workflow_with_assistant.id, assistant.name
     )
+
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)

codemie_test_harness/tests/workflow/assistant_tools/research/test_workflow_with_assistant_research_tools.py

@@ -6,7 +6,10 @@ from codemie_test_harness.tests.test_data.research_tools_test_data import (
     search_tools_test_data,
     interactions_tools_test_data,
 )
-from codemie_test_harness.tests.utils.base_utils import percent_of_relevant_titles
+from codemie_test_harness.tests.utils.base_utils import (
+    percent_of_relevant_titles,
+    assert_tool_triggered,
+)
 
 
 @pytest.mark.workflow
@@ -32,6 +35,11 @@ def test_workflow_with_search_tools(
         workflow_with_assistant.id, assistant.name, user_input=prompt
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     percent = percent_of_relevant_titles(response)
 
     assert_that(
@@ -65,4 +73,9 @@ def test_workflow_with_interaction_tools(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)

codemie_test_harness/tests/workflow/assistant_tools/servicenow/test_workflow_with_servicenow_tools.py

@@ -5,6 +5,7 @@ from codemie_test_harness.tests.test_data.servicenow_tools_test_data import (
     PROMPT,
     EXPECTED_RESPONSE,
 )
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 
 
 @pytest.mark.workflow
@@ -30,4 +31,9 @@ def test_workflow_with_assistant_with_servicenow_tools(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(ServiceNowTool.SERVICE_NOW, triggered_tools)
+
     similarity_check.check_similarity(response, EXPECTED_RESPONSE, 80)

codemie_test_harness/tests/workflow/assistant_tools/vcs/workflow_with_assistant_vcs_tools.py

@@ -6,6 +6,7 @@ from codemie_test_harness.tests.enums.tools import (
 from codemie_test_harness.tests.test_data.vcs_tools_test_data import (
     vcs_tools_test_data,
 )
+from codemie_test_harness.tests.utils.base_utils import assert_tool_triggered
 from codemie_test_harness.tests.utils.constants import vcs_integrations
 
 
@@ -36,4 +37,9 @@ def test_create_workflow_with_assistant_vcs_tool(
         workflow_with_assistant.id, assistant.name
     )
 
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        workflow_with_assistant
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)

codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_cloud_tools.py

@@ -26,7 +26,7 @@ from codemie_test_harness.tests.utils.base_utils import get_random_name
     ],
 )
 def test_workflow_with_cloud_tools_direct(
-    cloud_integration,
+    integration,
     workflow_utils,
     similarity_check,
     workflow_with_tool,
@@ -39,10 +39,10 @@ def test_workflow_with_cloud_tools_direct(
 ):
     tool_and_state_name = get_random_name()
 
-    integration = cloud_integration(credential_type, credentials)
+    cloud_integration = integration(credential_type, credentials)
 
     _workflow = workflow_with_tool(
-        tool_and_state_name, tool_name, integration=integration
+        tool_and_state_name, tool_name, integration=cloud_integration
     )
     response = workflow_utils.execute_workflow(
         _workflow.id, tool_and_state_name, user_input=json.dumps(prompt)
@@ -66,7 +66,7 @@ def test_workflow_with_cloud_tools_direct(
     ],
 )
 def test_workflow_with_cloud_tools_with_hardcoded_args(
-    cloud_integration,
+    integration,
     workflow_utils,
     similarity_check,
     workflow_with_tool,
@@ -79,10 +79,10 @@ def test_workflow_with_cloud_tools_with_hardcoded_args(
 ):
     tool_and_state_name = get_random_name()
 
-    integration = cloud_integration(credential_type, credentials)
+    cloud_integration = integration(credential_type, credentials)
 
     _workflow = workflow_with_tool(
-        tool_and_state_name, tool_name, integration=integration, tool_args=prompt
+        tool_and_state_name, tool_name, integration=cloud_integration, tool_args=prompt
     )
     response = workflow_utils.execute_workflow(_workflow.id, tool_and_state_name)
 
@@ -104,7 +104,7 @@ def test_workflow_with_cloud_tools_with_hardcoded_args(
     ],
 )
 def test_workflow_with_cloud_tools_with_overriding_args(
-    cloud_integration,
+    integration,
     workflow_utils,
     similarity_check,
     workflow_with_tool,
@@ -117,13 +117,16 @@ def test_workflow_with_cloud_tools_with_overriding_args(
 ):
     tool_and_state_name = get_random_name()
 
-    integration = cloud_integration(credential_type, credentials)
+    cloud_integration = integration(credential_type, credentials)
 
     args_copy = copy.deepcopy(prompt)
     args_copy = {key: random.randint(1, 10) for key in args_copy}
 
     _workflow = workflow_with_tool(
-        tool_and_state_name, tool_name, integration=integration, tool_args=args_copy
+        tool_and_state_name,
+        tool_name,
+        integration=cloud_integration,
+        tool_args=args_copy,
     )
     response = workflow_utils.execute_workflow(
         _workflow.id, tool_and_state_name, user_input=json.dumps(prompt)

codemie_test_harness/tests/workflow/virtual_assistant_tools/access_management/test_workflow_with_keycloak_tool.py

@@ -5,7 +5,10 @@ from codemie_test_harness.tests.test_data.keycloak_tool_test_data import (
     KEYCLOAK_TOOL_PROMPT,
     KEYCLOAK_TOOL_RESPONSE,
 )
-from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 
 
 @pytest.mark.workflow
@@ -29,4 +32,8 @@ def test_workflow_with_virtual_assistant_with_keycloak_tool(
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(AccessManagementTool.KEYCLOAK, triggered_tools)
     similarity_check.check_similarity(response, KEYCLOAK_TOOL_RESPONSE)

codemie_test_harness/tests/workflow/virtual_assistant_tools/ado/test_workflow_with_ado_test_plan_tools.py

@@ -9,7 +9,10 @@ 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.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 from codemie_test_harness.tests.utils.constants import ID_PATTERN
 from codemie_test_harness.tests.utils.json_utils import extract_id_from_ado_response
 
@@ -44,6 +47,10 @@ def test_workflow_with_ado_test_plan_get_tools(
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name, prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -74,6 +81,10 @@ def test_workflow_with_ado_test_plan_tools(
     create_response = workflow_utils.execute_workflow(
         create_test_suite_workflow.id, assistant_and_state_name
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        create_test_suite_workflow
+    )
+    assert_tool_triggered(AzureDevOpsTestPlanTool.CREATE_TEST_PLAN, triggered_tools)
 
     # Extract the test plan ID from the response
     test_plan_id = extract_id_from_ado_response(create_response, ID_PATTERN)
@@ -98,6 +109,10 @@ def test_workflow_with_ado_test_plan_tools(
     create_suite_response = workflow_utils.execute_workflow(
         create_test_suite_workflow.id, assistant_and_state_name, create_suite_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        create_test_suite_workflow
+    )
+    assert_tool_triggered(AzureDevOpsTestPlanTool.CREATE_TEST_SUITE, triggered_tools)
 
     # Extract the test suite ID from the response
     test_suite_id = extract_id_from_ado_response(create_suite_response, ID_PATTERN)
@@ -119,6 +134,10 @@ def test_workflow_with_ado_test_plan_tools(
     add_case_response = workflow_utils.execute_workflow(
         add_test_case_workflow.id, assistant_and_state_name, add_case_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        add_test_case_workflow
+    )
+    assert_tool_triggered(AzureDevOpsTestPlanTool.ADD_TEST_CASE, triggered_tools)
     add_case_expected = ADO_TEST_PLAN_ADD_TEST_CASE["expected_llm_answer"].format(
         test_suite_id, test_plan_id
     )
@@ -137,6 +156,10 @@ def test_workflow_with_ado_test_plan_tools(
     delete_suite_response = workflow_utils.execute_workflow(
         delete_test_suite_workflow.id, assistant_and_state_name, delete_suite_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        delete_test_suite_workflow
+    )
+    assert_tool_triggered(AzureDevOpsTestPlanTool.DELETE_TEST_SUITE, triggered_tools)
     delete_suite_expected = ADO_TEST_PLAN_DELETE_SUITE["expected_llm_answer"].format(
         test_suite_id, test_plan_id
     )
@@ -155,6 +178,10 @@ def test_workflow_with_ado_test_plan_tools(
     delete_plan_response = workflow_utils.execute_workflow(
         delete_test_plan_workflow.id, assistant_and_state_name, delete_plan_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        delete_test_plan_workflow
+    )
+    assert_tool_triggered(AzureDevOpsTestPlanTool.DELETE_TEST_PLAN, triggered_tools)
     delete_plan_expected = ADO_TEST_PLAN_DELETE_PLAN["expected_llm_answer"].format(
         test_plan_id
     )

codemie_test_harness/tests/workflow/virtual_assistant_tools/ado/test_workflow_with_ado_wiki_tools.py

@@ -8,7 +8,10 @@ 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.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 
 
 @pytest.mark.workflow
@@ -41,6 +44,10 @@ def test_workflow_with_ado_wiki_get_tools(
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name, prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -68,6 +75,10 @@ def test_workflow_with_ado_wiki_modify_tools(
     create_response = workflow_utils.execute_workflow(
         create_page_workflow.id, assistant_and_state_name, create_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        create_page_workflow
+    )
+    assert_tool_triggered(AzureDevOpsWikiTool.MODIFY_WIKI_PAGE, triggered_tools)
     similarity_check.check_similarity(create_response, create_expected)
 
     # 2. Rename the page
@@ -86,6 +97,10 @@ def test_workflow_with_ado_wiki_modify_tools(
     rename_response = workflow_utils.execute_workflow(
         rename_page_workflow.id, assistant_and_state_name, rename_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        rename_page_workflow
+    )
+    assert_tool_triggered(AzureDevOpsWikiTool.RENAME_WIKI_PAGE, triggered_tools)
     similarity_check.check_similarity(rename_response, rename_expected)
 
     # 3. Modify the page
@@ -104,6 +119,10 @@ def test_workflow_with_ado_wiki_modify_tools(
     modify_response = workflow_utils.execute_workflow(
         modify_page_workflow.id, assistant_and_state_name, modify_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        modify_page_workflow
+    )
+    assert_tool_triggered(AzureDevOpsWikiTool.MODIFY_WIKI_PAGE, triggered_tools)
     similarity_check.check_similarity(modify_response, modify_expected)
 
     # 4. Delete the page
@@ -122,4 +141,8 @@ def test_workflow_with_ado_wiki_modify_tools(
     delete_response = workflow_utils.execute_workflow(
         delete_page_workflow.id, assistant_and_state_name, delete_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        delete_page_workflow
+    )
+    assert_tool_triggered(AzureDevOpsWikiTool.DELETE_WIKI_PAGE_BY_PATH, triggered_tools)
     similarity_check.check_similarity(delete_response, delete_expected)

codemie_test_harness/tests/workflow/virtual_assistant_tools/ado/test_workflow_with_ado_work_item_tools.py

@@ -7,7 +7,10 @@ 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.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 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
 
@@ -42,6 +45,10 @@ def test_workflow_with_ado_work_item_get_tools(
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name, prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -73,6 +80,10 @@ def test_workflow_with_ado_work_item_modify_tools(
     create_response = workflow_utils.execute_workflow(
         create_item_workflow.id, assistant_and_state_name, create_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        create_item_workflow
+    )
+    assert_tool_triggered(AzureDevOpsWorkItemTool.CREATE_WORK_ITEM, triggered_tools)
     work_item_id = extract_id_from_ado_response(create_response, WORK_ITEM_ID_PATTERN)
     similarity_check.check_similarity(create_response, create_expected)
 
@@ -94,6 +105,10 @@ def test_workflow_with_ado_work_item_modify_tools(
     update_response = workflow_utils.execute_workflow(
         update_item_workflow.id, assistant_and_state_name, update_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        update_item_workflow
+    )
+    assert_tool_triggered(AzureDevOpsWorkItemTool.UPDATE_WORK_ITEM, triggered_tools)
     similarity_check.check_similarity(update_response, update_expected)
 
     # 3. Link the item
@@ -111,4 +126,8 @@ def test_workflow_with_ado_work_item_modify_tools(
     link_response = workflow_utils.execute_workflow(
         link_item_workflow.id, assistant_and_state_name, link_prompt
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        link_item_workflow
+    )
+    assert_tool_triggered(AzureDevOpsWorkItemTool.LINK_WORK_ITEMS, triggered_tools)
     similarity_check.check_similarity(link_response, link_expected)

codemie_test_harness/tests/workflow/virtual_assistant_tools/cloud/test_workflow_with_cloud_tools.py

@@ -1,7 +1,10 @@
 import pytest
 
 from codemie_test_harness.tests.test_data.cloud_tools_test_data import cloud_test_data
-from codemie_test_harness.tests.utils.base_utils import get_random_name
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 
 
 @pytest.mark.workflow
@@ -17,7 +20,7 @@ from codemie_test_harness.tests.utils.base_utils import get_random_name
 )
 def test_workflow_with_cloud_tools(
     workflow_with_virtual_assistant,
-    cloud_integration,
+    integration,
     workflow_utils,
     similarity_check,
     toolkit,
@@ -29,15 +32,21 @@ def test_workflow_with_cloud_tools(
 ):
     assistant_and_state_name = get_random_name()
 
-    integration = cloud_integration(credential_type, credentials)
+    cloud_integration = integration(credential_type, credentials)
 
     test_workflow = workflow_with_virtual_assistant(
         assistant_and_state_name,
         tool_name,
-        integration=integration,
+        integration=cloud_integration,
         task=prompt,
     )
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name
     )
+
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)

codemie_test_harness/tests/workflow/virtual_assistant_tools/codebase/test_workflow_with_codebase_tools.py

@@ -7,7 +7,10 @@ from codemie_test_harness.tests.test_data.codebase_tools_test_data import (
     sonar_tools_test_data,
 )
 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.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
+)
 
 
 @pytest.fixture(scope="module")
@@ -60,6 +63,12 @@ def test_workflow_with_codebase_tools(
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name
     )
+
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(tool_name, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)
 
 
@@ -98,4 +107,10 @@ def test_workflow_with_sonar_tools(
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name
     )
+
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(CodeBaseTool.SONAR, triggered_tools)
+
     similarity_check.check_similarity(response, expected_response)

codemie_test_harness/tests/workflow/virtual_assistant_tools/data_management/test_workflow_with_data_management_tools.py

@@ -1,6 +1,7 @@
 import pytest
 from codemie_sdk.models.integration import CredentialTypes
 
+from codemie_test_harness.tests.enums.integrations import DataBaseDialect
 from codemie_test_harness.tests.enums.tools import DataManagementTool
 from codemie_test_harness.tests.test_data.data_management_tools_test_data import (
     ELASTIC_TOOL_TASK,
@@ -11,10 +12,15 @@ from codemie_test_harness.tests.test_data.data_management_tools_test_data import
     SQL_TOOL_INSERT_TABLE_TASK,
     SQL_TOOL_QUERY_TABLE_TASK,
     RESPONSE_FOR_SQL,
+    INFLUXDB_QUERY_MEASUREMENT_TASK,
+    RESPONSE_FOR_INFLUXDB,
+)
+from codemie_test_harness.tests.utils.base_utils import (
+    get_random_name,
+    assert_tool_triggered,
 )
 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
 
 pytestmark = pytest.mark.skipif(
     EnvironmentResolver.is_localhost(),
@@ -49,6 +55,10 @@ def test_workflow_with_elastic_tools(
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(DataManagementTool.ELASTIC, triggered_tools)
     similarity_check.check_similarity(response, RESPONSE_FOR_ELASTIC)
 
 
@@ -81,17 +91,79 @@ def test_workflow_with_sql_tools(
         integration=integration,
         task=f"DB dialect is {db_dialect.value}. Run SQL tool and execute SQL queries to perform user requests.",
     )
+
+    # Step 1: Create table
     workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name, SQL_TOOL_CREATE_TABLE_TASK
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(DataManagementTool.SQL, triggered_tools)
+
+    # Step 2: Insert data
     workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name, SQL_TOOL_INSERT_TABLE_TASK
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(DataManagementTool.SQL, triggered_tools)
+
+    # Step 3: Query data
     response = workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name, SQL_TOOL_QUERY_TABLE_TASK
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(DataManagementTool.SQL, triggered_tools)
+
+    # Step 4: Delete table
     workflow_utils.execute_workflow(
         test_workflow.id, assistant_and_state_name, SQL_TOOL_DELETE_TABLE_TASK
     )
+    triggered_tools = workflow_utils.extract_triggered_tools_from_execution(
+        test_workflow
+    )
+    assert_tool_triggered(DataManagementTool.SQL, triggered_tools)
 
     similarity_check.check_similarity(response, RESPONSE_FOR_SQL)
+
+
+@pytest.mark.workflow
+@pytest.mark.virtual_workflow
+@pytest.mark.influx
+@pytest.mark.api
+@pytest.mark.testcase("EPMCDME-6431")
+@pytest.mark.skipif(
+    not EnvironmentResolver.is_sandbox(),
+    reason="InfluxDB is only available in sandbox environments",
+)
+def test_workflow_with_influxdb_tools(
+    workflow_with_virtual_assistant,
+    workflow_utils,
+    integration_utils,
+    similarity_check,
+):
+    """Test workflow execution with InfluxDB tools."""
+
+    assistant_and_state_name = get_random_name()
+    credential_values = CredentialsManager.sql_credentials(DataBaseDialect.INFLUX)
+    integration = integration_utils.create_integration(
+        CredentialTypes.SQL, credential_values
+    )
+
+    test_workflow = workflow_with_virtual_assistant(
+        assistant_and_state_name,
+        DataManagementTool.SQL,
+        integration=integration,
+        task=f"DB dialect is {DataBaseDialect.INFLUX.value}. Run SQL tool and execute InfluxQL queries to perform user requests.",
+    )
+
+    # Query the measurement
+    response = workflow_utils.execute_workflow(
+        test_workflow.id, assistant_and_state_name, INFLUXDB_QUERY_MEASUREMENT_TASK
+    )
+
+    similarity_check.check_similarity(response, RESPONSE_FOR_INFLUXDB)