def test_unbookmark_removes_timestamp(self): g = NetworkGraph() g.bookmark('content', 'fake_uuid') g.unbookmark('content', 'fake_uuid') with self.assertRaises(KeyError): g.bookmarked_on('fake_uuid')
def test_bookmark_sets_timestamp(self): g = NetworkGraph() g.bookmark('content', 'fake_uuid') self.assertTrue(isinstance(g.bookmarked_on('fake_uuid'), datetime))