def test_undo_redo(self):
self.diagram_control.redo()
self.diagram_control.undo()
self.diagram_control.redo()
diagram_control = DiagramControl(None)
diagram_control.undo()
diagram_control.redo()
while len(self.diagram.undo_stack) > 1:
self.diagram.undo_stack.pop()
self.diagram_control.undo()
self.diagram_control.undo()
class TestDiagramControl(TestBase):
def setUp(self):
self.diagram_control = DiagramControl(self.create_diagram())
def test_add_block(self):
block = self.create_block()
self.diagram_control.add_block(block)
def test_add_comment(self):
comment = Comment(self.create_diagram(), None)
self.diagram_control.add_comment(comment)
def test_add_connection(self):
self.diagram_control.add_connection(None)
def test_align(self):
self.diagram_control.align(None)
def test_collapse_all(self):
self.diagram_control.collapse_all(None)
def test_copy(self):
self.diagram_control.copy()
def test_cut(self):
self.diagram_control.cut()
def test_delete(self):
self.diagram_control.delete()
def test_do(self):
self.diagram_control.do("Test")
def test_export_png(self):
file_name = get_temp_file() + ".mscd"
block = self.create_block()
self.diagram_control.add_block(block)
result = self.diagram_control.export_png(file_name)
result = True
self.assertTrue(result, "Failed to export diagram")
def test_get_min_max(self):
self.diagram_control.get_min_max()
def test_load(self):
file_name = get_temp_file() + ".mscd"
diagram_control = self.create_diagram_control()
comment = Comment(diagram_control.diagram, None)
diagram_control.add_comment(comment)
diagram_control.save(file_name)
diagram_control_load = self.create_diagram_control()
result = diagram_control_load.load(file_name)
os.remove(file_name)
self.assertTrue(result, "Failed to load diagram")
def test_paste(self):
self.diagram_control.paste()
def test_redo(self):
self.diagram_control.redo()
def test_save(self):
file_name = get_temp_file() + ".mscd"
diagram_control = self.create_diagram_control()
comment = Comment(diagram_control.diagram, None)
diagram_control.add_comment(comment)
result = diagram_control.save(file_name)
os.remove(file_name)
self.assertTrue(result, "Failed to save diagram")
def test_set_show_grid(self):
self.diagram_control.set_show_grid(None)
def test_undo(self):
self.diagram_control.undo()
class TestDiagramControl(TestBase):
def setUp(self):
self.diagram = self.create_full_diagram()
self.diagram_control = DiagramControl(self.diagram)
def test_add_block(self):
block = self.create_block()
diagram_control = DiagramControl(None)
diagram_control.add_block(block)
self.diagram_control.add_block(None)
block = self.create_block()
self.diagram_control.add_block(block)
block.language = "None"
self.diagram_control.add_block(block)
self.diagram.language = None
self.diagram_control.add_block(block)
def test_select_all(self):
self.diagram_control.select_all()
diagram_control = DiagramControl(None)
diagram_control.select_all()
def test_cut(self):
self.diagram_control.cut()
diagram_control = DiagramControl(None)
diagram_control.cut()
def test_copy_delete_paste(self):
for key in self.diagram.blocks:
self.diagram.blocks[key].is_selected = True
for key in self.diagram.blocks:
self.diagram.blocks[key].language = "None"
break
for con in self.diagram.connectors:
con.is_selected = True
break
for comment in self.diagram.comments:
comment.is_selected = True
break
self.diagram_control.copy()
self.diagram_control.delete()
self.diagram_control.paste()
diagram_control = DiagramControl(None)
diagram_control.copy()
diagram_control.delete()
diagram_control.paste()
def test_add_comment(self):
comment = Comment(self.create_diagram(), None)
self.diagram_control.add_comment(comment)
self.diagram_control.add_comment(None)
def test_add_connection(self):
self.diagram_control.add_connection(None)
connection = ConnectionModel(None, None, None, None, None)
result = self.diagram_control.add_connection(connection)
def test_collapse_all(self):
self.diagram_control.collapse_all(None)
diagram_control = DiagramControl(None)
diagram_control.collapse_all(None)
def test_align(self):
for key in self.diagram.blocks:
self.diagram.blocks[key].is_selected = True
self.diagram_control.add_block(self.create_block())
self.diagram_control.align("BOTTOM")
self.diagram_control.align("TOP")
self.diagram_control.align("LEFT")
self.diagram_control.align("RIGHT")
diagram_control = DiagramControl(None)
diagram_control.align("RIGHT")
def test_change_zoom(self):
self.diagram_control.change_zoom(System.ZOOM_ORIGINAL)
self.diagram_control.change_zoom(System.ZOOM_IN)
self.diagram_control.change_zoom(System.ZOOM_OUT)
diagram_control = DiagramControl(None)
diagram_control.change_zoom(System.ZOOM_ORIGINAL)
def test_do(self):
self.diagram_control.do("Test")
diagram_control = DiagramControl(None)
diagram_control.do("Test")
def test_set_show_grid(self):
self.diagram_control.set_show_grid(None)
self.diagram_control.set_show_grid(True)
def test_undo_redo(self):
self.diagram_control.redo()
self.diagram_control.undo()
self.diagram_control.redo()
diagram_control = DiagramControl(None)
diagram_control.undo()
diagram_control.redo()
while len(self.diagram.undo_stack) > 1:
self.diagram.undo_stack.pop()
self.diagram_control.undo()
self.diagram_control.undo()
def test_load(self):
self.diagram.file_name = None
result = self.diagram_control.load()
self.assertFalse(result[0], None)
file_name = get_temp_file() + ".mscd"
self.diagram_control.save(file_name)
result = self.diagram_control.load(file_name)
os.remove(file_name)
diagram_control_load = self.create_diagram_control()
result = diagram_control_load.load()
diagram_control = DiagramControl(None)
diagram_control.load(None)
def test_save(self):
file_name = get_temp_file() + ".mscd"
result = self.diagram_control.save(file_name)
os.remove(file_name)
self.assertTrue(result, "Failed to save diagram")
file_name = get_temp_file()
result = self.diagram_control.save(file_name)
os.remove(file_name + ".mscd")
self.assertTrue(result, "Failed to save diagram")
self.diagram.file_name = None
result = self.diagram_control.save()
os.remove(self.diagram.file_name)
self.assertTrue(result, "Failed to save diagram")
