codemie-test-harness 0.1.213__py3-none-any.whl → 0.1.214__py3-none-any.whl

codemie_test_harness/tests/utils/base_utils.py

@@ -4,21 +4,11 @@ import re
 import string
 import time
 import unicodedata
-from time import sleep
 
 from hamcrest import assert_that, equal_to, contains_string
 
 from codemie_sdk import CodeMieClient
-from codemie_sdk.exceptions import ApiError
-from codemie_sdk.models.workflow_state import (
-    WorkflowExecutionState,
-    WorkflowExecutionStatusEnum,
-)
-from codemie_sdk.services.workflow_execution_state import (
-    WorkflowExecutionStateService,
-)
 from codemie_test_harness.tests import autotest_entity_prefix
-from codemie_test_harness.tests.utils.credentials_manager import CredentialsManager
 
 
 class BaseUtils:
@@ -47,29 +37,6 @@ def to_camel_case(input_string):
     return camel_case
 
 
-def wait_for_completion(
-    execution_state_service: WorkflowExecutionStateService,
-    state_id: str,
-    timeout: int = int(CredentialsManager.get_parameter("DEFAULT_TIMEOUT", "120")),
-    pool_interval: int = 3,
-) -> WorkflowExecutionState:
-    start_time = time.time()
-    while time.time() - start_time < timeout:
-        states = [row for row in execution_state_service.list() if row.id == state_id]
-        if len(states) == 0:
-            sleep(pool_interval)
-            continue
-        state = states[0]
-        if state.status == WorkflowExecutionStatusEnum.SUCCEEDED:
-            return state
-        elif state.status == WorkflowExecutionStatusEnum.FAILED:
-            raise ApiError(
-                f"State execution failed: {execution_state_service.get_output(state_id)}"
-            )
-        sleep(pool_interval)
-    raise TimeoutError("State was not executed within the timeout period.")
-
-
 def clean_json(json_str):
     try:
         # Attempt to parse the JSON string directly

codemie_test_harness/tests/utils/similarity_check.py

@@ -3,9 +3,9 @@ from hamcrest import assert_that, greater_than_or_equal_to
 
 from codemie_test_harness.tests.utils.base_utils import (
     wait_for_entity,
-    wait_for_completion,
     clean_json,
 )
+from codemie_test_harness.tests.utils.workflow_utils import WorkflowUtils
 
 
 class SimilarityCheck:
@@ -27,7 +27,7 @@ class SimilarityCheck:
             lambda: states_service.list(),
             entity_name="similarity_analysis",
         )
-        wait_for_completion(execution_state_service=states_service, state_id=state.id)
+        WorkflowUtils.wait_for_completion(service=states_service, _id=state.id)
         output = states_service.get_output(state_id=state.id).output
         output = int(clean_json(output).get("similarity_rank"))
         assert_that(

codemie_test_harness/tests/utils/workflow_utils.py

@@ -1,21 +1,37 @@
 import os
 from pathlib import Path
 from typing import Optional
+import time
 
+
+from codemie_sdk.exceptions import ApiError
 from codemie_sdk.models.workflow import (
     WorkflowCreateRequest,
     WorkflowUpdateRequest,
     WorkflowMode,
+    WorkflowExecution,
+)
+from codemie_sdk.models.workflow_state import (
+    WorkflowExecutionState,
+    WorkflowExecutionStatusEnum,
+)
+from codemie_sdk.services.workflow_execution import WorkflowExecutionService
+from codemie_sdk.services.workflow_execution_state import WorkflowExecutionStateService
+
+from codemie_test_harness.tests import (
+    PROJECT,
+    VERIFY_SSL,
+    API_DOMEN,
+    CredentialsManager,
 )
-
-from codemie_test_harness.tests import PROJECT, VERIFY_SSL, API_DOMEN
 from codemie_test_harness.tests.utils.base_utils import (
     BaseUtils,
     get_random_name,
     wait_for_entity,
-    wait_for_completion,
 )
 from codemie_test_harness.tests.utils.http_utils import RequestHandler
+from time import sleep
+
 
 workflow_endpoint = "/v1/workflows"
 
@@ -106,17 +122,21 @@ class WorkflowUtils(BaseUtils):
         file_name: Optional[str] = None,
     ):
         self.client.workflows.run(workflow, user_input=user_input, file_name=file_name)
