1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
from time import sleep
import pytest
from clear import ClearTests
from common import CommonTests
from has import HasTests
from cachelib import SimpleCache
class SillySerializer:
"""A pointless serializer only for testing"""
def dumps(self, value):
return repr(value).encode()
def loads(self, bvalue):
return eval(bvalue.decode())
class CustomCache(SimpleCache):
"""Our custom cache client with non-default serializer"""
# overwrite serializer
serializer = SillySerializer()
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
@pytest.fixture(autouse=True, params=[SimpleCache, CustomCache])
def cache_factory(request):
def _factory(self, *args, **kwargs):
return request.param(*args, **kwargs)
request.cls.cache_factory = _factory
class TestSimpleCache(CommonTests, HasTests, ClearTests):
def test_threshold(self):
threshold = len(self.sample_pairs) // 2
cache = self.cache_factory(threshold=threshold)
assert cache.set_many(self.sample_pairs)
assert abs(len(cache._cache) - threshold) <= 1
def test_prune_old_entries(self):
threshold = 2 * len(self.sample_pairs) - 1
cache = self.cache_factory(threshold=threshold)
for k, v in self.sample_pairs.items():
assert cache.set(f"{k}-t0.1", v, timeout=0.1)
assert cache.set(f"{k}-t5.0", v, timeout=5.0)
sleep(2)
for k, v in self.sample_pairs.items():
assert cache.set(k, v)
assert f"{k}-t5.0" in cache._cache.keys()
assert f"{k}-t0.1" not in cache._cache.keys()
|
