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"