def test_get_missing_key(self):
"Expected to throw"
cache = CacheManager()
with raises(ReferenceError):
cache.get(key=None)
def test_set_has_value(self):
"Expected to find the required value"
cache = CacheManager()
cache.set(key='key', value='value', expiration=1)
assert cache.get('key') == 'value'
def test_set_missing_value(self):
"Expected to set an empty key"
cache = CacheManager()
cache.set(key='key', value=None, expiration=1)
assert not cache.get('key')
def test_get_existent_key(self):
"Expected to return None"
cache = CacheManager()
key = 'key'
value = 'value'
cache.set(key, value)
assert cache.get(key) == value
class CacheTest(unittest.TestCase):
def setUp(self):
self.cache_manager = CacheManager()
def test_use_cache(self):
key1 = "honda"
self.cache_manager.add(key1, Car(key1, 32000))
car = self.cache_manager.get(key1)
self.assertEqual(key1, car.brand)
self.assertEqual(32000, car.price)
self.cache_manager.remove(key1)
self.assertIsNone(self.cache_manager.get(key1))
def test_can_expire_cache(self):
key1 = "mykey"
self.cache_manager.add(key1, 'Value', expiry_time=timedelta(minutes=2))
value = self.cache_manager.get(key1,
current_time=datetime.now() +
timedelta(minutes=5))
self.assertIsNone(value)
def test_get_non_existent_key(self):
"Expected to return None"
cache = CacheManager()
assert not cache.get('idonotexist')
