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 add_fragment(sequence_diagram_name=None):
fragment_name = input('Insert the Fragment name: ')
if sequence_diagram_name != None:
diagram_name = sequence_diagram_name
sequence_diagram = None
else:
diagram_name = input('Insert the Sequence Diagram name: ')
print('Create the Sequence Diagram that is represented by the fragment ' + fragment_name)
sequence_diagram = create_sequence_diagram(diagram_name)
sequence_diagram.set_fragments(Fragment(name=fragment_name, represented_by=diagram_name))
sequence_diagram = sequence_diagram_menu(sequence_diagram)
fragment = Fragment(name=fragment_name, represented_by=diagram_name, sequence_diagram=sequence_diagram)
return fragment
def setUp(self):
self.fragment = Fragment()
