cr-connect-workflow/tests/test_auto_set_primary_bpmn.py

36 lines
1.7 KiB
Python

from tests.base_test import BaseTest
import io
import json
from crc.models.workflow import WorkflowSpecInfo, WorkflowSpecInfoSchema
from crc.services.workflow_spec_service import WorkflowSpecService
class TestAutoSetPrimaryBPMN(BaseTest):
def test_auto_set_primary_bpmn(self):
category = self.assure_category_exists()
spec = WorkflowSpecInfo(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(WorkflowSpecInfoSchema().dump(spec)))
self.assert_success(rv)
db_spec = WorkflowSpecService().get_spec(spec.id)
self.assertEqual(spec.display_name, db_spec.display_name)
self.assertEqual('',db_spec.primary_process_id)
self.assertEqual('',db_spec.primary_file_name)
data = {}
data['file'] = io.BytesIO(self.minimal_bpmn("abcdef")), 'my_new_file.bpmn'
# Add a new BPMN file to the specification
rv = self.app.post(f'/v1.0/workflow-specification/{db_spec.id}/file', data=data,
follow_redirects=True,
content_type='multipart/form-data', headers=self.logged_in_headers())
self.assert_success(rv)
# Make sure we now have a primary bpmn
db_spec = WorkflowSpecService().get_spec(spec.id)
self.assertEqual(db_spec.primary_process_id, '1')
self.assertEqual(db_spec.primary_file_name, 'my_new_file.bpmn')