def test_all_target(self):
targets = [Obj(), Obj(), Obj()]
flow1 = SequenceFlow('1', Obj(), targets[0])
flow2 = SequenceFlow('4', Obj(), targets[1])
flow3 = SequenceFlow('7', Obj(), targets[2])
flows = [flow1, flow2, flow3]
collection = SequenceFlowCollection(*flows)
self.assertEqual(targets, collection.all_target_node())
def test_all_source(self):
sources = [Obj(), Obj(), Obj()]
flow1 = SequenceFlow('1', sources[0], Obj())
flow2 = SequenceFlow('4', sources[1], Obj())
flow3 = SequenceFlow('7', sources[2], Obj())
flows = [flow1, flow2, flow3]
collection = SequenceFlowCollection(*flows)
self.assertEqual(sources, collection.all_source_node())
def test_get_flow(self):
flow1 = SequenceFlow('1', Obj(), Obj())
flow2 = SequenceFlow('4', Obj(), Obj())
flow3 = SequenceFlow('7', Obj(), Obj())
flows = [flow1, flow2, flow3]
collection = SequenceFlowCollection(*flows)
self.assertEqual(flow1, collection.get_flow(flow1.id))
self.assertEqual(flow2, collection.get_flow(flow2.id))
self.assertEqual(flow3, collection.get_flow(flow3.id))
def test_unique_one(self):
flow1 = SequenceFlow('1', Obj(), Obj())
flow2 = SequenceFlow('4', Obj(), Obj())
flow3 = SequenceFlow('7', Obj(), Obj())
flows = [flow1, flow2, flow3]
not_unique_collection = SequenceFlowCollection(*flows)
unique_collection = SequenceFlowCollection(flow1)
self.assertEqual(flow1, unique_collection.unique_one())
self.assertRaises(InvalidOperationException, not_unique_collection.unique_one)
def test_add_flow(self):
flow1 = SequenceFlow('1', Obj(), Obj())
flow2 = SequenceFlow('4', Obj(), Obj())
flow3 = SequenceFlow('7', Obj(), Obj())
flow4 = SequenceFlow('10', Obj(), Obj())
flows = [flow1, flow2, flow3]
flows_after_added = [flow1, flow2, flow3, flow4]
flow_dict_after_added = {
flow1.id: flow1,
flow2.id: flow2,
flow3.id: flow3,
flow4.id: flow4
}
collection = SequenceFlowCollection(*flows)
collection.add_flow(flow4)
self.assertEqual(flows_after_added, collection.flows)
self.assertEqual(flow_dict_after_added, collection.flow_dict)
def test_sequence_flow_collection(self):
flow1 = SequenceFlow('1', Obj(), Obj())
flow2 = SequenceFlow('4', Obj(), Obj())
flow3 = SequenceFlow('7', Obj(), Obj())
flows = [flow1, flow2, flow3]
flow_dict = {flow1.id: flow1, flow2.id: flow2, flow3.id: flow3}
collection = SequenceFlowCollection(*flows)
self.assertEqual(flows, collection.flows)
self.assertEqual(flow_dict, collection.flow_dict)
def test_is_empty(self):
flow1 = SequenceFlow('1', Obj(), Obj())
flow2 = SequenceFlow('4', Obj(), Obj())
flow3 = SequenceFlow('7', Obj(), Obj())
flows = [flow1, flow2, flow3]
not_empty_collection = SequenceFlowCollection(*flows)
empty_collection = SequenceFlowCollection()
self.assertTrue(empty_collection.is_empty())
self.assertFalse(not_empty_collection.is_empty())