-        executions = self.client.workflows.executions(workflow)
+        executions_service = self.client.workflows.executions(workflow)
         execution_id = next(
-            row.execution_id for row in executions.list() if row.prompt == user_input
+            row.execution_id
+            for row in executions_service.list()
+            if row.prompt == user_input
         )
-        states_service = executions.states(execution_id)
+        states_service = executions_service.states(execution_id)
         state = wait_for_entity(
             lambda: states_service.list(),
             entity_name=execution_name,
         )
 
-        wait_for_completion(execution_state_service=states_service, state_id=state.id)
+        self.wait_for_completion(service=states_service, _id=state.id)
+        self.wait_for_completion(service=executions_service, _id=execution_id)
+
         return states_service.get_output(state_id=state.id).output
 
     def run_workflow(self, workflow_id, user_input=""):
@@ -279,3 +299,39 @@ class WorkflowUtils(BaseUtils):
                 process_thought(thought)
 
         return triggered_tools
+
+    @staticmethod
+    def wait_for_completion(
+        service: WorkflowExecutionStateService | WorkflowExecutionService,
+        _id: str,
+        timeout: int = int(CredentialsManager.get_parameter("DEFAULT_TIMEOUT", "120")),
+        pool_interval: int = 3,
+    ) -> WorkflowExecutionState | WorkflowExecution:
+        is_workflow_execution_state_state = isinstance(
+            service, WorkflowExecutionStateService
+        )
+        start_time = time.time()
+        while time.time() - start_time < timeout:
+            executions_or_states = [
+                row
+                for row in service.list()
+                if row.id == _id or row.execution_id == _id
+            ]
+            if len(executions_or_states) == 0:
+                sleep(pool_interval)
+                continue
+            execution_or_state = executions_or_states[0]
+            if execution_or_state.status == WorkflowExecutionStatusEnum.SUCCEEDED:
+                return execution_or_state
+            elif execution_or_state.status == WorkflowExecutionStatusEnum.FAILED:
+                if is_workflow_execution_state_state:
+                    raise ApiError(f"State execution failed: {service.get_output(_id)}")
+                else:
+                    raise ApiError(
+                        f"Execution failed. Execution id: {execution_or_state.execution_id}"
+                    )
+            sleep(pool_interval)
+        if is_workflow_execution_state_state:
+            raise TimeoutError("State was not executed within the timeout period.")
+        else:
+            raise TimeoutError("Workflow was not executed within the timeout period.")

codemie_test_harness/tests/utils/yaml_utils.py

@@ -54,7 +54,6 @@ class StateModel(BaseModel):
 
 
 class WorkflowYamlModel(BaseModel):
-    enable_summarization_node: bool = False
     tools: List[ToolModel] = Field(default_factory=list)
     assistants: List[AssistantModel] = Field(default_factory=list)
     states: List[StateModel] = Field(default_factory=list)

{codemie_test_harness-0.1.213.dist-info → codemie_test_harness-0.1.214.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: codemie-test-harness
-Version: 0.1.213
+Version: 0.1.214
 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.213)
+Requires-Dist: codemie-sdk-python (==0.1.214)
 Requires-Dist: pytest (>=8.4.1,<9.0.0)
 Requires-Dist: pytest-playwright (>=0.7.0,<0.8.0)
 Requires-Dist: pytest-repeat (>=0.9.3,<0.10.0)

{codemie_test_harness-0.1.213.dist-info → codemie_test_harness-0.1.214.dist-info}/RECORD

@@ -269,7 +269,7 @@ codemie_test_harness/tests/ui/workflows/test_workflows.py,sha256=zRBFiQYhJ_MWKGG
 codemie_test_harness/tests/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/utils/assistant_utils.py,sha256=2s1MikCfIcM3UlkcuwaedyxI2hePuWT3h8F3SDjBCtk,8022
 codemie_test_harness/tests/utils/aws_parameters_store.py,sha256=YAVpvwElkKZJZvzSVxtOue1Gjs-kvSBS2y5QvIlz484,3267
