def test_pushing_works_like_a_stack(self): push_stream(self.cache, self.user_id, 'junk') push_stream(self.cache, self.user_id, self.token) self.assertEqual(self.token, pop_stream(self.cache, self.user_id)) self.assertEqual('junk', pop_stream(self.cache, self.user_id)) self.assertIsNone(pop_stream(self.cache, self.user_id))
def test_set_overrides_stack(self): push_stream(self.cache, self.user_id, '1') push_stream(self.cache, self.user_id, '2') self.assertEqual('2', top_stream(self.cache, self.user_id)) set_stream(self.cache, self.user_id, '3') self.assertEqual('3', pop_stream(self.cache, self.user_id)) self.assertIsNone(pop_stream(self.cache, self.user_id))
def test_set_overrides_stack(self): push_stream(self.cache, self.user_id, "1") push_stream(self.cache, self.user_id, "2") self.assertEqual("2", top_stream(self.cache, self.user_id)) set_stream(self.cache, self.user_id, "3") self.assertEqual("3", pop_stream(self.cache, self.user_id)) self.assertIsNone(pop_stream(self.cache, self.user_id))
def test_adding_removing_stream(self): self.assertTrue(push_stream(self.cache, self.user_id, self.token)) # peak at the top self.assertEqual(self.token, top_stream(self.cache, self.user_id)) self.assertIsNone(top_stream(self.cache, 'not dave')) # pop it off self.assertEqual(self.token, pop_stream(self.cache, self.user_id)) self.assertIsNone(top_stream(self.cache, self.user_id))