def test_add_save(self): cda = CommentDataAccess("test.ini") r = Comment() r.set_id("id3") r.set_user_id("user3") r.set_message("message3") r.set_time(2015, 6, 11, 12) self.assertTrue(cda.add_comment(r)) cda.save() with open("test.ini", "rt") as f: self.assertEqual(f.readline(), "[id3]\n") self.assertEqual(f.readline(), "time = 06/11/15 12:00:00\n") self.assertEqual(f.readline(), "user id = user3\n") self.assertEqual(f.readline(), "message = message3\n")
def test_add_then_list(self): cda = CommentDataAccess("test.ini") cda.load() c = Comment() c.set_id("id4") c.set_user_id("user4") c.set_message("message3") c.set_time(2015, 6, 12, 16) self.assertTrue(cda.add_comment(c)) allCmt = cda.list_all_comments() self.assertEqual(len(allCmt), 4) lastCmt = allCmt[3] self.assertFalse(lastCmt is c) self.assertEqual(lastCmt.get_id(), "id4") self.assertEqual(lastCmt.get_user_id(), "user4") self.assertEqual(lastCmt.get_message(), "message3") self.assertEqual(lastCmt.get_time_as_str(), "06/12/15 16:00:00")
def test_add_autoid(self): cda = CommentDataAccess("test.ini") cda.load() r = Comment() r.set_id("id3") r.set_user_id("user4") r.set_message("message3") r.set_time(2015, 6, 12, 16) self.assertTrue(cda.add_comment(r)) cda.save() with open("test.ini", "rt") as f: self.assertEqual(f.readline(), "[id1]\n") self.assertEqual(f.readline(), "time = 06/10/15 12:00:00\n") self.assertEqual(f.readline(), "user id = user1\n") self.assertEqual(f.readline(), "message = message1\n") self.assertEqual(f.readline(), "\n") self.assertEqual(f.readline(), "[id2]\n") self.assertEqual(f.readline(), "time = 06/10/15 12:00:00\n") self.assertEqual(f.readline(), "user id = user2\n") self.assertEqual(f.readline(), "message = message2\n") self.assertEqual(f.readline(), "\n") self.assertEqual(f.readline(), "[id3]\n") self.assertEqual(f.readline(), "time = 06/11/15 12:00:00\n") self.assertEqual(f.readline(), "user id = user3\n") self.assertEqual(f.readline(), "message = message3\n") self.assertEqual(f.readline(), "\n") self.assertEqual(f.readline(), "[0]\n") self.assertEqual(f.readline(), "time = 06/12/15 16:00:00\n") self.assertEqual(f.readline(), "user id = user4\n") self.assertEqual(f.readline(), "message = message3\n") self.assertEqual(f.readline(), "\n") self.assertEqual(f.readline(), "")