mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-15 08:04:49 +01:00
test redis: minor tweak to use @patch for fakeredis mock
This commit is contained in:
parent
c309637a3a
commit
46d013ab19
@ -10,21 +10,24 @@ from .testutils import key_ts_res
|
||||
|
||||
import pytest
|
||||
|
||||
import redis
|
||||
import fakeredis
|
||||
from fakeredis import FakeStrictRedis
|
||||
from mock import patch
|
||||
|
||||
redis.StrictRedis = fakeredis.FakeStrictRedis
|
||||
redis.Redis = fakeredis.FakeRedis
|
||||
redismock = patch('redis.StrictRedis', FakeStrictRedis)
|
||||
redismock.start()
|
||||
|
||||
def setup_module():
|
||||
global r
|
||||
r = fakeredis.FakeStrictRedis(db=2)
|
||||
r = FakeStrictRedis.from_url('redis://localhost:6379/2')
|
||||
r.delete('test:rediscdx')
|
||||
with open('testdata/iana.cdxj', 'rb') as fh:
|
||||
for line in fh:
|
||||
r.zadd('test:rediscdx', 0, line.rstrip())
|
||||
|
||||
|
||||
def teardown_module():
|
||||
redismock.stop()
|
||||
|
||||
|
||||
local_sources = [
|
||||
FileIndexSource('testdata/iana.cdxj'),
|
||||
RedisIndexSource('redis://localhost:6379/2/test:rediscdx')
|
||||
|
Loading…
x
Reference in New Issue
Block a user