codemie-test-harness 0.1.148__py3-none-any.whl → 0.1.149__py3-none-any.whl

codemie_test_harness/tests/assistant/tools/datamanagement/test_assistant_with_data_management_tools.py

@@ -4,6 +4,7 @@ import uuid
 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 (
     Toolkit,
     DataManagementTool,
@@ -46,7 +47,7 @@ def test_create_assistant_with_elastic_tool(
 @pytest.mark.parametrize(
     "db_dialect",
     sql_tools_test_data,
-    ids=[f"{row}" for row in sql_tools_test_data],
+    ids=[DataBaseDialect.MY_SQL, DataBaseDialect.POSTGRES, DataBaseDialect.MS_SQL],
 )
 @pytest.mark.testcase("EPMCDME-6132")
 @pytest.mark.skipif(

codemie_test_harness/tests/conftest.py

@@ -1,6 +1,7 @@
 import os
 import subprocess
 import uuid
+import atexit
 from time import sleep
 from typing import List, Optional
 from pathlib import Path
@@ -56,6 +57,25 @@ from codemie_test_harness.tests.utils.yaml_utils import (
 logger = setup_logger(__name__)
 
 
+def force_finish_rp_launch():
+    from reportportal_client._internal.local import current
+
+    client = current()
+    if client and hasattr(client, "finish_launch") and hasattr(client, "launch_uuid"):
+        if (
+            client.launch_uuid
+            and hasattr(client, "use_own_launch")
+            and client.use_own_launch
+        ):
+            from datetime import datetime, timezone
+
+            end_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%S.%fZ")
+            client.finish_launch(end_time)
+
+
+atexit.register(force_finish_rp_launch)
+
+
 def pytest_configure(config):
     """Pytest hook that runs before test collection starts.
 
@@ -93,22 +113,6 @@ def pytest_configure(config):
         )
 
 
-def pytest_unconfigure(config):
-    """Called before test process is exited - cleanup ReportPortal and other resources."""
-    from reportportal_client._internal.local import current
-    from datetime import datetime, timezone
-
-    client = current()
-    if client and hasattr(client, "finish_launch") and hasattr(client, "launch_uuid"):
-        if (
-            client.launch_uuid
-            and hasattr(client, "use_own_launch")
-            and client.use_own_launch
-        ):
-            end_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%S.%fZ")
-            client.finish_launch(end_time)
-
-
 @pytest.fixture(scope="session")
 def client():
     return get_client()

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

@@ -3,6 +3,8 @@ import os
 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, Toolkit
 from codemie_test_harness.tests.test_data.data_management_tools_test_data import (
     ELASTIC_TOOL_TASK,
@@ -46,7 +48,7 @@ def test_workflow_with_assistant_with_elastic_tools(
 @pytest.mark.parametrize(
     "db_dialect",
     sql_tools_test_data,
-    ids=[f"{row}" for row in sql_tools_test_data],
+    ids=[DataBaseDialect.MY_SQL, DataBaseDialect.POSTGRES, DataBaseDialect.MS_SQL],
 )
 @pytest.mark.skipif(
     os.getenv("ENV") == "local", reason="Skipping this test on local environment"

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

@@ -3,6 +3,8 @@ import os
 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,
@@ -51,7 +53,7 @@ def test_workflow_with_elastic_tools(
 @pytest.mark.parametrize(
     "db_dialect",
     sql_tools_test_data,
-    ids=[f"{row}" for row in sql_tools_test_data],
+    ids=[DataBaseDialect.MY_SQL, DataBaseDialect.POSTGRES, DataBaseDialect.MS_SQL],
 )
 @pytest.mark.skipif(
     os.getenv("ENV") == "local", reason="Skipping this test on local environment"

{codemie_test_harness-0.1.148.dist-info → codemie_test_harness-0.1.149.dist-info}/METADATA

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

@@ -34,7 +34,7 @@ codemie_test_harness/tests/assistant/tools/cloud/test_cloud_tools.py,sha256=mymG
 codemie_test_harness/tests/assistant/tools/codebase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/assistant/tools/codebase/test_codebase_tools.py,sha256=KbQqf9fQGN1sPM7nKTkIsq07NpYClSm7O1wAp3rGQJg,1624
 codemie_test_harness/tests/assistant/tools/datamanagement/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-codemie_test_harness/tests/assistant/tools/datamanagement/test_assistant_with_data_management_tools.py,sha256=I8SMMWt0ci6vF1mIbd_pYJF--zGiy8L7rutwwz7tMAw,2576
+codemie_test_harness/tests/assistant/tools/datamanagement/test_assistant_with_data_management_tools.py,sha256=DQ13-x2hduqakOIayvCpgKaq95D60FhFdTs1VjIe3A8,2683
 codemie_test_harness/tests/assistant/tools/filemanagement/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/assistant/tools/filemanagement/test_assistant_with_file_management_tools.py,sha256=CrnJMvfQS3qPxpklaYBonof3W1J729wt1UefaM4tL3g,3826
 codemie_test_harness/tests/assistant/tools/git/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -57,7 +57,7 @@ codemie_test_harness/tests/assistant/tools/servicenow/__init__.py,sha256=47DEQpj
 codemie_test_harness/tests/assistant/tools/servicenow/test_servicenow_tools.py,sha256=JV_n6COaCsd0_rxQBJvvLZfPdo8BsCbG8Ti-UpzFvfQ,644
 codemie_test_harness/tests/assistant/tools/vcs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/assistant/tools/vcs/test_assistant_with_vcs_tools.py,sha256=5-QybLWmYSqHG4Sqr-IN-vNcStV4skSwCJQN44Nfats,922
-codemie_test_harness/tests/conftest.py,sha256=BRq0QKY9OHpqUGe6D-sfckCS8EW6vdHn8iDcb3qYOsk,28994
+codemie_test_harness/tests/conftest.py,sha256=8B0Jf0xoX2pt64WQY4h0p7eKfReypRWpftZ50jSDdG0,28966
 codemie_test_harness/tests/conversations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/conversations/test_conversations_endpoints.py,sha256=SQraRVrjjE4mYTs4EuGzMrO96DuOuKTiYQ4ZRBhHx70,3913
 codemie_test_harness/tests/e2e/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -250,7 +250,7 @@ codemie_test_harness/tests/workflow/assistant_tools/cloud/test_workflow_with_ass
 codemie_test_harness/tests/workflow/assistant_tools/codebase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/workflow/assistant_tools/codebase/test_worfklow_with_assistant_codebase_tools.py,sha256=d_4CJbikU5QzeRjwk0GyY0HE_iI_ojSG0gNCss2SVHQ,1962
 codemie_test_harness/tests/workflow/assistant_tools/data_management/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-codemie_test_harness/tests/workflow/assistant_tools/data_management/test_workflow_with_assistant_with_data_management_tools.py,sha256=jBTVA13jvHxa4a926PD5Ljlf4FSHvptjMVdE_6MUFoY,2849
+codemie_test_harness/tests/workflow/assistant_tools/data_management/test_workflow_with_assistant_with_data_management_tools.py,sha256=jDpCLegJDi8D_iMtoPu5K7ooa-sqBOwqyh-91lT-wzE,2957
 codemie_test_harness/tests/workflow/assistant_tools/default_integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/workflow/assistant_tools/default_integrations/test_default_integrations_for_tool.py,sha256=05YTrr9IqR9gR-jPVdNNzbSiwWbK7yHOUunPttRwoTk,8153
 codemie_test_harness/tests/workflow/assistant_tools/default_integrations/test_default_integrations_for_tool_kit.py,sha256=QmgFweqg-pSQs29O4jnanVlUZjoVejqbO35oqV2MGzk,8419
@@ -310,7 +310,7 @@ codemie_test_harness/tests/workflow/virtual_assistant_tools/cloud/test_workflow_
 codemie_test_harness/tests/workflow/virtual_assistant_tools/codebase/__init__.py,sha256=kCzJLR44IfB9ZncaLNOtyUqlzvyOG3BXclsDFS5MaQg,46
 codemie_test_harness/tests/workflow/virtual_assistant_tools/codebase/test_workflow_with_codebase_tools.py,sha256=xDuIH03ZEsnbUTp3Rz775mcXZPKrTvXbCQXpDNypWIk,3091
 codemie_test_harness/tests/workflow/virtual_assistant_tools/data_management/__init__.py,sha256=Vk3MBJAVfJA8uxRJdY3aH_5lCWkTBcE_l9xw5FzUt7I,53
-codemie_test_harness/tests/workflow/virtual_assistant_tools/data_management/test_workflow_with_data_management_tools.py,sha256=IXPDzG3ZCpTPi-HjD5h72NQbgdb8a1gRIi9929P6W0M,3137
+codemie_test_harness/tests/workflow/virtual_assistant_tools/data_management/test_workflow_with_data_management_tools.py,sha256=Znpd8WsAfsemSgAGaOBz2Ou2x31USdga3n5NIkDHt_U,3245
 codemie_test_harness/tests/workflow/virtual_assistant_tools/default_integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/workflow/virtual_assistant_tools/default_integrations/test_default_integrations_for_tool.py,sha256=mQKv_R98aiUXlPWhQruPBLPtin4gvBq8aykOvQOLRxc,8091
 codemie_test_harness/tests/workflow/virtual_assistant_tools/default_integrations/test_default_integrations_for_tool_kit.py,sha256=gC0Qj98ekHtKRRiNgtZTkb2k8TJJ5_ZVzvUQ_9r6k6c,8377
@@ -336,7 +336,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.148.dist-info/METADATA,sha256=2y0snuPrG9cFdIjTDfnL5Z71Nhaq026a-Kjei0P-36k,8998
-codemie_test_harness-0.1.148.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-codemie_test_harness-0.1.148.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
-codemie_test_harness-0.1.148.dist-info/RECORD,,
+codemie_test_harness-0.1.149.dist-info/METADATA,sha256=OI6YmvBvp-IxM9jGPP-TzNi50E-WmAOagHFD7b9nHFg,8998
+codemie_test_harness-0.1.149.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+codemie_test_harness-0.1.149.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
+codemie_test_harness-0.1.149.dist-info/RECORD,,