def test_event_entity(self): ''' Function for testing the event class ''' event = Event('1', '20/12/2012', '10:59', 'Wedding') self.assertEqual(event.get_id_event(), '1') self.assertEqual(event.get_date(), '20/12/2012') self.assertEqual(event.get_time(), '10:59') self.assertEqual(event.get_descr(), 'Wedding')
def test_create_assignment(self): ''' Functon that tests the create assignment feature ''' person = Person('1', 'Geo', 'Olanu') event = Event('1', "20/07/2019", "10:58", "Wedding") assignment = Assignment(person.get_id_pers(), event.get_id_event()) self.assertEqual(assignment.get_person_id(), '1') self.assertEqual(assignment.get_event_id(), '1')
class test_store_assig(TestCase): def setUp(self): self.fileName = "testassig.txt" self.repo = AssignmentFileRepository(self.fileName) self.repo.removeAll() self.person3 = Person("5", "Andrei", "Roman") self.person1 = Person("2", "Alex", "Caracal") self.person2 = Person("3", "Andrada", "Valcea") self.event1 = Event("1", "20/11/2012", "10:58", "Funeral") self.event2 = Event("2", "20/09/2013", "11:21", "Party") self.event3 = Event("3", "19/06/2017", "12:31", "Wedding") self.assignment1 = Assignment(self.person1.get_id_pers(), self.event1.get_id_event()) self.assignment2 = Assignment(self.person2.get_id_pers(), self.event2.get_id_event()) self.assignment3 = Assignment(self.person3.get_id_pers(), self.event3.get_id_event()) self.repo.store_assig(self.assignment1) self.repo.store_assig(self.assignment2) self.repo.store_assig(self.assignment3) self.assertEqual(self.repo.size(), 3) self.repo2 = AssignmentFileRepository(self.fileName) self.assignemnt4 = Assignment(self.person1.get_id_pers(), self.event2.get_id_event()) self.repo2.store_assig(self.assignemnt4) def tearDown(self): self.repo.removeAll() def test_store_assig(self): self.assertEqual(self.repo.search_assig(self.assignment1), self.assignment1) self.assertEqual(self.repo2.size(), 4) if __name__ == "__main__": unittest_main()
def test_event_repo(self): ''' Function to test the event repository store ''' event = Event('1', "20/06/2010", "10:56", "Funeral") repo = EventRepository(EventValidator) self.assertEqual(repo.size(), 0) repo.store_event(event) self.assertEqual(repo.size(), 1) event = Event('2', "15/06/2011", "12:03", "Wedding") repo.store_event(event) self.assertEqual(repo.size(), 2) repo.delete_event(event.get_id_event()) self.assertEqual(repo.size(), 1) event = Event("1", "12/12/2012", "11:07", "Birthday Party")