def test_clear_workflow_status_by_days(self): for i in range(10): wf_exec_id = uuid4() wf_id = uuid4() workflow_status = WorkflowStatus(wf_exec_id, wf_id, 'test') workflow_status.running() workflow_status.started_at = datetime.datetime.today( ) - datetime.timedelta(days=40) action_exec_id = uuid4() action_id = uuid4() action_status = ActionStatus(action_exec_id, action_id, 'name', 'test_app', 'test_action') workflow_status._action_statuses.append(action_status) workflow_status.completed() workflow_status.completed_at = datetime.datetime.today( ) - datetime.timedelta(days=40) self.app.running_context.execution_db.session.add(workflow_status) self.app.running_context.execution_db.session.commit() self.delete_with_status_check('/api/workflowqueue/cleardb?days=30', headers=self.headers, status_code=NO_CONTENT) wf_stats = self.app.running_context.execution_db.session.query( WorkflowStatus).all() self.assertEqual(len(wf_stats), 0) action_stats = self.app.running_context.execution_db.session.query( ActionStatus).all() self.assertEqual(len(action_stats), 0)
def get_workflow_status(self, workflow_execution_id, status): workflow_id = uuid4() workflow_status = WorkflowStatus(workflow_execution_id, workflow_id, 'workflow1') action_execution_id = uuid4() action_id = uuid4() self.app.running_context.execution_db.session.add(workflow_status) action_status = ActionStatus(action_execution_id, action_id, 'my action', 'the_app', 'the_action') self.app.running_context.execution_db.session.add(action_status) workflow_status.add_action_status(action_status) expected = { 'execution_id': str(workflow_execution_id), 'workflow_id': str(workflow_id), 'name': 'workflow1', 'status': status.name, 'current_action': action_status.as_json(summary=True)} return expected, workflow_status
def get_workflow_status(self, workflow_execution_id, status): workflow_id = uuid4() workflow_status = WorkflowStatus(workflow_execution_id, workflow_id, 'workflow1') action_execution_id = uuid4() action_id = uuid4() self.app.running_context.execution_db.session.add(workflow_status) action_status = ActionStatus(action_execution_id, action_id, 'my action', 'the_app', 'the_action') self.app.running_context.execution_db.session.add(action_status) workflow_status.add_action_status(action_status) expected = { 'execution_id': str(workflow_execution_id), 'workflow_id': str(workflow_id), 'name': 'workflow1', 'status': status.name, 'current_action': action_status.as_json(summary=True)} return expected, workflow_status
def test_read_all_workflow_status_with_action(self): wf_exec_id = uuid4() wf_id = uuid4() workflow_status = WorkflowStatus(wf_exec_id, wf_id, 'test') workflow_status.running() action_exec_id = uuid4() action_id = uuid4() action_status = ActionStatus(action_exec_id, action_id, 'name', 'test_app', 'test_action') workflow_status._action_statuses.append(action_status) self.app.running_context.execution_db.session.add(workflow_status) self.app.running_context.execution_db.session.commit() response = self.get_with_status_check('/api/workflowqueue', headers=self.headers) self.assertEqual(len(response), 1) response = response[0] self.assertIn('started_at', response) response.pop('started_at') expected = {'execution_id': str(wf_exec_id), 'workflow_id': str(wf_id), 'name': 'test', 'status': 'running', 'current_action': { 'execution_id': str(action_exec_id), 'action_id': str(action_id), 'action_name': 'test_action', 'app_name': 'test_app', 'name': 'name'}} self.assertDictEqual(response, expected)
def test_read_workflow_status(self): wf_exec_id = uuid4() wf_id = uuid4() workflow_status = WorkflowStatus(wf_exec_id, wf_id, 'test') workflow_status.running() action_exec_id = uuid4() action_id = uuid4() action_status = ActionStatus(action_exec_id, action_id, 'name', 'test_app', 'test_action') workflow_status._action_statuses.append(action_status) self.app.running_context.execution_db.session.add(workflow_status) self.app.running_context.execution_db.session.commit() response = self.get_with_status_check('/api/workflowqueue/{}'.format( str(wf_exec_id)), headers=self.headers) self.assertIn('started_at', response) response.pop('started_at') self.assertIn('action_statuses', response) self.assertIn('started_at', response['action_statuses'][-1]) response['action_statuses'][-1].pop('started_at') expected = { 'execution_id': str(wf_exec_id), 'workflow_id': str(wf_id), 'name': 'test', 'status': 'running', 'action_statuses': [{ 'execution_id': str(action_exec_id), 'action_id': str(action_id), 'name': 'name', 'app_name': 'test_app', 'action_name': 'test_action', 'status': 'executing', 'arguments': [] }] } self.assertDictEqual(response, expected)
def __action_start_callback(sender, **kwargs): workflow_execution_id = kwargs['data']['workflow']['execution_id'] current_app.running_context.execution_db.session.expire_all() action_status = current_app.running_context.execution_db.session.query(ActionStatus).filter_by( execution_id=sender['execution_id']).first() if action_status: action_status.status = ActionStatusEnum.executing else: workflow_status = current_app.running_context.execution_db.session.query(WorkflowStatus).filter_by( execution_id=workflow_execution_id).first() arguments = sender['arguments'] if 'arguments' in sender else [] action_status = ActionStatus(sender['execution_id'], sender['id'], sender['name'], sender['app_name'], sender['action_name'], json.dumps(arguments)) workflow_status.add_action_status(action_status) current_app.running_context.execution_db.session.add(action_status) current_app.running_context.execution_db.session.commit()
def make_generic_action_statuses(number): return [ ActionStatus(uuid4(), uuid4(), 'app', 'action', 'name') for _ in range(number) ]