Actual flaky test fix

Author: andystaples

tests/durabletask-azuremanaged/test_dts_orchestration_e2e.py

@@ -265,9 +265,9 @@ def orchestrator(ctx: task.OrchestrationContext, _):
 #         # Raise an event to the orchestration and confirm that it does NOT complete
 #         task_hub_client.raise_orchestration_event(id, "my_event", data=42)
 #         try:
-#             state = task_hub_client.wait_for_orchestration_completion(id, timeout=30)
+#             state = task_hub_client.wait_for_orchestration_completion(id, timeout=3)
 #             assert False, "Orchestration should not have completed"
-#         except TimeoutError:
+#         except (TimeoutError, _InactiveRpcError):
 #             pass
 
 #         # Resume the orchestration and wait for it to complete

tests/durabletask/test_orchestration_e2e.py

@@ -10,6 +10,8 @@
 
 from durabletask import client, task, worker
 
+from grpc._channel import _InactiveRpcError
+
 # NOTE: These tests assume a sidecar process is running. Example command:
 #       go install github.com/microsoft/durabletask-go@main
 #       durabletask-go --port 4001
@@ -244,9 +246,9 @@ def orchestrator(ctx: task.OrchestrationContext, _):
         # Raise an event to the orchestration and confirm that it does NOT complete
         task_hub_client.raise_orchestration_event(id, "my_event", data=42)
         try:
-            state = task_hub_client.wait_for_orchestration_completion(id, timeout=30)
+            state = task_hub_client.wait_for_orchestration_completion(id, timeout=3)
             assert False, "Orchestration should not have completed"
-        except TimeoutError:
+        except (TimeoutError, _InactiveRpcError):
             pass
 
         # Resume the orchestration and wait for it to complete