1
0
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:
Ilya Kreymer 2016-03-10 21:35:01 -08:00
parent c309637a3a
commit 46d013ab19

View File

@ -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')