-codemie_test_harness/tests/utils/base_utils.py,sha256=SJV-m8KuC6MAU4W5a1sWLz6Nw-1-buc0f-W1hbA1OIY,8270
+codemie_test_harness/tests/utils/base_utils.py,sha256=MaYrJIQVmOYjqMIkfr7fJYgZQhnIlRr-3Xp2QlA1Ak0,7000
 codemie_test_harness/tests/utils/client_factory.py,sha256=xGta0ZaVYzWfwJ4cu3f89KkGc_R5Bq-9lqnhr57x_2w,972
 codemie_test_harness/tests/utils/confluence_utils.py,sha256=auhip1ntqSDsHWAoWCxQxfuNv05BinS6TWXyg_F2dfc,4544
 codemie_test_harness/tests/utils/constants.py,sha256=aGs0gdHB38Ozd-UyCKNpWRoQXMy3D0bjmfxiPcdZdqY,1165
@@ -290,11 +290,11 @@ codemie_test_harness/tests/utils/notification_utils.py,sha256=WyrO8Udp4O8-C4eI6w
 codemie_test_harness/tests/utils/provider_utils.py,sha256=NSI5vG1IPpzsytuwQAGbx9Fe0xN6OYQ9Gp9mvjO0H2g,5115
 codemie_test_harness/tests/utils/pytest_utils.py,sha256=k-mEjX2qpnh37sqKpJqYhZT6BV9974y_KaAhv8Xj9GI,284
 codemie_test_harness/tests/utils/search_utils.py,sha256=SrXiB2d9wiI5ka9bgg0CD73GOX_1mqi2Hz5FBm5DsEU,1435
-codemie_test_harness/tests/utils/similarity_check.py,sha256=1U66NGh6esISKABodtVobE2WnuFt0f6vcK3qUri6ZqU,1485
+codemie_test_harness/tests/utils/similarity_check.py,sha256=2URqvD3Ft7efwLmhh2iYVsXrYboP9f-_B_ekZmJn0ac,1527
 codemie_test_harness/tests/utils/user_utils.py,sha256=zJNrmL3Fb7iGuaVRobUMwJ2Og6NqEPcM_9lw60m18T8,242
 codemie_test_harness/tests/utils/webhook_utils.py,sha256=YjyLwAqQjR12vYFOUmYhJCJIyZvKm4SvU-1oIjIYNqg,340
-codemie_test_harness/tests/utils/workflow_utils.py,sha256=d0dT5ciaLCDrRPZ-uwgGvWBhOdDExebH_h_Q3oJ4KOY,9261
-codemie_test_harness/tests/utils/yaml_utils.py,sha256=y9fUf4u4G4SoCktPOwaC5x71iaDKhktbz_XUfI9kNis,1661
+codemie_test_harness/tests/utils/workflow_utils.py,sha256=tCSJ-6QyumveTP15ZiNzlkBgGxg97GcKV5Kkc02k_Zs,11409
+codemie_test_harness/tests/utils/yaml_utils.py,sha256=iIdEl-rUUh1LgzAmD_mjfftthhvlzXyCuA37yBoH0Gw,1617
 codemie_test_harness/tests/webhook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codemie_test_harness/tests/webhook/test_webhook_service.py,sha256=POmxQG0tpcNW9-yKQ62CcnQpUEFYlTOs0_4H9MijIHY,8127
 codemie_test_harness/tests/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -402,7 +402,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=D835gaRbCnB4va5mi9TdA_u9StSpGXQ_fgzwW0S2pwo,1173
 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=Se9imIiBYuJU78m1pLu0g4ZmHygKZjr6JjIWkGXTy1Q,1364
-codemie_test_harness-0.1.213.dist-info/METADATA,sha256=yFQaB6HKgMkLqmiIDGbo0-wu4bwEWhyGznZEIxz7FiI,27184
-codemie_test_harness-0.1.213.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-codemie_test_harness-0.1.213.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
-codemie_test_harness-0.1.213.dist-info/RECORD,,
+codemie_test_harness-0.1.214.dist-info/METADATA,sha256=P60BQniiZByraDIAxld3bYLAQ1l5dhPAk8OLFvxwKVs,27184
+codemie_test_harness-0.1.214.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+codemie_test_harness-0.1.214.dist-info/entry_points.txt,sha256=n98t-EOM5M1mnMl_j2X4siyeO9zr0WD9a5LF7JyElIM,73
+codemie_test_harness-0.1.214.dist-info/RECORD,,