test fix

jssmith · jssmith · commit a1297b172904 · 2025-06-29T21:06:01.000-07:00
diff --git a/tests/expense/test_expense_workflow.py b/tests/expense/test_expense_workflow.py
@@ -362,10 +362,18 @@ async def register_decision_timeout_check(expense_id: str) -> None:
                 # Check that we're called with 10 minute timeout
                 activity_info = activity.info()
                 timeout_calls.append(("wait", activity_info.start_to_close_timeout))
-                # Simulate automatic decision if one was scheduled
+                # In time-skipping mode, send the decision immediately
                 if expense_id in mock_ui.scheduled_decisions:
-                    # Decision will be sent by the scheduled task
-                    pass
+                    decision = mock_ui.scheduled_decisions[expense_id]
+                    if expense_id in mock_ui.workflow_map:
+                        workflow_id = mock_ui.workflow_map[expense_id]
+                        handle = client.get_workflow_handle(workflow_id)
+                        try:
+                            # Send signal immediately when registering
+                            await handle.signal("expense_decision_signal", decision)
+                        except Exception:
+                            # Ignore errors in time-skipping mode
+                            pass
                 return None
 
             return register_decision_timeout_check
