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()