From 8f8d9d30e43123610b81092bae28f126ae5d2f7b Mon Sep 17 00:00:00 2001 From: mike cullerton Date: Wed, 12 May 2021 13:52:18 -0400 Subject: [PATCH] Added test for get_primary_workflow --- tests/workflow/test_workflow_service.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/workflow/test_workflow_service.py b/tests/workflow/test_workflow_service.py index a4af4edc..dc6eb01e 100644 --- a/tests/workflow/test_workflow_service.py +++ b/tests/workflow/test_workflow_service.py @@ -10,6 +10,7 @@ from example_data import ExampleDataLoader from crc import db from crc.models.task_event import TaskEventModel from crc.models.api_models import Task +from crc.models.file import FileModel from crc.api.common import ApiError @@ -114,3 +115,12 @@ class TestWorkflowService(BaseTest): result2 = WorkflowService.get_dot_value(path, {"a.b.c":"garbage"}) self.assertEqual("garbage", result2) + + def test_get_primary_workflow(self): + + workflow = self.create_workflow('hello_world') + workflow_spec_id = workflow.workflow_spec.id + primary_workflow = WorkflowService.get_primary_from_workflow_spec(workflow_spec_id) + self.assertIsInstance(primary_workflow, FileModel) + self.assertEqual(workflow_spec_id, primary_workflow.workflow_spec_id) + self.assertEqual('hello_world.bpmn', primary_workflow.name)