def test_delete(self): self.objects.delete(Student(3, "random", 7)) self.assertEqual(len(self.objects), 1) try: self.objects.delete(Student(5, "random", 7)) except RepositoryError: pass
def test_add(self): self.assertEqual(len(self.objects), 2) self.objects.save(Student(1, "Alex", 5)) self.assertEqual(len(self.objects), 3) try: self.objects.save(Student("dsc", "Alex", 5)) except StudentError: pass
def add_student(self, id, name, group): """ Adds a new student :param id: student id :param name: student name :param group: student group :return: nothing """ self.__repository.save(Student(id, name, group)) UndoManager.register_operation(self.__repository, UndoHandler.ADD_STUDENT, Student(id, name, group))
def test_update(self): self.objects.update(Student(3, "Alexandru", 7)) stud = self.objects.get_all() self.assertEqual(stud[0].name, "Alexandru") self.objects.update(Student(3, "Lukas", 8)) stud = self.objects.get_all() self.assertEqual(stud[0].group, 8) try: self.objects.update(Student(5, "Alexandru", 7)) except RepositoryError: pass try: self.objects.update(Student("tf", "Lukas", 3)) except StudentError: pass
def delete_student(self, id): """ Deletes a student :param id: the id of the student :return: nothing """ UndoManager.register_operation(self.__repository, UndoHandler.DELETE_STUDENT, self.find_student(id), "delete_student") self.__repository.delete(Student(id, "random", 0))
def update_student(self, id, arg=None, group=None): """ Updates the name and/or group of the student :param id: student id :param arg: new name/ new group :param group: new group :return: nothing """ student = self.find_student(id) try: arg = int(arg) if group is None: UndoManager.register_operation(self.__repository, UndoHandler.UPDATE_STUDENT, student, "update_student") self.__repository.update(Student(id, student.name, arg)) except ValueError: if group is None: UndoManager.register_operation(self.__repository, UndoHandler.UPDATE_STUDENT, student, "update_student") self.__repository.update(Student(id, arg, student.group)) else: UndoManager.register_operation(self.__repository, UndoHandler.UPDATE_STUDENT, student, "update_student") self.__repository.update(Student(id, arg, group))
def test_get(self): stud = self.students.get_all_students() self.assertEqual(len(stud), 2) self.assertEqual(stud[0], Student(4, "Lukas", 4))
def test_find(self): stud = self.students.find_student(4) self.assertEqual(stud, Student(4, "Lukas", 4)) stud = self.students.find_student(5) self.assertEqual(stud, None)
def test_get(self): list = self.grades.get_all_grades() self.assertEqual(len(list), 2) self.assertEqual(list[0], Grade(1, 4, 8.4, Student(4, "Lukas", 4)))
def setUp(self): self.objects = Repository(StudentValidator) self.objects.save(Student(3, "Lukas", 4)) self.objects.save(Student(2, "Victor", 6))
def test_get(self): stud = self.objects.get_all() self.assertEqual(len(stud), 2) self.assertEqual(stud[0], Student(3, "Lukas", 4))
def test_find(self): stud = self.objects.find_by_id(3) self.assertEqual(stud, Student(3, "Lukas", 4)) stud = self.objects.find_by_id(5) self.assertEqual(stud, None)