spiff-arena/tests/SpiffWorkflow/bpmn/ResetSubProcessTest.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

85 lines
3.4 KiB
Python

# -*- coding: utf-8 -*-
import unittest
from SpiffWorkflow.bpmn.workflow import BpmnWorkflow
from tests.SpiffWorkflow.bpmn.BpmnWorkflowTestCase import BpmnWorkflowTestCase
__author__ = 'matth'
class ResetSubProcessTest(BpmnWorkflowTestCase):
"""Assure we can reset a token to a previous task when we have
a sub-workflow."""
def setUp(self):
spec, subprocesses = self.load_workflow_spec('resetworkflowA-*.bpmn', 'TopLevel')
self.workflow = BpmnWorkflow(spec, subprocesses)
def reload_save_restore(self):
spec, subprocesses = self.load_workflow_spec('resetworkflowB-*.bpmn', 'TopLevel')
self.workflow = BpmnWorkflow(spec, subprocesses)
# Save and restore the workflow, without including the spec.
# When loading the spec, use a slightly different spec.
self.workflow.do_engine_steps()
state = self.serializer.serialize_json(self.workflow)
self.workflow = self.serializer.deserialize_json(state)
self.workflow.spec = spec
self.workflow.subprocesses = subprocesses
def testSaveRestore(self):
self.actualTest(True)
def testResetToOuterWorkflowWhileInSubWorkflow(self):
self.workflow.do_engine_steps()
top_level_task = self.workflow.get_ready_user_tasks()[0]
self.workflow.complete_task_from_id(top_level_task.id)
self.workflow.do_engine_steps()
task = self.workflow.get_ready_user_tasks()[0]
self.save_restore()
top_level_task = self.workflow.get_tasks_from_spec_name('Task1')[0]
top_level_task.reset_token({}, reset_data=True)
task = self.workflow.get_ready_user_tasks()[0]
self.assertEqual(task.get_name(), 'Task1')
def actualTest(self, save_restore=False):
self.workflow.do_engine_steps()
self.assertEqual(1, len(self.workflow.get_ready_user_tasks()))
task = self.workflow.get_ready_user_tasks()[0]
self.workflow.complete_task_from_id(task.id)
self.workflow.do_engine_steps()
task = self.workflow.get_ready_user_tasks()[0]
self.assertEqual(task.get_name(),'SubTask2')
self.workflow.complete_task_from_id(task.id)
self.workflow.do_engine_steps()
task = self.workflow.get_tasks_from_spec_name('Task1')[0]
task.reset_token(self.workflow.last_task.data)
self.workflow.do_engine_steps()
self.reload_save_restore()
task = self.workflow.get_ready_user_tasks()[0]
self.assertEqual(task.get_name(),'Task1')
self.workflow.complete_task_from_id(task.id)
self.workflow.do_engine_steps()
task = self.workflow.get_ready_user_tasks()[0]
self.assertEqual(task.get_name(),'Subtask2')
self.workflow.complete_task_from_id(task.id)
self.workflow.do_engine_steps()
task = self.workflow.get_ready_user_tasks()[0]
self.assertEqual(task.get_name(),'Subtask2A')
self.workflow.complete_task_from_id(task.id)
self.workflow.do_engine_steps()
task = self.workflow.get_ready_user_tasks()[0]
self.assertEqual(task.get_name(),'Task2')
self.workflow.complete_task_from_id(task.id)
self.workflow.do_engine_steps()
self.assertTrue(self.workflow.is_completed())
def suite():
return unittest.TestLoader().loadTestsFromTestCase(ResetSubProcessTest)
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=2).run(suite())