def setUp(self):
self.util = Util()
from utils.utils import Util
util = Util()
class SequenceDiagramElement():
def __init__(self, nome=''):
self.nome = nome
def __eq__(self, sequence_diagram_element): # pragma: no cover
return self.nome == sequence_diagram_element.nome
def __str__(self): # pragma: no cover
return 'nome: {}\n'.format(self.nome)
def set_nome(self, nome):
self.nome = nome
def get_nome(self):
return self.nome
def dispose(self):
self.nome = ""
class Fragment(SequenceDiagramElement):
def __init__(self, nome='', represented_by=None):
super().__init__(nome)
self.represented_by = represented_by
def __eq__(self, fragment): # pragma: no cover
return self.nome == fragment.nome and \
class TestMessage(unittest.TestCase):
def setUp(self):
self.util = Util()
@parameterized.expand([
[{'A1': ActivityDiagramElement(name='A1', element_type=Util().START_NODE),
'A2': ActivityDiagramElement(name='A2', element_type=Util().TRANSITION_NODE)}],
[{'B1':ActivityDiagramElement(name='B1', element_type=Util().START_NODE), \
'B2': ActivityDiagramElement(name='B2', element_type=Util().DECISION_NODE), \
'B3': ActivityDiagramElement(name='B3', element_type=Util().MERGE_NODE)}],
[{'C1': ActivityDiagramElement(name='C1', element_type=Util().START_NODE)}],
])
def test_check_start_node_existence_true(self, nodes):
response = self.util.check_start_node_existence(nodes, None)
self.assertTrue(response)
@parameterized.expand([
[{'A1': ActivityDiagramElement(name='A1', element_type=Util().END_NODE), \
'A2': ActivityDiagramElement(name='A2', element_type=Util().TRANSITION_NODE)}],
[{'B1':ActivityDiagramElement(name='B1', element_type=Util().MERGE_NODE), \
'B2': ActivityDiagramElement(name='B2', element_type=Util().DECISION_NODE), \
'B3': ActivityDiagramElement(name='B3', element_type=Util().MERGE_NODE)}],
[{'C1': ActivityDiagramElement(name='C1', element_type=Util().ACTIVITY_NODE)}],
])
def test_check_start_node_existence_false(self, nodes):
with self.assertRaises(OrderError):
response = self.util.check_start_node_existence(nodes, None)
@parameterized.expand([
[{'A1': ActivityDiagramElement(name='A1', element_type=Util().START_NODE), \
'A2': ActivityDiagramElement(name='A2', element_type=Util().DECISION_NODE)}],
[{'B1':ActivityDiagramElement(name='B1', element_type=Util().START_NODE), \
'B2': ActivityDiagramElement(name='B2', element_type=Util().DECISION_NODE), \
'B3': ActivityDiagramElement(name='B3', element_type=Util().DECISION_NODE)}],
[{'C1': ActivityDiagramElement(name='C1', element_type=Util().DECISION_NODE)}],
])
def test_check_join_possibility(self, nodes):
response = self.util.check_join_possibility(nodes)
self.assertTrue(response)
@parameterized.expand([
[{'A1': ActivityDiagramElement(name='A1', element_type=Util().START_NODE), \
'A2': ActivityDiagramElement(name='A2', element_type=Util().MERGE_NODE)}],
[{'B1':ActivityDiagramElement(name='B1', element_type=Util().START_NODE), \
'B2': ActivityDiagramElement(name='B2', element_type=Util().MERGE_NODE), \
'B3': ActivityDiagramElement(name='B3', element_type=Util().MERGE_NODE)}],
[{'C1': ActivityDiagramElement(name='C1', element_type=Util().START_NODE)}],
])
def test_check_join_possibility_false(self, nodes):
with self.assertRaises(OrderError):
response = self.util.check_join_possibility(nodes)
@parameterized.expand([
[False, False, False, ('<ActivityDiagram name="ActivityDiagram1">\n'
' <ActivityDiagramElements>\n'
' <StartNode name="StartNode1"/>\n'
' <DecisionNode name="DecisionNode1"/>\n'
' <MergeNode name="MergeNode1"/>\n'
' <EndNode name="EndNode1"/>\n'
' </ActivityDiagramElements>\n'
' <ActivityDiagramTransitions>\n'
' <Transition name="FirstTr" prob="0.5"/>\n'
' <Transition name="Tr2" prob="0.5"/>\n'
' <Transition name="LastTr" prob="0.5"/>\n'
' </ActivityDiagramTransitions>\n'
'</ActivityDiagram>\n')],
[True, False, False, ('<ActivityDiagram name="ActivityDiagram1">\n'
' <ActivityDiagramElements>\n'
' <StartNode name="StartNode1"/>\n'
' <DecisionNode name="DecisionNode1"/>\n'
' <MergeNode name="MergeNode1"/>\n'
' <Activity name="ActivityNode1"/>\n'
' <EndNode name="EndNode1"/>\n'
' </ActivityDiagramElements>\n'
' <ActivityDiagramTransitions>\n'
' <Transition name="FirstTr" prob="0.5"/>\n'
' <Transition name="Tr2" prob="0.5"/>\n'
' <Transition name="Tr3" prob="0.5"/>\n'
' <Transition name="LastTr" prob="0.5"/>\n'
' </ActivityDiagramTransitions>\n'
'</ActivityDiagram>\n'
'<SequenceDiagrams>\n'
' <Lifelines>\n'
' <Lifeline name="LifeLineX"/>\n'
' </Lifelines>\n'
'</SequenceDiagrams>\n')],
[True, True, False, ('<ActivityDiagram name="ActivityDiagram1">\n'
' <ActivityDiagramElements>\n'
' <StartNode name="StartNode1"/>\n'
' <DecisionNode name="DecisionNode1"/>\n'
' <MergeNode name="MergeNode1"/>\n'
' <Activity name="ActivityNode1"/>\n'
' <EndNode name="EndNode1"/>\n'
' </ActivityDiagramElements>\n'
' <ActivityDiagramTransitions>\n'
' <Transition name="FirstTr" prob="0.5"/>\n'
' <Transition name="Tr2" prob="0.5"/>\n'
' <Transition name="Tr3" prob="0.5"/>\n'
' <Transition name="LastTr" prob="0.5"/>\n'
' </ActivityDiagramTransitions>\n'
'</ActivityDiagram>\n'
'<SequenceDiagrams>\n'
' <Lifelines>\n'
' <Lifeline name="LifeLineX"/>\n'
' </Lifelines>\n'
' <Fragments>\n'
' <Optional name="Fragment1" representedBy="SequenceDiagram"/>\n'
' </Fragments>\n'
' <SequenceDiagram name="SequenceDiagram">\n'
' <Message name="MensagemX" prob="0.5" source="LifeLineX" target="LifeLineX"/>\n'
' <Fragment name="Fragment1"/>\n'
' </SequenceDiagram>\n'
'</SequenceDiagrams>\n')],
[True, True, True, ('<ActivityDiagram name="ActivityDiagram1">\n'
' <ActivityDiagramElements>\n'
' <StartNode name="StartNode1"/>\n'
' <DecisionNode name="DecisionNode1"/>\n'
' <MergeNode name="MergeNode1"/>\n'
' <Activity name="ActivityNode1"/>\n'
' <EndNode name="EndNode1"/>\n'
' </ActivityDiagramElements>\n'
' <ActivityDiagramTransitions>\n'
' <Transition name="FirstTr" prob="0.5"/>\n'
' <Transition name="Tr2" prob="0.5"/>\n'
' <Transition name="Tr3" prob="0.5"/>\n'
' <Transition name="LastTr" prob="0.5"/>\n'
' </ActivityDiagramTransitions>\n'
'</ActivityDiagram>\n'
'<SequenceDiagrams>\n'
' <Lifelines>\n'
' <Lifeline name="LifeLineX"/>\n'
' </Lifelines>\n'
' <Fragments>\n'
' <Optional name="Fragment1" representedBy="SequenceDiagram"/>\n'
' <Optional name="Fragment2" representedBy="SequenceDiagram"/>\n'
' </Fragments>\n'
' <SequenceDiagram name="SequenceDiagram">\n'
' <Message name="MensagemX" prob="0.5" source="LifeLineX" target="LifeLineX"/>\n'
' <Fragment name="Fragment1"/>\n'
' </SequenceDiagram>\n'
' <SequenceDiagram name="SequenceDiagram">\n'
' <Message name="MensagemX" prob="0.5" source="LifeLineX" target="LifeLineX"/>\n'
' <Fragment name="Fragment2"/>\n'
' </SequenceDiagram>\n'
'</SequenceDiagrams>\n')],
])
def test_generate_diagram(self, has_activity, has_fragment,
has_multiple_fragments, result_diagram):
activity_diagram = self.mock_activity_diagram(has_activity, has_fragment,
has_multiple_fragments)
self.util.generate_diagram(activity_diagram)
diagram_file = open(f'xmls/{activity_diagram.name}.xml', 'r')
diagram_from_file = diagram_file.read()
self.assertEqual(diagram_from_file, result_diagram)
diagram_file.close()
os.remove(f'xmls/{activity_diagram.name}.xml')
def mock_activity_diagram(self, has_activity=False, has_fragment=False, has_multiple_fragments=False):
activity_diagram = ActivityDiagram(name='ActivityDiagram1')
# Start Node
start_node = ActivityDiagramElement(name='StartNode1', element_type=self.util.START_NODE)
activity_diagram.set_elements(start_node)
activity_diagram.set_start_node(start_node)
# Decision Node
decision_node = ActivityDiagramElement(name='DecisionNode1',
element_type=self.util.DECISION_NODE)
activity_diagram.set_elements(decision_node)
activity_diagram.set_transitions(self.create_transition(name='FirstTr', source_node=start_node,
target_node=decision_node))
# Merge Node
merge_node = ActivityDiagramElement(name='MergeNode1', element_type=self.util.MERGE_NODE)
activity_diagram.set_elements(merge_node)
activity_diagram.set_transitions(self.create_transition(name='Tr2', source_node=decision_node,
target_node=merge_node))
last_node = merge_node
if has_activity:
activity_node = ActivityDiagramElement(name='ActivityNode1', element_type=self.util.ACTIVITY_NODE)
activity_diagram.set_elements(activity_node)
activity_diagram.set_transitions(self.create_transition(name='Tr3',
source_node=merge_node,
target_node=activity_node))
last_node = activity_node
sequence_diagram = self.mock_sequence_diagram(has_fragment, has_multiple_fragments)
activity_diagram.set_sequence_diagrams(sequence_diagram)
# EndNode
end_node = ActivityDiagramElement(name='EndNode1', element_type=self.util.END_NODE)
activity_diagram.set_elements(end_node)
activity_diagram.set_transitions(self.create_transition(name='LastTr', source_node=last_node,
target_node=end_node))
return activity_diagram
def mock_sequence_diagram(self, has_fragment, has_multiple_fragments):
sequence_diagram = SequenceDiagram(name='SequenceDiagram', guard_condition=True)
# LifeLines
lifelines = {0 : Lifeline(id=0, name='LifeLineX')}
sequence_diagram.set_life_lines(lifelines)
# Message
message = Message(name='MensagemX', source=lifelines[0],
target=lifelines[0], prob=0.5,
message_type='Synchronous')
sequence_diagram.set_messages(message)
if has_fragment:
fragment = Fragment(name='Fragment1',
represented_by=sequence_diagram.name,
sequence_diagram=sequence_diagram)
sequence_diagram.set_fragments(fragment)
if has_multiple_fragments:
fragment = Fragment(name='Fragment2',
represented_by=sequence_diagram.name,
sequence_diagram=self.mock_sequence_diagram(False, False))
sequence_diagram.set_fragments(fragment)
if has_fragment:
sequence_diagram.get_fragments()[0].sequence_diagram = sequence_diagram
return sequence_diagram
def create_transition(self, name, source_node, target_node):
return Transition(name=name,
prob=0.5,
source=source_node,
target=target_node,
element_type=self.util.TRANSITION_NODE)