def test_add_save(self): rda = RoomReservationDataAccess("test.ini") r = RoomReservation() r.set_id("id3") r.set_user_id("user id3") r.set_reserved_room_id("room3") r.set_start_time(2015, 6, 11, 12) r.set_end_time(2015, 6, 12, 16) self.assertTrue(rda.add_reservation(r)) rda.save() with open("test.ini", "rt") as f: self.assertEqual(f.readline(), "[id3]\n") self.assertEqual(f.readline(), "start time = 06/11/15 12:00:00\n") self.assertEqual(f.readline(), "end time = 06/12/15 16:00:00\n") self.assertEqual(f.readline(), "user id = user id3\n") self.assertEqual(f.readline(), "room id = room3\n")
def test_add_then_list(self): rda = RoomReservationDataAccess("test.ini") rda.load() r = RoomReservation() r.set_id("id4") r.set_user_id("user id4") r.set_reserved_room_id("room3") r.set_start_time(2015, 6, 12, 16) r.set_end_time(2015, 6, 12, 18) self.assertTrue(rda.add_reservation(r)) allRsv = rda.list_all_reservations() self.assertEqual(len(allRsv), 4) lastRsv = allRsv[3] self.assertFalse(lastRsv is r) self.assertEqual(lastRsv.get_id(), "id4") self.assertEqual(lastRsv.get_user_id(), "user id4") self.assertEqual(lastRsv.get_reserved_room_id(), "room3") self.assertEqual(lastRsv.get_start_time_as_str(), "06/12/15 16:00:00") self.assertEqual(lastRsv.get_end_time_as_str(), "06/12/15 18:00:00")