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