def task_events(self, task_id):
events = [
x for x in TaskEvent.select().join(Task).where(
Task.task_id == task_id).order_by(TaskEvent.timestamp)
]
if len(events):
return events
else:
raise ZeroResults(f'No results returned for task_id: {task_id}')
def task_completed_event(self, task_id):
events = [
x for x in (TaskEvent.select().join(Task).where(
(Task.task_id == task_id)
& (TaskEvent.status.in_(['failed', 'completed']))))
]
if len(events):
return events[0]
else:
raise ZeroResults(f'No results returned for task_id: {task_id}')
def test_get_task_events_no_events(api, client, testtask):
task_id = testtask().json['task']['task_id']
task = Task.get(Task.task_id == task_id)
TaskEvent.delete() \
.where(TaskEvent.task == task) \
.execute()
events = TaskEvent.select() \
.join(Task) \
.where(Task.task_id == task_id)
assert len([x for x in events]) == 0, 'events where not cleared out'
resp = client.get(api.url_for(GMan, task_id=task_id, events='events'))
assert resp.status_code == 404