class Test_Cache(unittest.TestCase):
def setUp(self):
self.cache = Cache(3)
def test_should_keep_last_items(self):
self.cache.append(1)
self.cache.append(2)
self.cache.append(3)
self.cache.append(4)
self.assertEqual(list(self.cache.items), [2, 3, 4])
def test_should_consume_cache(self):
self.cache.append(1)
self.cache.append(2)
self.cache.append(3)
self.cache.append(4)
self.assertEqual(list(self.cache.consume()), [2, 3, 4])
self.cache.append(5)
self.assertEqual(list(self.cache.consume()), [5])
self.cache.append(6)
self.cache.append(7)
self.assertEqual(list(self.cache.consume()), [6, 7])
def test_should_return_empty_if_empty(self):
valor = list(self.cache.consume())
self.assertEqual(valor, [])
def test_should_return_cache_length(self):
self.cache.append(1)
self.cache.append(2)
self.assertEqual(len(self.cache), 2)
self.cache.append(3)
self.cache.append(4)
self.cache.append(5)
self.assertEqual(len(self.cache), 3)