def test_update_then_save(self):
ada = TravelArrangementDataAccess("test.ini")
ada.load()
a = TravelArrangement()
a.set_id("id1")
a.set_name("name 1.1")
a.set_description("description 1.1")
self.assertTrue(ada.update(a))
ada.save()
with open("test.ini", "rt") as f:
self.assertEqual(f.readline(), "[id1]\n")
self.assertEqual(f.readline(), "name = name 1.1\n")
self.assertEqual(f.readline(), "description = description 1.1\n")
self.assertEqual(f.readline(), "\n")
self.assertEqual(f.readline(), "[id2]\n")
self.assertEqual(f.readline(), "name = name 2\n")
self.assertEqual(f.readline(), "description = desc 2\n")
self.assertEqual(f.readline(), "\n")
self.assertEqual(f.readline(), "[id3]\n")
self.assertEqual(f.readline(), "name = name 3\n")
self.assertEqual(f.readline(), "description = desc 3\n")
self.assertEqual(f.readline(), "\n")
self.assertEqual(f.readline(), "")
def test_update_then_list(self):
ada = TravelArrangementDataAccess("test.ini")
ada.load()
a = TravelArrangement()
a.set_id("id1")
a.set_name("name 1.1")
a.set_description("desc 1.1")
self.assertTrue(ada.update(a))
tlist = ada.list_all()
self.assertEqual(len(tlist), 3)
a = tlist[0]
self.assertEqual(a.get_id(), "id1")
self.assertEqual(a.get_name(), "name 1.1")
self.assertEqual(a.get_description(), "desc 1.1")
a = tlist[1]
self.assertEqual(a.get_id(), "id2")
self.assertEqual(a.get_name(), "name 2")
self.assertEqual(a.get_description(), "desc 2")
a = tlist[2]
self.assertEqual(a.get_id(), "id3")
self.assertEqual(a.get_name(), "name 3")
self.assertEqual(a.get_description(), "desc 3")
