3
3
import asyncio
4
4
import gc
5
5
import pytest
6
- import redis as redisdb
7
6
import time
8
7
import uuid
9
8
from docker import Client as DockerClient
10
- from pymemcache .client .base import Client
11
9
12
10
13
- @pytest .yield_fixture
11
+ @pytest .yield_fixture ( scope = 'session' )
14
12
def loop (request ):
15
13
loop = asyncio .new_event_loop ()
16
14
asyncio .set_event_loop (None )
@@ -42,7 +40,7 @@ def pytest_addoption(parser):
42
40
43
41
44
42
@pytest .yield_fixture (scope = 'session' )
45
- def redis_server (docker , session_id , request ):
43
+ def redis_server (docker , session_id , loop , request ):
46
44
if not request .config .option .no_pull :
47
45
docker .pull ('redis:{}' .format ('latest' ))
48
46
container = docker .create_container (
@@ -57,10 +55,12 @@ def redis_server(docker, session_id, request):
57
55
delay = 0.001
58
56
for i in range (100 ):
59
57
try :
60
- conn = redisdb .StrictRedis (host = host , port = 6379 , db = 0 )
61
- conn .set ('foo' , 'bar' )
58
+ conn = loop .run_until_complete (
59
+ aioredis .create_connection ((host , 6379 ))
60
+ )
61
+ loop .run_until_complete (conn .execute ('SET' , 'foo' , 'bar' ))
62
62
break
63
- except redisdb . exceptions . ConnectionError as e :
63
+ except ConnectionRefusedError as e :
64
64
time .sleep (delay )
65
65
delay *= 2
66
66
else :
@@ -77,7 +77,7 @@ def redis_params(redis_server):
77
77
return dict (** redis_server ['redis_params' ])
78
78
79
79
80
- @pytest .yield_fixture ()
80
+ @pytest .yield_fixture
81
81
def redis (loop , redis_params ):
82
82
pool = None
83
83
@@ -97,7 +97,7 @@ def start(*args, no_loop=False, **kwargs):
97
97
98
98
99
99
@pytest .yield_fixture (scope = 'session' )
100
- def memcached_server (docker , session_id , request ):
100
+ def memcached_server (docker , session_id , loop , request ):
101
101
if not request .config .option .no_pull :
102
102
docker .pull ('memcached:{}' .format ('latest' ))
103
103
container = docker .create_container (
@@ -112,8 +112,8 @@ def memcached_server(docker, session_id, request):
112
112
delay = 0.001
113
113
for i in range (100 ):
114
114
try :
115
- client = Client (( host , 11211 ) )
116
- client . set ('foo' , 'bar' )
115
+ conn = aiomcache . Client (host , 11211 , loop = loop )
116
+ loop . run_until_complete ( conn . set (b 'foo' , b 'bar') )
117
117
break
118
118
except ConnectionRefusedError as e :
119
119
time .sleep (delay )
0 commit comments