def test_push_get(self): buff = TokenBuffer() msg = "some_random_buffer_data" buff.push(msg) buff_content = buff.get() self.assertEqual(msg, buff_content)
def test_get_clears(self): buff = TokenBuffer() msg = "some_random_buffer_data" buff.push(msg) buff_content1 = buff.get() buff_content2 = buff.get() self.assertEqual(msg, buff_content1) self.assertEqual("", buff_content2)