class TestUndoRedo(unittest.TestCase): def setUp(self): self.__undoRedo = UndoRedo() self.__students = Repository() self.__assignments = Repository() self.__grades = Repository() self.__ui = UI(self.__students, self.__assignments, self.__grades, self.__undoRedo) def tearDown(self): self.__undoRedo = None self.__ui = None self.__students = None self.__assignments = None self.__grades = None def test_undoRedo(self): command = 'add 1 Cozma Marin 908' self.__ui.add(command, True) assert len(self.__students) == 1 self.__ui.undo('undo') self.__ui.undo('undo') lst = self.__students.getAll() assert len(self.__students) == 0 self.__ui.redo('redo') self.__ui.redo('redo')
class TestControllerStudents(unittest.TestCase): def setUp(self): self.__repo=Repository() self.__groups={} self.__idlist=[] self.__assignmentDict={} self.__undoRedo=UndoRedo() self.__control=controllerStudent(self.__repo,self.__undoRedo) def tearDown(self): self.__repo=None self.__groups=None self.__idlist=None self.__assignmentDict=None self.__control=None self.__undoRedo=None def test_addStudent(self): x="add 23 Popescu Marin 915" self.__control.addStudent(x,self.__groups,self.__idlist,self.__assignmentDict) with self.assertRaises(ValueError): self.__control.addStudent(x,self.__groups,self.__idlist,self.__assignmentDict) lst=self.__repo.getAll() x="add 2 Popescu Maria 915" self.__control.addStudent(x,self.__groups,self.__idlist,self.__assignmentDict) assert str(lst[0])=="23, Popescu Marin, 915" def test_removeStudent(self): x="add 23 Popescu Marin 915" self.__control.addStudent(x,self.__groups,self.__idlist,self.__assignmentDict) x="add 45 Boc Emil 911" self.__control.addStudent(x,self.__groups,self.__idlist,self.__assignmentDict) x="remove student 23" self.__control.removeStudent(x,self.__groups) with self.assertRaises(ValueError): x='remove student 1000' self.__control.removeStudent(x,self.__groups) lst=self.__repo.getAll() assert str(lst[0])=='45, Boc Emil, 911' def test_updateStudent(self): x="add 45 Boc Emil 911" self.__control.addStudent(x,self.__groups,self.__idlist,self.__assignmentDict) x='add 48 Dorin Mihai 912' self.__control.addStudent(x,self.__groups,self.__idlist,self.__assignmentDict) x='update student 45 912' dictt={911:[45],912:[48]} self.__control.updateStudent(x,dictt) lst=self.__repo.getAll() assert str(lst[0])=='45, Boc Emil, 912' with self.assertRaises(ValueError): x='update student 100 912' self.__control.updateStudent(x,self.__groups) with self.assertRaises(ValueError): x='update student 100 789' self.__control.updateStudent(x, self.__groups) with self.assertRaises(ValueError): x='update student 45 912' self.__control.updateStudent(x, self.__groups)