spiff-arena/tests/SpiffWorkflow/camunda/BusinessRuleTaskParserTest.py
Jon Herron 0892db6fa7 Squashed 'SpiffWorkflow/' content from commit 63db3e4
git-subtree-dir: SpiffWorkflow
git-subtree-split: 63db3e45947ec66b8d0efc2c74064004f8ff482c
2022-10-12 10:19:53 -04:00

56 lines
2.1 KiB
Python

import unittest
from unittest.mock import patch
from SpiffWorkflow.task import TaskState
from SpiffWorkflow.bpmn.workflow import BpmnWorkflow
from .BaseTestCase import BaseTestCase
class BusinessRuleTaskParserTest(BaseTestCase):
def setUp(self):
self.spec, subprocesses = self.load_workflow_spec(
'ExclusiveGatewayIfElseAndDecision.bpmn',
'Process_1',
'test_integer_decision.dmn')
self.workflow = BpmnWorkflow(self.spec)
def testDmnHappy(self):
self.workflow.get_tasks(TaskState.READY)[0].set_data(x=3)
self.workflow.do_engine_steps()
self.assertDictEqual(self.workflow.data, {'x': 3, 'y': 'A'})
self.assertDictEqual(self.workflow.last_task.data, {'x': 3, 'y': 'A'})
def testDmnSaveRestore(self):
self.workflow.get_tasks(TaskState.READY)[0].set_data(x=3)
self.save_restore()
self.workflow.do_engine_steps()
self.save_restore()
self.assertDictEqual(self.workflow.data, {'x': 3, 'y': 'A'})
self.assertDictEqual(self.workflow.last_task.data, {'x': 3, 'y': 'A'})
@patch('SpiffWorkflow.dmn.engine.DMNEngine.DMNEngine.evaluate')
def testDmnExecHasAccessToTask(self, mock_engine):
"""At one time, the Execute and Evaluate methods received a Task object
but the DMN evaluate method did not get a task object. While this is
an optional argument, it should always exist if executed in the context
of a BPMNWorkflow"""
self.workflow.get_tasks(TaskState.READY)[0].set_data(x=3)
self.workflow.do_engine_steps()
task = self.workflow.get_tasks_from_spec_name('TaskDecision')[0]
name, args, kwargs = mock_engine.mock_calls[0]
self.assertIn(task, args)
def testDmnUsesSameScriptEngineAsBPMN(self):
self.workflow.get_tasks(TaskState.READY)[0].set_data(x=3)
self.workflow.do_engine_steps()
def suite():
return unittest.TestLoader().loadTestsFromTestCase(BusinessRuleTaskParserTest)
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=2).run(suite())