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