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")
