Real fix

Author: andystaples

durabletask/testing/in_memory_backend.py

@@ -309,7 +309,14 @@ def RaiseEvent(self, request: pb.RaiseEventRequest, context):
             )
             instance.pending_events.append(event)
             instance.last_updated_at = datetime.now(timezone.utc)
-            self._enqueue_orchestration(instance.instance_id)
+
+            # Don't dispatch work for suspended or terminal orchestrations;
+            # suspended events will be delivered when the orchestration is
+            # resumed, and terminal orchestrations can't process new events.
+            not_terminal = not self._is_terminal_status(instance.status)
+            not_suspended = instance.status != pb.ORCHESTRATION_STATUS_SUSPENDED
+            if not_terminal and not_suspended:
+                self._enqueue_orchestration(instance.instance_id)
 
         self._logger.info(f"Raised event '{request.name}' for instance '{request.instanceId}'")
         return pb.RaiseEventResponse()

tests/durabletask/test_orchestration_async_e2e.py

@@ -128,13 +128,10 @@ def orchestrator(ctx: task.OrchestrationContext, _):
                 assert state is not None
             assert state.runtime_status == client.OrchestrationStatus.SUSPENDED
 
-            # Small delay to ensure the suspension is fully enforced
-            await asyncio.sleep(1)
-
             # Raise an event and confirm that it does NOT complete while suspended
             await c.raise_orchestration_event(id, "my_event", data=42)
             try:
-                state = await c.wait_for_orchestration_completion(id, timeout=5)
+                state = await c.wait_for_orchestration_completion(id, timeout=3)
                 assert False, "Orchestration should not have completed"
             except TimeoutError:
                 pass

tests/durabletask/test_orchestration_e2e.py

@@ -269,13 +269,10 @@ def orchestrator(ctx: task.OrchestrationContext, _):
             assert state is not None
         assert state.runtime_status == client.OrchestrationStatus.SUSPENDED
 
-        # Small delay to ensure the suspension is fully enforced
-        time.sleep(1)
-
         # 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=5)
+            state = task_hub_client.wait_for_orchestration_completion(id, timeout=3)
             assert False, "Orchestration should not have completed"
         except TimeoutError:
             pass