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