Extends test

Signed-off-by: Elena Kolevska <[email protected]>

Author: elena-kolevska

ext/dapr-ext-workflow/tests/test_workflow_client.py

@@ -30,7 +30,7 @@
 mock_resume_result = 'resume001'
 mock_purge_result = 'purge001'
 mock_instance_id = 'instance001'
-wf_exists = False
+wf_status = 'not-found'
 
 
 class SimulatedRpcError(RpcError):
@@ -57,11 +57,15 @@ def schedule_new_orchestration(
         return mock_schedule_result
 
     def get_orchestration_state(self, instance_id, fetch_payloads):
-        global wf_exists
-        if not wf_exists:
+        global wf_status
+        if wf_status == 'not-found':
             raise SimulatedRpcError(code='UNKNOWN', details='no such instance exists')
-
-        return self._inner_get_orchestration_state(instance_id, client.OrchestrationStatus.PENDING)
+        elif wf_status == 'found':
+            return self._inner_get_orchestration_state(
+                instance_id, client.OrchestrationStatus.PENDING
+            )
+        else:
+            raise SimulatedRpcError(code='UNKNOWN', details='unknown error')
 
     def wait_for_orchestration_start(self, instance_id, fetch_payloads, timeout):
         return self._inner_get_orchestration_state(instance_id, client.OrchestrationStatus.RUNNING)
@@ -118,13 +122,20 @@ def test_client_functions(self):
             )
             assert actual_schedule_result == mock_schedule_result
 
+            global wf_status
+            wf_status = 'not-found'
             actual_get_result = wfClient.get_workflow_state(
                 instance_id=mock_instance_id, fetch_payloads=True
             )
             assert actual_get_result is None
 
-            global wf_exists
-            wf_exists = True
+            wf_status = 'error'
+            with self.assertRaises(RpcError):
+                wfClient.get_workflow_state(instance_id=mock_instance_id, fetch_payloads=True)
+
+            assert actual_get_result is None
+
+            wf_status = 'found'
             actual_get_result = wfClient.get_workflow_state(
                 instance_id=mock_instance_id, fetch_payloads=True
             )