class StudentTests(unittest.TestCase):
def setUp(self):
self.__student_repository = Repository()
self.__student_repository.add(Student(1, "Test Student", 917))
self.__student_repository.add(Student(2, "Student Test", 916))
self.addValidInput = Student(3, "Le Student", 916)
self.addInvalidInput = Student(1, "Le Student", "da")
self.removeValidInput = Student(1, None, None)
self.removeInvalidInput = Student(5, None, None)
self.updateValidInput = Student(1, None, None)
self.updateInvalidInput = Student(5, None, None)
self.updateParameter = "New Student"
def test__add_student__validInput__addsStudent(self):
self.__student_repository.add(self.addValidInput)
students = self.__student_repository.get_all_items()
self.assertEqual(students, [
Student(1, "Test Student", 917),
Student(2, "Student Test", 916),
Student(3, "Le Student", 916)
])
def test__add_student__invalidInput__raisesError(self):
try:
self.__student_repository.add(self.addInvalidInput)
except RepoError:
pass
self.assertRaises(RepoError)
def test__remove_student__validInput__removesStudent(self):
self.__student_repository.remove(self.removeValidInput)
students = self.__student_repository.get_all_items()
self.assertEqual(students, [Student(2, "Student Test", 916)])
def test__remove_student__invalidInput__raisesError(self):
try:
self.__student_repository.remove(self.removeInvalidInput)
except RepoError:
pass
self.assertRaises(RepoError)
class AssignmentTests(unittest.TestCase):
def setUp(self):
self.__assignment_repository = Repository()
self.__assignment_repository.add(Assignment(1, "Test Assignment", 917))
self.__assignment_repository.add(Assignment(2, "Assignment Test", 916))
self.addValidInput = Assignment(3, "Le Assignment", 916)
self.addInvalidInput = Assignment(1, "Le Assignment", "da")
self.removeValidInput = Assignment(1, None, None)
self.removeInvalidInput = Assignment(5, None, None)
self.updateValidInput = Assignment(1, None, None)
self.updateInvalidInput = Assignment(5, None, None)
self.updateParameter = "New Assignment"
def test__add_assignment__validInput__addsAssignment(self):
self.__assignment_repository.add(self.addValidInput)
assignments = self.__assignment_repository.get_all_items()
self.assertEqual(assignments, [
Assignment(1, "Test Assignment", 917),
Assignment(2, "Assignment Test", 916),
Assignment(3, "Le Assignment", 916)
])
def test__add_assignment__invalidInput__raisesError(self):
try:
self.__assignment_repository.add(self.addInvalidInput)
except RepoError:
pass
self.assertRaises(RepoError)
def test__remove_assignment__validInput__removesAssignment(self):
self.__assignment_repository.remove(self.removeValidInput)
assignments = self.__assignment_repository.get_all_items()
self.assertEqual(assignments, [Assignment(2, "Assignment Test", 916)])
def test__remove_assignment__invalidInput__raisesError(self):
try:
self.__assignment_repository.remove(self.removeInvalidInput)
except RepoError:
pass
self.assertRaises(RepoError)
