fix test

Signed-off-by: Tim Li <[email protected]>

Author: timl3136

tests/cadence/worker/test_task_handler_integration.py

@@ -123,7 +123,7 @@ def track_context_activation():
             context_activated = True
 
         with patch('cadence.worker._decision_task_handler.WorkflowEngine', return_value=mock_engine):
-            with patch('cadence.worker._decision_task_handler.Context') as mock_context_class:
+            with patch('cadence._internal.workflow.workflow_engine.Context') as mock_context_class:
                 mock_context = Mock()
                 mock_context._activate = Mock(return_value=contextmanager(lambda: track_context_activation())())
                 mock_context_class.return_value = mock_context
@@ -239,7 +239,7 @@ def track_context_cleanup():
             context_cleaned_up = True
 
         with patch('cadence.worker._decision_task_handler.WorkflowEngine', return_value=mock_engine):
-            with patch('cadence.worker._decision_task_handler.Context') as mock_context_class:
+            with patch('cadence._internal.workflow.workflow_engine.Context') as mock_context_class:
                 mock_context = Mock()
                 mock_context._activate = Mock(return_value=contextmanager(lambda: track_context_cleanup())())
                 mock_context_class.return_value = mock_context