Adds test

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

Author: elena-kolevska

ext/dapr-ext-workflow/dapr/ext/workflow/dapr_workflow_client.py

@@ -139,7 +139,7 @@ def get_workflow_state(
                 self._logger.warning(f"Workflow instance not found: {instance_id}")
                 return None
             self._logger.error(f"Unhandled RPC error while fetching workflow state: {error.code()} - {error.details()}")
-            raise error
+            raise
 
     def wait_for_workflow_start(
         self, instance_id: str, *, fetch_payloads: bool = False, timeout_in_seconds: int = 60

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

@@ -21,6 +21,7 @@
 from dapr.ext.workflow.dapr_workflow_client import DaprWorkflowClient
 from durabletask import client
 import durabletask.internal.orchestrator_service_pb2 as pb
+from grpc import StatusCode, RpcError
 
 mock_schedule_result = 'workflow001'
 mock_raise_event_result = 'event001'
@@ -29,6 +30,18 @@
 mock_resume_result = 'resume001'
 mock_purge_result = 'purge001'
 mock_instance_id = 'instance001'
+wf_exists = False
+
+class SimulatedRpcError(RpcError):
+    def __init__(self, code, details):
+        self._code = code
+        self._details = details
+
+    def code(self):
+        return self._code
+
+    def details(self):
+        return self._details
 
 
 class FakeTaskHubGrpcClient:
@@ -43,8 +56,13 @@ def schedule_new_orchestration(
         return mock_schedule_result
 
     def get_orchestration_state(self, instance_id, fetch_payloads):
+        global wf_exists
+        if not wf_exists:
+            raise SimulatedRpcError(code="UNKNOWN", details="no such instance exists")
+
         return self._inner_get_orchestration_state(instance_id, client.OrchestrationStatus.PENDING)
 
+
     def wait_for_orchestration_start(self, instance_id, fetch_payloads, timeout):
         return self._inner_get_orchestration_state(instance_id, client.OrchestrationStatus.RUNNING)
 
@@ -100,6 +118,13 @@ def test_client_functions(self):
             )
             assert actual_schedule_result == mock_schedule_result
 
+            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
             actual_get_result = wfClient.get_workflow_state(
                 instance_id=mock_instance_id, fetch_payloads=True
             )