mirror of
https://github.com/sartography/spiff-arena.git
synced 2025-01-15 20:14:56 +00:00
c237e218b2
0e61be85 Merge pull request #289 from sartography/improvement/execution-and-serialization-cleanup 527684da fix some typos in the class & method docs 0dff44a4 Merge branch 'main' into improvement/execution-and-serialization-cleanup 64737498 Allow for other PythonScriptEngine environments besides task data (#288) dd63e916 remove some unused tests & diagrams 24aae519 clean up various small stuff 3b2dc35d use context when opening files for parsing 69eec3eb update class/method docs 24528dfb move all spec conversion classes to top level 5af33b11 remove some unused methods related to old serializer 931b90fb reorganize serializer 4e81ed29 consolidate pointless serializer classes d62acf02 change task_spec._update_hook to return a boolean indicating whether the task is ready git-subtree-dir: SpiffWorkflow git-subtree-split: 0e61be85c47474a33037e6f398e64c96e02f13ad
60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
import unittest
|
|
|
|
from SpiffWorkflow.exceptions import WorkflowTaskException
|
|
from SpiffWorkflow.task import TaskState
|
|
from SpiffWorkflow.bpmn.PythonScriptEngine import PythonScriptEngine
|
|
from SpiffWorkflow.bpmn.PythonScriptEngineEnvironment import TaskDataEnvironment
|
|
from SpiffWorkflow.bpmn.workflow import BpmnWorkflow
|
|
from tests.SpiffWorkflow.bpmn.BpmnWorkflowTestCase import BpmnWorkflowTestCase
|
|
|
|
__author__ = 'McDonald, danfunk'
|
|
|
|
def my_custom_function(txt):
|
|
return str(txt).upper()
|
|
|
|
class CustomBpmnScriptEngine(PythonScriptEngine):
|
|
"""This is a custom script processor that can be easily injected into Spiff Workflow.
|
|
It will execute python code read in from the bpmn. It will also make any scripts in the
|
|
scripts directory available for execution. """
|
|
def __init__(self):
|
|
environment = TaskDataEnvironment({'custom_function': my_custom_function})
|
|
super().__init__(environment=environment)
|
|
|
|
|
|
class CustomInlineScriptTest(BpmnWorkflowTestCase):
|
|
|
|
def setUp(self):
|
|
spec, subprocesses = self.load_workflow_spec('custom_function_test*', 'top_workflow')
|
|
script_engine = CustomBpmnScriptEngine()
|
|
self.workflow = BpmnWorkflow(spec, subprocesses, script_engine=script_engine)
|
|
|
|
def testRunThroughHappy(self):
|
|
self.actual_test(save_restore=False)
|
|
|
|
def testRunThroughSaveRestore(self):
|
|
self.actual_test(save_restore=False)
|
|
|
|
def actual_test(self, save_restore):
|
|
if save_restore: self.save_restore()
|
|
self.workflow.do_engine_steps()
|
|
if save_restore: self.save_restore()
|
|
data = self.workflow.last_task.data
|
|
self.assertEqual(data['c1'], 'HELLO')
|
|
self.assertEqual(data['c2'], 'GOODBYE')
|
|
self.assertEqual(data['c3'], 'ARRIVEDERCI')
|
|
|
|
def test_overwrite_function_with_local_variable(self):
|
|
ready_task = self.workflow.get_tasks(TaskState.READY)[0]
|
|
ready_task.data = {'custom_function': "bill"}
|
|
with self.assertRaises(WorkflowTaskException) as e:
|
|
self.workflow.do_engine_steps()
|
|
self.assertTrue('' in str(e.exception))
|
|
self.assertTrue('custom_function' in str(e.exception))
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromTestCase(CustomInlineScriptTest)
|
|
if __name__ == '__main__':
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|