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
57
58
|
import pytest
from auto_pytabs.core import Cache
@pytest.fixture()
def cache() -> Cache:
cache = Cache()
cache.clear_all()
cache._init_cache_dir()
yield cache
cache.clear_all()
def test_get_not_found(cache) -> None:
assert cache.get("foo") is None
def test_get(cache) -> None:
cache.cache_content_dir.joinpath("foo").write_text("bar")
cache._load()
assert cache.get("foo") == "bar"
assert cache._cache["foo"] == "bar"
def test_set(cache) -> None:
cache.set("foo", "bar")
assert cache.get("foo") == "bar"
@pytest.mark.parametrize("evict", [True, False])
def test_persist(cache: Cache, evict: bool) -> None:
test_file_one = cache.cache_content_dir.joinpath("one")
test_file_two = cache.cache_content_dir.joinpath("two")
test_file_one.write_text("foo")
test_file_two.write_text("bar")
cache._load()
cache.get("two")
cache.set("three", "baz")
cache.persist(evict=evict)
if evict:
assert not test_file_one.exists()
else:
assert test_file_one.exists()
assert cache.cache_content_dir.joinpath("two").exists()
assert cache.cache_content_dir.joinpath("three").exists()
def test_clear_all(cache) -> None:
cache.set("foo", "bar")
cache.clear_all()
assert not cache.cache_dir.exists()
|
