def marker_recorded_event(marker_header_json):
event = HistoryEvent()
event.event_id = 20
event.event_type = EventType.MarkerRecorded
event.marker_recorded_event_attributes = MarkerRecordedEventAttributes()
event.marker_recorded_event_attributes.marker_name = "the-marker-name"
event.marker_recorded_event_attributes.header = Header()
event.marker_recorded_event_attributes.header.fields[MUTABLE_MARKER_HEADER_KEY] = bytes(marker_header_json, "utf-8")
event.marker_recorded_event_attributes.details = b'blah-blah'
return event
def version_marker_recorded_event(marker_header_json):
marker_header_json = MarkerHeader(id="abc", event_id=55, access_count=0).to_json()
event = HistoryEvent()
event.event_id = 20
event.event_type = EventType.MarkerRecorded
event.marker_recorded_event_attributes = MarkerRecordedEventAttributes()
event.marker_recorded_event_attributes.marker_name = VERSION_MARKER_NAME
event.marker_recorded_event_attributes.header = Header()
event.marker_recorded_event_attributes.header.fields[MUTABLE_MARKER_HEADER_KEY] = bytes(marker_header_json, "utf-8")
event.marker_recorded_event_attributes.details = b'4'
return event
def test_handle_decision_task_failed(self):
event = HistoryEvent(event_id=15)
event.event_type = EventType.DecisionTaskFailed
event.decision_task_failed_event_attributes = DecisionTaskFailedEventAttributes(
)
event.decision_task_failed_event_attributes.cause = DecisionTaskFailedCause.RESET_WORKFLOW
event.decision_task_failed_event_attributes.new_run_id = "the-new-run-id"
self.decider.decision_context = decision_context = MagicMock()
self.decider.handle_decision_task_failed(event)
decision_context.set_current_run_id.assert_called()
args, kwargs = decision_context.set_current_run_id.call_args_list[0]
assert args[0] == "the-new-run-id"
