def setUp(self): self.db = connect_db(':memory:') init_db(self.db) self.s = EntryService(self.db, TimestampService())
class EntryServiceTests(unittest.TestCase): def setUp(self): self.db = connect_db(':memory:') init_db(self.db) self.s = EntryService(self.db, TimestampService()) def tearDown(self): self.db.close() def test_add_entry(self): #when self.s.add_entry('teohm', 'content') #then list = self.s.list_entries('teohm') self.assertEqual(len(list), 1) entry = list[0] self.assertEqual(entry['content'], 'content') def test_add_entry_returned_object(self): #when entry = self.s.add_entry('teohm', 'content', '2010-01-01T00:00:00Z') entry2 = self.s.add_entry('teohm', 'content') #then assert entry['content'] == 'content' assert entry['timestamp'] == '2010-01-01T00:00:00Z' assert entry2['content'] == 'content' assert entry2['timestamp'] is not None def test_list_no_entry(self): #when list = self.s.list_entries('teohm') #then self.assertEqual(len(list), 0) def test_list_entries(self): #when self.s.add_entry('teohm', 'content', '2010-01-01T00:00:00Z') self.s.add_entry('teohm', 'content', '2010-01-01T00:00:01Z') self.s.add_entry('teohm', 'content', '2010-01-01T00:00:02Z') self.s.add_entry('foo', 'content', '2010-01-01T00:00:03Z') #then list = self.s.list_entries('teohm') self.assertEqual(len(list), 3) self.assertTrue(list[0]['timestamp'] > list[1]['timestamp']) self.assertTrue(list[1]['timestamp'] > list[2]['timestamp'])