def setUp(self):
ex_gateway1 = ExclusiveGateway(id='1')
next_node1 = ParallelGateway(id='1', converge_gateway_id='cvg')
next_node2 = ParallelGateway(id='2', converge_gateway_id='cvg')
flow1 = SequenceFlow('flow1', ex_gateway1, next_node1)
flow2 = SequenceFlow('flow2', ex_gateway1, next_node2)
condition1 = Condition('a == 1', flow1)
condition2 = Condition('a != 1', flow2)
ex_gateway1.add_condition(condition1)
ex_gateway1.add_condition(condition2)
ex_gateway1.outgoing.add_flow(flow1)
ex_gateway1.outgoing.add_flow(flow2)
next_node1.incoming.add_flow(flow1)
next_node2.incoming.add_flow(flow2)
self.gateway_for_test_determine = ex_gateway1
ex_gateway2 = ExclusiveGateway(id='2')
next_node3 = ParallelGateway(id='3', converge_gateway_id='cvg')
next_node4 = ParallelGateway(id='4', converge_gateway_id='cvg')
next_node5 = ParallelGateway(id='5', converge_gateway_id='cvg')
flow3 = SequenceFlow('flow3', ex_gateway2, next_node3)
flow4 = SequenceFlow('flow4', ex_gateway2, next_node4)
flow5 = SequenceFlow('flow5', ex_gateway2, next_node5, is_default=True)
condition3 = Condition('a == 1', flow3)
condition4 = Condition('a != 1', flow4)
ex_gateway2.add_condition(condition3)
ex_gateway2.add_condition(condition4)
ex_gateway2.outgoing.add_flow(flow3)
ex_gateway2.outgoing.add_flow(flow4)
ex_gateway2.outgoing.add_flow(flow5)
next_node3.incoming.add_flow(flow3)
next_node4.incoming.add_flow(flow4)
next_node5.incoming.add_flow(flow5)
self.gateway_for_test_next = ex_gateway2
def get_exclusive_gateway_pipeline(result):
start_event_a_id = node_uniqid()
act_b_id = node_uniqid()
gateway_c_id = node_uniqid()
act_d_id = node_uniqid()
act_e_id = node_uniqid()
act_f_id = node_uniqid()
gateway_g_id = node_uniqid()
end_event_h_id = node_uniqid()
start_event_a = EmptyStartEvent(start_event_a_id)
b_act = ServiceActivity(act_b_id,
service=EchoService(),
data=DataObject({'a': 3}))
c_gateway = ExclusiveGateway(gateway_c_id, gateway_g_id)
d_act = ServiceActivity(act_d_id,
service=LogService(),
data=DataObject({'node': 'd'}))
e_act = ServiceActivity(act_e_id,
service=EchoService(),
data=DataObject({'node': 'e'}))
f_act = ServiceActivity(act_f_id,
service=TestService(),
data=DataObject({'node': 'f'}))
g_gateway = ConvergeGateway(gateway_g_id)
end_event_h = EmptyEndEvent(end_event_h_id)
ab_flow = SequenceFlow('ab', start_event_a, b_act)
bc_flow = SequenceFlow('bc', b_act, c_gateway)
cd_flow = SequenceFlow('cd', c_gateway, d_act)
ce_flow = SequenceFlow('ce', c_gateway, e_act)
cf_flow = SequenceFlow('cf', c_gateway, f_act)
dg_flow = SequenceFlow('dg', d_act, g_gateway)
eg_flow = SequenceFlow('eg', e_act, g_gateway)
fg_flow = SequenceFlow('fg', f_act, g_gateway)
gh_flow = SequenceFlow('gh', g_gateway, end_event_h)
start_event_a.outgoing.add_flow(ab_flow)
b_act.incoming.add_flow(ab_flow)
b_act.outgoing.add_flow(bc_flow)
c_gateway.incoming.add_flow(bc_flow)
c_gateway.outgoing.add_flow(cd_flow)
c_gateway.outgoing.add_flow(ce_flow)
c_gateway.outgoing.add_flow(cf_flow)
d_act.incoming.add_flow(cd_flow)
d_act.outgoing.add_flow(dg_flow)
e_act.incoming.add_flow(ce_flow)
e_act.outgoing.add_flow(eg_flow)
f_act.incoming.add_flow(cf_flow)
f_act.outgoing.add_flow(fg_flow)
g_gateway.incoming.add_flow(dg_flow)
g_gateway.incoming.add_flow(eg_flow)
g_gateway.incoming.add_flow(fg_flow)
g_gateway.outgoing.add_flow(gh_flow)
end_event_h.incoming.add_flow(gh_flow)
c_gateway.add_condition(Condition('result == 1', cd_flow))
c_gateway.add_condition(Condition('result == 2', ce_flow))
c_gateway.add_condition(Condition('result == 3', cf_flow))
spec = PipelineSpec(start_event_a,
end_event_h, [
ab_flow, bc_flow, cd_flow, ce_flow, cf_flow,
dg_flow, eg_flow, fg_flow, gh_flow
], [b_act, d_act, e_act, f_act],
[c_gateway, g_gateway],
data=DataObject({}),
context=context.Context(act_outputs={},
scope={'result': result}))
return Pipeline(node_uniqid(), spec)
def test_add_condition(self):
ex_gateway = ExclusiveGateway(id='1')
flow1 = SequenceFlow('flow1', ex_gateway, None)
self.assertEqual([], ex_gateway.conditions)
ex_gateway.add_condition(flow1)
self.assertEqual([flow1], ex_gateway.conditions)