backup/pywb
1
0
Fork 0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-24 06:59:52 +01:00
Code Issues Projects Releases Activity

test redis: minor tweak to use @patch for fakeredis mock

Browse Source
This commit is contained in:
Ilya Kreymer 2016-03-10 21:35:01 -08:00
parent c309637a3a
commit 46d013ab19
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
webagg/test/test_indexsource.py
View File

@ -10,21 +10,24 @@ from .testutils import key_ts_res
import pytest import pytest
import redis from fakeredis import FakeStrictRedis
import fakeredis from mock import patch
redis.StrictRedis = fakeredis.FakeStrictRedis redismock = patch('redis.StrictRedis', FakeStrictRedis)
redis.Redis = fakeredis.FakeRedis redismock.start()
def setup_module(): def setup_module():
global r r = FakeStrictRedis.from_url('redis://localhost:6379/2')
r = fakeredis.FakeStrictRedis(db=2)
r.delete('test:rediscdx') r.delete('test:rediscdx')
with open('testdata/iana.cdxj', 'rb') as fh: with open('testdata/iana.cdxj', 'rb') as fh:
for line in fh: for line in fh:
r.zadd('test:rediscdx', 0, line.rstrip()) r.zadd('test:rediscdx', 0, line.rstrip())
def teardown_module():
redismock.stop()
local_sources = [ local_sources = [
FileIndexSource('testdata/iana.cdxj'), FileIndexSource('testdata/iana.cdxj'),
RedisIndexSource('redis://localhost:6379/2/test:rediscdx') RedisIndexSource('redis://localhost:6379/2/test:rediscdx')