def _fire_timer(context: ContextBuilder, id_counter: int, deadlines: List[datetime]) -> Tuple[ContextBuilder, int]: """Add timer fired events to the context. Parameters ---------- context: ContextBuilder Orchestration context mock, to which we'll add the event completion events id_counter: int The current event counter deadlines: List[datetime] List of dates at which to fire the timers Returns ------- Tuple[ContextBuilder, int]: The updated context, the updated id_counter """ for id_, fire_at in deadlines: context.add_timer_fired_event(id_=id_, fire_at=fire_at) id_counter += 1 return context, id_counter
def add_timer_fired_events(context_builder: ContextBuilder, id_: int, timestamp: str, is_played: bool = True): fire_at: str = context_builder.add_timer_created_event(id_, timestamp) context_builder.add_orchestrator_completed_event() context_builder.add_orchestrator_started_event() context_builder.add_timer_fired_event(id_=id_, fire_at=fire_at, is_played=is_played)
def add_retry_timer_events(context_builder: ContextBuilder, id_: int): fire_at = context_builder.add_timer_created_event(id_) context_builder.add_orchestrator_completed_event() context_builder.add_orchestrator_started_event() context_builder.add_timer_fired_event(id_=id_, fire_at=fire_at)