mirror of
https://github.com/sartography/spiff-arena.git
synced 2025-01-18 21:41:04 +00:00
99 lines
4.1 KiB
Python
99 lines
4.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from SpiffWorkflow.task import TaskState
|
||
|
|
from SpiffWorkflow.bpmn.workflow import BpmnWorkflow
|
||
|
|
|
||
|
|
from tests.SpiffWorkflow.camunda.BaseTestCase import BaseTestCase
|
||
|
|
|
||
|
|
__author__ = 'kellym'
|
||
|
|
|
||
|
|
|
||
|
|
class NIMessageBoundaryTest(BaseTestCase):
|
||
|
|
"""
|
||
|
|
Non-Interrupting Timer boundary test
|
||
|
|
"""
|
||
|
|
def setUp(self):
|
||
|
|
spec, subprocesses = self.load_workflow_spec('noninterrupting-MessageBoundary.bpmn', 'MessageBoundary')
|
||
|
|
self.workflow = BpmnWorkflow(spec, subprocesses)
|
||
|
|
|
||
|
|
def testRunThroughHappy(self):
|
||
|
|
self.actual_test(save_restore=False)
|
||
|
|
|
||
|
|
def testThroughSaveRestore(self):
|
||
|
|
self.actual_test(save_restore=True)
|
||
|
|
|
||
|
|
|
||
|
|
def actual_test(self,save_restore = False):
|
||
|
|
|
||
|
|
ready_tasks = self.workflow.get_tasks(TaskState.READY)
|
||
|
|
self.assertEqual(1, len(ready_tasks))
|
||
|
|
self.workflow.complete_task_from_id(ready_tasks[0].id)
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
|
||
|
|
# first we run through a couple of steps where we answer No to each
|
||
|
|
# question
|
||
|
|
answers = {'Activity_WorkLate':('flag_task','No'),
|
||
|
|
'Activity_DoWork': ('work_done','No')}
|
||
|
|
for x in range(3):
|
||
|
|
ready_tasks = self.workflow.get_tasks(TaskState.READY)
|
||
|
|
for task in ready_tasks:
|
||
|
|
response = answers.get(task.task_spec.name,None)
|
||
|
|
self.assertEqual(response==None,
|
||
|
|
False,
|
||
|
|
'We got a ready task that we did not expect - %s'%(
|
||
|
|
task.task_spec.name))
|
||
|
|
task.data[response[0]] = response[1]
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
# if we have a list of tasks - that list becomes invalid
|
||
|
|
# after we do a save restore, so I'm completing the list
|
||
|
|
# before doing the save restore.
|
||
|
|
if save_restore:
|
||
|
|
self.save_restore()
|
||
|
|
|
||
|
|
|
||
|
|
answers = {'Activity_WorkLate':('flag_task','Yes'),
|
||
|
|
'Activity_DoWork': ('work_done','No'),
|
||
|
|
'Activity_WorkLateReason':('work_late_reason','covid-19')}
|
||
|
|
for x in range(3):
|
||
|
|
ready_tasks = self.workflow.get_tasks(TaskState.READY)
|
||
|
|
for task in ready_tasks:
|
||
|
|
response = answers.get(task.task_spec.name,None)
|
||
|
|
self.assertEqual(response==None,
|
||
|
|
False,
|
||
|
|
'We got a ready task that we did not expect - %s'%(
|
||
|
|
task.task_spec.name))
|
||
|
|
task.data[response[0]] = response[1]
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
|
||
|
|
ready_tasks = self.workflow.get_tasks(TaskState.READY)
|
||
|
|
self.assertEqual(len(ready_tasks),1)
|
||
|
|
task = ready_tasks[0]
|
||
|
|
self.assertEqual(task.task_spec.name,'Activity_DoWork')
|
||
|
|
task.data['work_done'] = 'Yes'
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
ready_tasks = self.workflow.get_tasks(TaskState.READY)
|
||
|
|
self.assertEqual(len(ready_tasks), 1)
|
||
|
|
task = ready_tasks[0]
|
||
|
|
self.assertEqual(task.task_spec.name, 'Activity_WorkCompleted')
|
||
|
|
task.data['work_completed'] = 'Lots of Stuff'
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
self.assertEqual(self.workflow.is_completed(),True)
|
||
|
|
self.assertEqual(self.workflow.last_task.data,{'Event_InterruptBoundary_Response': 'Youre late!',
|
||
|
|
'flag_task': 'Yes',
|
||
|
|
'work_done': 'Yes',
|
||
|
|
'work_completed': 'Lots of Stuff',
|
||
|
|
'work_late_reason': 'covid-19'})
|
||
|
|
|
||
|
|
|
||
|
|
def suite():
|
||
|
|
return unittest.TestLoader().loadTestsFromTestCase(NIMessageBoundaryTest)
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|