def add_completed_event(context_builder: ContextBuilder, id_: int, name: str,
result):
context_builder.add_task_scheduled_event(name=name, id_=id_)
context_builder.add_orchestrator_completed_event()
context_builder.add_orchestrator_started_event()
context_builder.add_task_completed_event(id_=id_,
result=json.dumps(result))
def _complete_event(context: ContextBuilder,
id_counter: int) -> Tuple[ContextBuilder, int]:
"""Add event / task completions 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
Returns
-------
Tuple[ContextBuilder, int]
The updated context, the updated id_counter
"""
for id_, city in zip(scheduled_ids, CITIES):
result = f"\"{RESULT_PREFIX}{city}\""
context.add_task_completed_event(id_=id_, result=result)
id_counter += 1
return context, id_counter
def add_completed_http_events(context_builder: ContextBuilder, id_: int,
result: str):
context_builder.add_task_scheduled_event(name=HTTP_ACTION_NAME, id_=id_)
context_builder.add_orchestrator_completed_event()
context_builder.add_orchestrator_started_event()
context_builder.add_task_completed_event(id_=id_, result=result)
def add_hello_completed_events(context_builder: ContextBuilder, id_: int,
result: str):
context_builder.add_task_scheduled_event(name='Hello', id_=id_)
context_builder.add_orchestrator_completed_event()
context_builder.add_orchestrator_started_event()
context_builder.add_task_completed_event(id_=id_, result=result)
