80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
from cache.FastCache import FastCache
|
|
from core.global_symbols import NotFound
|
|
|
|
|
|
def test_i_can_put_an_retrieve_values():
|
|
cache = FastCache()
|
|
cache.put("key", "value")
|
|
|
|
assert cache.get("key") == "value"
|
|
assert cache.cache == {"key": "value"}
|
|
assert cache.lru == ["key"]
|
|
|
|
|
|
def test_i_can_put_and_retrieve_multiple_items():
|
|
cache = FastCache()
|
|
cache.put("key1", "value1")
|
|
cache.put("key2", "value2")
|
|
cache.put("key3", "value3")
|
|
|
|
assert cache.cache == {"key1": "value1", "key2": "value2", "key3": "value3"}
|
|
assert cache.lru == ["key1", "key2", "key3"]
|
|
|
|
|
|
def test_i_the_least_used_is_remove_first():
|
|
cache = FastCache(3)
|
|
cache.put("key1", "value1")
|
|
cache.put("key2", "value2")
|
|
cache.put("key3", "value3")
|
|
|
|
cache.put("key4", "value4")
|
|
assert cache.cache == {"key4": "value4", "key2": "value2", "key3": "value3"}
|
|
assert cache.lru == ["key2", "key3", "key4"]
|
|
|
|
cache.put("key5", "value5")
|
|
assert cache.cache == {"key4": "value4", "key5": "value5", "key3": "value3"}
|
|
assert cache.lru == ["key3", "key4", "key5"]
|
|
|
|
|
|
def test_i_can_put_the_same_key_several_times():
|
|
cache = FastCache()
|
|
cache.put("key1", "value1")
|
|
cache.put("key2", "value2")
|
|
cache.put("key1", "value3")
|
|
|
|
assert cache.cache == {"key1": "value3", "key2": "value2"}
|
|
assert cache.lru == ["key2", "key1"]
|
|
|
|
|
|
def test_none_is_returned_when_not_found():
|
|
cache = FastCache()
|
|
assert cache.get("foo") is NotFound
|
|
|
|
|
|
def test_i_can_evict_by_key():
|
|
cache = FastCache()
|
|
cache.put("key1", "value1")
|
|
cache.put("to_keep1", "to_keep_value1")
|
|
cache.put("key2", "value2")
|
|
cache.put("to_keep2", "to_keep_value2")
|
|
cache.put("key3", "value3")
|
|
cache.put("to_keep3", "to_keep_value3")
|
|
|
|
cache.evict_by_key(lambda k: k.startswith("key"))
|
|
assert cache.cache == {"to_keep1": "to_keep_value1",
|
|
"to_keep2": "to_keep_value2",
|
|
"to_keep3": "to_keep_value3"}
|
|
|
|
assert cache.lru == ["to_keep1", "to_keep2", "to_keep3"]
|
|
|
|
|
|
def test_i_can_get_default_value():
|
|
cache = FastCache(max_size=3, default=lambda key: key + 1)
|
|
|
|
assert cache.get(1) == 2
|
|
assert cache.get(2) == 3
|
|
assert cache.get(3) == 4
|
|
assert cache.get(4) == 5
|
|
|
|
assert cache.cache == {2: 3, 3: 4, 4: 5} # only 3 values
|