def version_decision_context(version_marker_recorded_event): decider = ReplayDecider(execution_id=Mock(), workflow_type=Mock(), worker=Mock()) decision_context = DecisionContext(decider=decider) decider.decision_context = decision_context decider.decision_context.workflow_clock.version_handler.decision_context = decision_context decision_context.decider.decision_events = DecisionEvents(events=[], decision_events=[version_marker_recorded_event], replay=True, replay_current_time_milliseconds=0, next_decision_event_id=20) decision_context.decider.next_decision_event_id = 20 return decision_context
def mock_decision_context(decider: ReplayDecider): decision_context = MagicMock() decider.decision_context = decision_context return decision_context