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
59
60
61
62
63
64
65
66
|
import sys
from contextlib import ExitStack
from unittest.mock import create_autospec, patch
import pytest
from aiocache import caches
from aiocache.base import BaseCache
from aiocache.plugins import BasePlugin
if sys.version_info < (3, 8):
# Missing AsyncMock on 3.7
collect_ignore_glob = ["*"]
@pytest.fixture(autouse=True)
def reset_caches():
caches.set_config(
{
"default": {
"cache": "aiocache.SimpleMemoryCache",
"serializer": {"class": "aiocache.serializers.NullSerializer"},
}
}
)
@pytest.fixture
def mock_cache(mocker):
return create_autospec(BaseCache, instance=True)
@pytest.fixture
def mock_base_cache():
"""Return BaseCache instance with unimplemented methods mocked out."""
plugin = create_autospec(BasePlugin, instance=True)
cache = BaseCache(timeout=0.002, plugins=(plugin,))
methods = ("_add", "_get", "_gets", "_set", "_multi_get", "_multi_set", "_delete",
"_exists", "_increment", "_expire", "_clear", "_raw", "_close",
"_redlock_release", "acquire_conn", "release_conn")
with ExitStack() as stack:
for f in methods:
stack.enter_context(patch.object(cache, f, autospec=True))
stack.enter_context(patch.object(cache, "_serializer", autospec=True))
yield cache
@pytest.fixture
def base_cache():
return BaseCache()
@pytest.fixture
async def redis_cache():
from aiocache.backends.redis import RedisCache
async with RedisCache() as cache:
yield cache
@pytest.fixture
async def memcached_cache():
from aiocache.backends.memcached import MemcachedCache
async with MemcachedCache() as cache:
yield cache
|
