mirror of
https://github.com/status-im/spiff-arena.git
synced 2025-01-17 13:41:41 +00:00
223 lines
8.9 KiB
Python
223 lines
8.9 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from SpiffWorkflow.bpmn.workflow import BpmnWorkflow
|
||
|
|
from SpiffWorkflow.exceptions import WorkflowException
|
||
|
|
|
||
|
|
from tests.SpiffWorkflow.camunda.BaseTestCase import BaseTestCase
|
||
|
|
|
||
|
|
__author__ = 'matth'
|
||
|
|
|
||
|
|
|
||
|
|
class MultiInstanceArrayTest(BaseTestCase):
|
||
|
|
"""The example bpmn diagram tests both a set cardinality from user input
|
||
|
|
as well as looping over an existing array."""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
spec, subprocesses = self.load_workflow_spec('multi_instance_array.bpmn', 'MultiInstanceArray')
|
||
|
|
self.workflow = BpmnWorkflow(spec, subprocesses)
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
|
||
|
|
def testRunThroughHappy(self):
|
||
|
|
self.actual_test(False)
|
||
|
|
|
||
|
|
def testRunThroughSaveRestore(self):
|
||
|
|
self.actual_test(True)
|
||
|
|
|
||
|
|
|
||
|
|
def testRunThroughHappyList(self):
|
||
|
|
self.actual_test2(False)
|
||
|
|
|
||
|
|
def testRunThroughSaveRestoreList(self):
|
||
|
|
self.actual_test2(True)
|
||
|
|
|
||
|
|
def testRunThroughHappyDict(self):
|
||
|
|
self.actual_test_with_dict(False)
|
||
|
|
|
||
|
|
def testRunThroughSaveRestoreDict(self):
|
||
|
|
self.actual_test_with_dict(True)
|
||
|
|
|
||
|
|
def testGetTaskExtensions(self):
|
||
|
|
self.actual_test_for_extensions(False)
|
||
|
|
|
||
|
|
|
||
|
|
def actual_test(self, save_restore=False):
|
||
|
|
|
||
|
|
# Set initial array size to 3 in the first user form.
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
taskinfo = task.task_info()
|
||
|
|
self.assertEqual(taskinfo,{'is_looping':False,
|
||
|
|
'is_sequential_mi':False,
|
||
|
|
'is_parallel_mi':False,
|
||
|
|
'mi_count':0,
|
||
|
|
'mi_index':0})
|
||
|
|
self.assertEqual("Activity_FamSize", task.task_spec.name)
|
||
|
|
task.update_data({"Family": {"Size": 3}})
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
|
||
|
|
# Set the names of the 3 family members.
|
||
|
|
for i in range(3):
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
taskinfo = task.task_info()
|
||
|
|
self.assertEqual(taskinfo, {'is_looping': False,
|
||
|
|
'is_sequential_mi': True,
|
||
|
|
'is_parallel_mi': False,
|
||
|
|
'mi_count': 3,
|
||
|
|
'mi_index': i+1})
|
||
|
|
if i > 0:
|
||
|
|
self.assertEqual("FamilyMemberTask"+"_%d"%(i-1), task.task_spec.name)
|
||
|
|
else:
|
||
|
|
self.assertEqual("FamilyMemberTask", task.task_spec.name)
|
||
|
|
|
||
|
|
|
||
|
|
task.update_data({"FamilyMember": {"FirstName": "The Funk #%i" % i}})
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
|
||
|
|
self.assertEqual({'1': {'FirstName': 'The Funk #0'},
|
||
|
|
'2': {'FirstName': 'The Funk #1'},
|
||
|
|
'3': {'FirstName': 'The Funk #2'}},
|
||
|
|
task.data["Family"]["Members"])
|
||
|
|
#### NB - start here
|
||
|
|
### Data is not correctly getting to the next task upon complete of the last task
|
||
|
|
### after do_engine_steps, the next task in the list should be the same as task.data
|
||
|
|
### but it is not.
|
||
|
|
|
||
|
|
### invalid copy of data?? ## appears that parent is not hooked up correctly
|
||
|
|
|
||
|
|
# Set the birthdays of the 3 family members.
|
||
|
|
for i in range(3):
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
if i > 0:
|
||
|
|
self.assertEqual("FamilyMemberBday"+"_%d"%(i-1), task.task_spec.name)
|
||
|
|
else:
|
||
|
|
self.assertEqual("FamilyMemberBday", task.task_spec.name)
|
||
|
|
task.update_data({"CurrentFamilyMember": {"Birthdate": "10/0%i/1985" % i}})
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
self.assertTrue(self.workflow.is_completed())
|
||
|
|
|
||
|
|
self.assertEqual({'1': {'FirstName': 'The Funk #0', "Birthdate": "10/00/1985"},
|
||
|
|
'2': {'FirstName': 'The Funk #1', "Birthdate": "10/01/1985"},
|
||
|
|
'3': {'FirstName': 'The Funk #2', "Birthdate": "10/02/1985"}},
|
||
|
|
self.workflow.last_task.data["Family"]["Members"])
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def actual_test2(self, save_restore=False):
|
||
|
|
|
||
|
|
# Set initial array size to 3 in the first user form.
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
self.assertEqual("Activity_FamSize", task.task_spec.name)
|
||
|
|
task.update_data({"Family":{"Size": 3}})
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
|
||
|
|
# Set the names of the 3 family members.
|
||
|
|
for i in range(3):
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
if i > 0:
|
||
|
|
self.assertEqual("FamilyMemberTask"+"_%d"%(i-1), task.task_spec.name)
|
||
|
|
else:
|
||
|
|
self.assertEqual("FamilyMemberTask", task.task_spec.name)
|
||
|
|
task.update_data({"FamilyMember": {"FirstName": "The Funk #%i" % i}})
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
|
||
|
|
self.assertEqual({'1': {'FirstName': 'The Funk #0'},
|
||
|
|
'2': {'FirstName': 'The Funk #1'},
|
||
|
|
'3': {'FirstName': 'The Funk #2'}},
|
||
|
|
task.data["Family"]["Members"])
|
||
|
|
|
||
|
|
|
||
|
|
# Make sure that if we have a list as both input and output
|
||
|
|
# collection, that we raise an exception
|
||
|
|
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
task.data['Family']['Members'] = ['The Funk #0','The Funk #1','The Funk #2']
|
||
|
|
self.assertEqual("FamilyMemberBday", task.task_spec.name)
|
||
|
|
task.update_data(
|
||
|
|
{"CurrentFamilyMember": {"Birthdate": "10/0%i/1985" % i}})
|
||
|
|
with self.assertRaises(WorkflowException) as context:
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
|
||
|
|
|
||
|
|
def actual_test_with_dict(self, save_restore=False):
|
||
|
|
|
||
|
|
# Set initial array size to 3 in the first user form.
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
self.assertEqual("Activity_FamSize", task.task_spec.name)
|
||
|
|
task.update_data({"Family":{"Size": 3}})
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
|
||
|
|
# Set the names of the 3 family members.
|
||
|
|
for i in range(3):
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
if i > 0:
|
||
|
|
self.assertEqual("FamilyMemberTask"+"_%d"%(i-1), task.task_spec.name)
|
||
|
|
else:
|
||
|
|
self.assertEqual("FamilyMemberTask", task.task_spec.name)
|
||
|
|
task.update_data({"FamilyMember": {"FirstName": "The Funk #%i" % i}})
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
|
||
|
|
self.assertEqual({'1': {'FirstName': 'The Funk #0'},
|
||
|
|
'2': {'FirstName': 'The Funk #1'},
|
||
|
|
'3': {'FirstName': 'The Funk #2'}},
|
||
|
|
task.data["Family"]["Members"])
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
# Set the birthdays of the 3 family members.
|
||
|
|
for i in range(3):
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
if i == 0:
|
||
|
|
# Modify so that the dict keys are alpha rather than int
|
||
|
|
task.data["Family"]["Members"] = {
|
||
|
|
"a": {'FirstName': 'The Funk #0'},
|
||
|
|
"b": {'FirstName': 'The Funk #1'},
|
||
|
|
"c": {'FirstName': 'The Funk #2'}}
|
||
|
|
if (i > 0):
|
||
|
|
self.assertEqual("FamilyMemberBday"+"_%d"%(i-1), task.task_spec.name)
|
||
|
|
else:
|
||
|
|
self.assertEqual("FamilyMemberBday", task.task_spec.name)
|
||
|
|
task.update_data(
|
||
|
|
{"CurrentFamilyMember": {"Birthdate": "10/0%i/1985" % i}})
|
||
|
|
self.workflow.complete_task_from_id(task.id)
|
||
|
|
# if save_restore: self.save_restore()
|
||
|
|
|
||
|
|
self.workflow.do_engine_steps()
|
||
|
|
if save_restore: self.save_restore()
|
||
|
|
self.assertTrue(self.workflow.is_completed())
|
||
|
|
|
||
|
|
self.assertEqual({"a": {'FirstName': 'The Funk #0', "Birthdate": "10/00/1985"},
|
||
|
|
"b": {'FirstName': 'The Funk #1', "Birthdate": "10/01/1985"},
|
||
|
|
"c": {'FirstName': 'The Funk #2', "Birthdate": "10/02/1985"}},
|
||
|
|
self.workflow.last_task.data["Family"]["Members"])
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def actual_test_for_extensions(self, save_restore=False):
|
||
|
|
|
||
|
|
# Set initial array size to 3 in the first user form.
|
||
|
|
task = self.workflow.get_ready_user_tasks()[0]
|
||
|
|
self.assertEqual("Activity_FamSize", task.task_spec.name)
|
||
|
|
extensions = task.task_spec.extensions # assume bpmn
|
||
|
|
self.assertEqual(extensions,{'Test1':'Value1','Test2':'Value2'})
|
||
|
|
|
||
|
|
|
||
|
|
def suite():
|
||
|
|
return unittest.TestLoader().loadTestsFromTestCase(MultiInstanceArrayTest)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|