import json from tests.base_test import BaseTest from crc import session from crc.models.file import FileModel from crc.models.workflow import WorkflowSpecModel, WorkflowSpecModelSchema, WorkflowModel, WorkflowSpecCategoryModel, WorkflowSpecCategoryModelSchema from example_data import ExampleDataLoader class TestWorkflowSpec(BaseTest): def test_list_workflow_specifications(self): self.load_example_data() spec = session.query(WorkflowSpecModel).first() rv = self.app.get('/v1.0/workflow-specification', follow_redirects=True, content_type="application/json",headers=self.logged_in_headers()) self.assert_success(rv) json_data = json.loads(rv.get_data(as_text=True)) specs = WorkflowSpecModelSchema(many=True).load(json_data, session=session) spec2 = specs[0] self.assertEqual(spec.id, spec2.id) self.assertEqual(spec.display_name, spec2.display_name) self.assertEqual(spec.description, spec2.description) def test_add_new_workflow_specification(self): self.load_example_data() num_before = session.query(WorkflowSpecModel).count() category_id = session.query(WorkflowSpecCategoryModel).first().id category_count = session.query(WorkflowSpecModel).filter_by(category_id=category_id).count() spec = WorkflowSpecModel(id='make_cookies', display_name='Cooooookies', description='Om nom nom delicious cookies', category_id=category_id, standalone=False) rv = self.app.post('/v1.0/workflow-specification', headers=self.logged_in_headers(), content_type="application/json", data=json.dumps(WorkflowSpecModelSchema().dump(spec))) self.assert_success(rv) db_spec = session.query(WorkflowSpecModel).filter_by(id='make_cookies').first() self.assertEqual(spec.display_name, db_spec.display_name) num_after = session.query(WorkflowSpecModel).count() self.assertEqual(num_after, num_before + 1) self.assertEqual(category_count, db_spec.display_order) category_count_after = session.query(WorkflowSpecModel).filter_by(category_id=category_id).count() self.assertEqual(category_count_after, category_count + 1) def test_get_workflow_specification(self): self.load_example_data() db_spec = session.query(WorkflowSpecModel).first() rv = self.app.get('/v1.0/workflow-specification/%s' % db_spec.id, headers=self.logged_in_headers()) self.assert_success(rv) json_data = json.loads(rv.get_data(as_text=True)) api_spec = WorkflowSpecModelSchema().load(json_data, session=session) self.assertEqual(db_spec, api_spec) def test_update_workflow_specification(self): self.load_example_data() category_id = 99 category = WorkflowSpecCategoryModel(id=category_id, display_name="It's a trap!", display_order=0) session.add(category) session.commit() db_spec_before: WorkflowSpecModel = session.query(WorkflowSpecModel).first() spec_id = db_spec_before.id self.assertNotEqual(db_spec_before.category_id, category_id) db_spec_before.category_id = category_id rv = self.app.put('/v1.0/workflow-specification/%s' % spec_id, content_type="application/json", headers=self.logged_in_headers(), data=json.dumps(WorkflowSpecModelSchema().dump(db_spec_before))) self.assert_success(rv) json_data = json.loads(rv.get_data(as_text=True)) api_spec = WorkflowSpecModelSchema().load(json_data, session=session) self.assertEqual(db_spec_before, api_spec) db_spec_after: WorkflowSpecModel = session.query(WorkflowSpecModel).filter_by(id=spec_id).first() self.assertIsNotNone(db_spec_after.category_id) self.assertIsNotNone(db_spec_after.category) self.assertEqual(db_spec_after.category.display_name, category.display_name) self.assertEqual(db_spec_after.category.display_order, category.display_order) def test_delete_workflow_specification(self): self.load_example_data() spec_id = 'random_fact' self.load_test_spec(spec_id) num_specs_before = session.query(WorkflowSpecModel).filter_by(id=spec_id).count() self.assertEqual(num_specs_before, 1) num_files_before = session.query(FileModel).filter_by(workflow_spec_id=spec_id).count() num_workflows_before = session.query(WorkflowModel).filter_by(workflow_spec_id=spec_id).count() self.assertGreater(num_files_before + num_workflows_before, 0) rv = self.app.delete('/v1.0/workflow-specification/' + spec_id, headers=self.logged_in_headers()) self.assert_success(rv) num_specs_after = session.query(WorkflowSpecModel).filter_by(id=spec_id).count() self.assertEqual(0, num_specs_after) # Make sure that all items in the database with the workflow spec ID are deleted as well. num_files_after = session.query(FileModel).filter_by(workflow_spec_id=spec_id).count() num_workflows_after = session.query(WorkflowModel).filter_by(workflow_spec_id=spec_id).count() self.assertEqual(num_files_after + num_workflows_after, 0) def test_display_order_after_delete_spec(self): self.load_example_data() workflow_spec_category = session.query(WorkflowSpecCategoryModel).first() spec_model_1 = WorkflowSpecModel(id='test_spec_1', display_name='Test Spec 1', description='Test Spec 1 Description', category_id=workflow_spec_category.id, display_order=1, standalone=False) spec_model_2 = WorkflowSpecModel(id='test_spec_2', display_name='Test Spec 2', description='Test Spec 2 Description', category_id=workflow_spec_category.id, display_order=2, standalone=False) spec_model_3 = WorkflowSpecModel(id='test_spec_3', display_name='Test Spec 3', description='Test Spec 3 Description', category_id=workflow_spec_category.id, display_order=3, standalone=False) session.add(spec_model_1) session.add(spec_model_2) session.add(spec_model_3) session.commit() self.app.delete('/v1.0/workflow-specification/test_spec_2', headers=self.logged_in_headers()) test_order = 0 specs = session.query(WorkflowSpecModel).\ filter(WorkflowSpecModel.category_id == workflow_spec_category.id).\ order_by(WorkflowSpecModel.display_order).\ all() for test_spec in specs: self.assertEqual(test_order, test_spec.display_order) test_order += 1 def test_get_standalone_workflow_specs(self): self.load_example_data() category = session.query(WorkflowSpecCategoryModel).first() ExampleDataLoader().create_spec('hello_world', 'Hello World', category_id=category.id, standalone=True, from_tests=True) rv = self.app.get('/v1.0/workflow-specification?standalone=true', headers=self.logged_in_headers()) self.assertEqual(1, len(rv.json)) ExampleDataLoader().create_spec('email_script', 'Email Script', category_id=category.id, standalone=True, from_tests=True) rv = self.app.get('/v1.0/workflow-specification?standalone=true', headers=self.logged_in_headers()) self.assertEqual(2, len(rv.json)) def test_get_workflow_from_workflow_spec(self): self.load_example_data() spec = ExampleDataLoader().create_spec('hello_world', 'Hello World', standalone=True, from_tests=True) rv = self.app.post(f'/v1.0/workflow-specification/{spec.id}', headers=self.logged_in_headers()) self.assert_success(rv) self.assertEqual('hello_world', rv.json['workflow_spec_id']) self.assertEqual('Task_GetName', rv.json['next_task']['name']) def test_add_workflow_spec_category(self): self.load_example_data() count = session.query(WorkflowSpecCategoryModel).count() category = WorkflowSpecCategoryModel( id=count, display_name='Another Test Category', display_order=0 ) rv = self.app.post(f'/v1.0/workflow-specification-category', headers=self.logged_in_headers(), content_type="application/json", data=json.dumps(WorkflowSpecCategoryModelSchema().dump(category)) ) self.assert_success(rv) result = session.query(WorkflowSpecCategoryModel).filter(WorkflowSpecCategoryModel.id==count).first() self.assertEqual('Another Test Category', result.display_name) self.assertEqual(count, result.id) def test_update_workflow_spec_category(self): self.load_example_data() category = session.query(WorkflowSpecCategoryModel).first() display_name_before = category.display_name new_display_name = display_name_before + '_asdf' self.assertNotEqual(display_name_before, new_display_name) category.display_name = new_display_name rv = self.app.put(f'/v1.0/workflow-specification-category/{category.id}', content_type="application/json", headers=self.logged_in_headers(), data=json.dumps(WorkflowSpecCategoryModelSchema().dump(category))) self.assert_success(rv) json_data = json.loads(rv.get_data(as_text=True)) self.assertEqual(new_display_name, json_data['display_name']) def test_delete_workflow_spec_category(self): self.load_example_data() category_model_1 = WorkflowSpecCategoryModel( id=1, display_name='Test Category 1', display_order=1 ) category_model_2 = WorkflowSpecCategoryModel( id=2, display_name='Test Category 2', display_order=2 ) category_model_3 = WorkflowSpecCategoryModel( id=3, display_name='Test Category 3', display_order=3 ) session.add(category_model_1) session.add(category_model_2) session.add(category_model_3) session.commit() self.app.delete('/v1.0/workflow-spec-category/2', headers=self.logged_in_headers()) test_order = 0 categories = session.query(WorkflowSpecCategoryModel).order_by(WorkflowSpecCategoryModel.display_order).all() for test_category in categories: self.assertEqual(test_order, test_category.display_order) test_order += 1 def test_add_library_with_category_id(self): self.load_example_data() category_id = session.query(WorkflowSpecCategoryModel).first().id spec = WorkflowSpecModel(id='test_spec', display_name='Test Spec', description='Library with a category id', category_id=category_id, standalone=False, library=True) rv = self.app.post('/v1.0/workflow-specification', headers=self.logged_in_headers(), content_type="application/json", data=json.dumps(WorkflowSpecModelSchema().dump(spec))) self.assert_success(rv) # libraries don't get category_ids # so, the category_id should not get set self.assertIsNone(rv.json['category_id'])