def test_add_save(self):
ada = TravelArrangementDataAccess("test.ini")
r = TravelArrangement()
r.set_id("id3")
r.set_name("name 3")
r.set_description("desc 3")
self.assertTrue(ada.add(r))
ada.save()
with open("test.ini", "rt") as f:
self.assertEqual(f.readline(), "[id3]\n")
self.assertEqual(f.readline(), "name = name 3\n")
self.assertEqual(f.readline(), "description = desc 3\n")
def test_add_then_list(self):
ada = TravelArrangementDataAccess("test.ini")
ada.load()
a = TravelArrangement()
a.set_id("id4")
a.set_name("name 4")
a.set_description("desc 4")
self.assertTrue(ada.add(a))
allCmt = ada.list_all()
self.assertEqual(len(allCmt), 4)
lastCmt = allCmt[3]
self.assertFalse(lastCmt is a)
self.assertEqual(lastCmt.get_id(), "id4")
self.assertEqual(lastCmt.get_name(), "name 4")
self.assertEqual(lastCmt.get_description(), "desc 4")
