codemie-test-harness 0.1.134__py3-none-any.whl → 0.1.135__py3-none-any.whl

codemie_test_harness/tests/enums/integrations.py

@@ -4,5 +4,6 @@ from enum import Enum
 class DataBaseDialect(str, Enum):
     """Enum for DB Dialect names."""
 
+    MS_SQL = "mssql"
     MY_SQL = "mysql"
     POSTGRES = "postgres"

codemie_test_harness/tests/test_data/data_management_tools_test_data.py

@@ -1,3 +1,7 @@
+import os
+
+import pytest
+
 from codemie_test_harness.tests.enums.integrations import DataBaseDialect
 
 ELASTIC_TOOL_TASK = """
@@ -81,7 +85,15 @@ RESPONSE_FOR_SQL = """
     | Sarah      | Connor    | sarah.connor@email.com  | Security   |
 """
 
+# Define test data for SQL tools based on environment
 sql_tools_test_data = [
     DataBaseDialect.MY_SQL,
     DataBaseDialect.POSTGRES,
+    pytest.param(
+        DataBaseDialect.MS_SQL,
+        marks=pytest.mark.skipif(
+            os.getenv("ENV") not in ("aws", "azure", "gcp"),
+            reason="MS SQL is only available in staging environments",
+        ),
+    ),
 ]

codemie_test_harness/tests/test_data/direct_tools/data_management_tools_test_data.py

@@ -1,3 +1,7 @@
+import os
+
+import pytest
+
 from codemie_test_harness.tests.enums.tools import DataManagementTool, Toolkit
 from codemie_test_harness.tests.enums.integrations import DataBaseDialect
 
@@ -18,4 +22,27 @@ sql_tools_test_data = [
         },
         [{"table_name": "users"}, {"table_name": "products"}],
     ),
+    pytest.param(
+        Toolkit.DATA_MANAGEMENT,
+        DataManagementTool.SQL,
+        DataBaseDialect.MS_SQL,
+        {
+            "sql_query": """SELECT table_name
+                            FROM
+                            information_schema.tables
+                            WHERE
+                            table_type = 'BASE TABLE'
+                            AND
+                            table_catalog = 'autotests'
+                            AND
+                            table_schema = 'dbo';
+                            """
+        },
+        [{"table_name": "Users"}, {"table_name": "Products"}],
+        marks=pytest.mark.skipif(
+            os.getenv("ENV") not in ("aws", "azure", "gcp"),
+            reason="MS SQL is only available in staging environments",
+        ),
+        id="DataBaseDialect.MS_SQL",
+    ),
 ]

codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_data_management_tools.py

@@ -22,6 +22,7 @@ pytestmark = pytest.mark.skipif(
 @pytest.mark.parametrize(
     "toolkit,tool_name,db_dialect,prompt,expected_response",
     sql_tools_test_data,
+    ids=[f"{row[2]}" for row in sql_tools_test_data],
 )
 def test_workflow_with_sql_tools_direct(
     workflow_with_tool,
@@ -59,6 +60,7 @@ def test_workflow_with_sql_tools_direct(
 @pytest.mark.parametrize(
     "toolkit,tool_name,db_dialect,prompt,expected_response",
     sql_tools_test_data,
+    ids=[f"{row[2]}" for row in sql_tools_test_data],
 )
 def test_workflow_with_sql_tools_with_hardcoded_args(
     workflow_with_tool,
@@ -97,6 +99,7 @@ def test_workflow_with_sql_tools_with_hardcoded_args(
 @pytest.mark.parametrize(
     "toolkit,tool_name,db_dialect,prompt,expected_response",
     sql_tools_test_data,
+    ids=[f"{row[2]}" for row in sql_tools_test_data],
 )
 def test_workflow_with_sql_tools_with_overriding_args(
     workflow_with_tool,

{codemie_test_harness-0.1.134.dist-info → codemie_test_harness-0.1.135.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: codemie-test-harness
-Version: 0.1.134
+Version: 0.1.135
 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.134)
+Requires-Dist: codemie-sdk-python (==0.1.135)
 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.134.dist-info → codemie_test_harness-0.1.135.dist-info}/RECORD

@@ -59,7 +59,7 @@ codemie_test_harness/tests/conftest.py,sha256=2EFM-VIh5BvJAlVCK4GTggGu4UbHWt5_dg
 codemie_test_harness/tests/e2e/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/e2e/test_e2e.py,sha256=RUvSu_uNInU1OFfkNft3-Cmd0KZV7JdHqCxSK2ZWXHw,6257
 codemie_test_harness/tests/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-codemie_test_harness/tests/enums/integrations.py,sha256=WEfqnOLqa4m8CH8BjQ9bxty0-w4nsZGy0Ok6SuQoqow,143
+codemie_test_harness/tests/enums/integrations.py,sha256=hG_cEKN0N5gtGHyrh3wuCpNAWhPyNOJaKsAw16iNshg,164
 codemie_test_harness/tests/enums/model_types.py,sha256=k07srBYIF9uRwbLWY9pBNYzi0s3Jdt8ajES9p-UmrYo,1185
 codemie_test_harness/tests/enums/tools.py,sha256=pGsHRGmPcLUqNtR2bKdRgsxdHw7HylXOWd9CPKPGOb0,5072
 codemie_test_harness/tests/integrations/__init__.py,sha256=5vnZbxvYQ1Y91O8DJG3QfBHWcdYsMii59cMBzsvHBCg,237
@@ -94,14 +94,14 @@ codemie_test_harness/tests/test_data/ado_wiki_tools_test_data.py,sha256=xvgEja5v
 codemie_test_harness/tests/test_data/ado_work_item_tools_test_data.py,sha256=MHou6QGjufyX5t1oexnq2Y4UCjlE17eeyMuIz6Ml62s,5693
 codemie_test_harness/tests/test_data/cloud_tools_test_data.py,sha256=vq0a5vx9cyxVnen-Cx5KCg_szO4Lje0UfFbp85XtaxY,4701
 codemie_test_harness/tests/test_data/codebase_tools_test_data.py,sha256=dpNKPtEZsBsjhnR3-pFBSkJRlExgjJ6fztK1spj04yI,3131
-codemie_test_harness/tests/test_data/data_management_tools_test_data.py,sha256=JS8bbp7kA4flBBrymRfRqnS-sIcwvXbx24LYkeZEMVE,3801
+codemie_test_harness/tests/test_data/data_management_tools_test_data.py,sha256=olCcn8If6AETpmgNBE-ekmXAKlPGqt1MHGkxLacdU_0,4115
 codemie_test_harness/tests/test_data/direct_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/test_data/direct_tools/ado_test_plan_tools_test_data.py,sha256=bDhoeFRlmafSZdRgOTOz122Q5tja7TsqtOxSTypeGVA,13120
 codemie_test_harness/tests/test_data/direct_tools/ado_wiki_tools_test_data.py,sha256=qKNYW0kP32IQrRee8-H1mEWTUa5gb4UP0Me_4owtgeQ,1432
 codemie_test_harness/tests/test_data/direct_tools/ado_work_item_tools_test_data.py,sha256=L-MhuQyRxRbb1WDje1rONjIxc_QxkgzaM5ssMWglcp0,7417
 codemie_test_harness/tests/test_data/direct_tools/cloud_tools_test_data.py,sha256=aehfRdIo6uSh00jJUGhfXxdFIQa6_Rw7OUF6Lo1LXTM,35201
 codemie_test_harness/tests/test_data/direct_tools/codebase_tools_test_data.py,sha256=GYAs-CGyvZGxkuJKOi1Wod_hK1noSw8zmnlF7UE78eA,11087
-codemie_test_harness/tests/test_data/direct_tools/data_management_tools_test_data.py,sha256=AZjFvT20Ghz4mP7pSNq-sEuf-PLEPXedmeayCuOvecQ,714
+codemie_test_harness/tests/test_data/direct_tools/data_management_tools_test_data.py,sha256=Eeba0oxKUmBEin2Scbv3VnLOEQRLmK6QR0qk5cRYRNM,1583
 codemie_test_harness/tests/test_data/direct_tools/direct_tools_test_data.py,sha256=y9awl1IA6EXGXyz05QzcNdt5z7Rk9J5LzIbfi4CFE3s,3233
 codemie_test_harness/tests/test_data/direct_tools/file_management_tools_test_data.py,sha256=8Y8D_pmHVnzhSnYNeiJng3dQYj5k1GkBZ8-BLHlB5kE,1380
 codemie_test_harness/tests/test_data/direct_tools/notification_tools_test_data.py,sha256=Blr7qIBEMxAh2_WRvFQr1hYt6wnez03v1b2SpEWha-o,1533
@@ -275,7 +275,7 @@ codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_ado_
 codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_ado_work_item_tools.py,sha256=4h3z3tUTqBLZPjDQCbnW2CkoGX0gEjZlDkYbLHSilZA,2799
 codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_cloud_tools.py,sha256=uYK0ASA0BpVYh_ozvVRLkEkHABDlkuu0VxuwyXBkHIQ,3539
 codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_codebase_tools.py,sha256=Kul_o9d4KIPzWUhKNOv82A9S_8-Z0ALaTeO5AO2Ltzs,3821
-codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_data_management_tools.py,sha256=igtMKIsL-_NnaN9-Hez697vwrA_-a1QZeafh1IzN4Bg,3737
+codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_data_management_tools.py,sha256=eIIHRovi_tNvINjOVYOQRB9uWIzfebwnDF4fWdx47Og,3899
 codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_file_management_tools.py,sha256=W1Q8fwfUzslnMQX7ObM8o1r5kx_GyZTjVGh-1979VNQ,2673
 codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_notification_tools.py,sha256=esLfjDk2zGBpqX0D0VJYQKW4qwuvXy1afiFI8PyfMbc,2931
 codemie_test_harness/tests/workflow/direct_tools_calling/test_workflow_with_open_api_tools.py,sha256=9il7_0U3m8Ekvw5bCP0WQlJ-27RpKhViyqlrjqsS5_Y,2949
@@ -321,7 +321,7 @@ codemie_test_harness/tests/workflow/virtual_assistant_tools/servicenow/__init__.
 codemie_test_harness/tests/workflow/virtual_assistant_tools/servicenow/test_workflow_with_servicenow_tools.py,sha256=Cxe5Yuy9JE_IEsnQOThpVIZQqpfLuHfDkF6JwLngDc8,890
 codemie_test_harness/tests/workflow/virtual_assistant_tools/vcs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/workflow/virtual_assistant_tools/vcs/test_workflow_with_vcs_tools.py,sha256=tm0NvUf_UtzLPTYJWykYpsNoxoVs-Eh80guDmRwImwg,1106
-codemie_test_harness-0.1.134.dist-info/METADATA,sha256=_C34qXz4_gJ9nGFuh1iAbE4fl9w1pEnheiyvxxiVkx8,8998
-codemie_test_harness-0.1.134.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-codemie_test_harness-0.1.134.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
-codemie_test_harness-0.1.134.dist-info/RECORD,,
+codemie_test_harness-0.1.135.dist-info/METADATA,sha256=mEwdZM_OH1YLuq6wAZl-_86HU8QDdO5luFGYWA9U3Vg,8998
+codemie_test_harness-0.1.135.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+codemie_test_harness-0.1.135.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
+codemie_test_harness-0.1.135.dist-info/RECORD,,