def test_should_raise_exception_if_raise_set_to_true(app):
app.config['REDIS_ENABLED'] = True
redis_client = RedisClient()
redis_client.init_app(app)
redis_client.redis_store.get = Mock(side_effect=Exception('get failed'))
redis_client.redis_store.set = Mock(side_effect=Exception('set failed'))
redis_client.redis_store.incr = Mock(side_effect=Exception('inc failed'))
redis_client.redis_store.pipeline = Mock(
side_effect=Exception('pipeline failed'))
redis_client.redis_store.expire = Mock(
side_effect=Exception('expire failed'))
redis_client.redis_store.delete = Mock(
side_effect=Exception('delete failed'))
with pytest.raises(Exception) as e:
redis_client.get('test', raise_exception=True)
assert str(e.value) == 'get failed'
with pytest.raises(Exception) as e:
redis_client.set('test', 'test', raise_exception=True)
assert str(e.value) == 'set failed'
with pytest.raises(Exception) as e:
redis_client.incr('test', raise_exception=True)
assert str(e.value) == 'inc failed'
with pytest.raises(Exception) as e:
redis_client.exceeded_rate_limit('test',
100,
200,
raise_exception=True)
assert str(e.value) == 'pipeline failed'
with pytest.raises(Exception) as e:
redis_client.expire('test', 0, raise_exception=True)
assert str(e.value) == 'expire failed'
with pytest.raises(Exception) as e:
redis_client.delete('test', raise_exception=True)
assert str(e.value) == 'delete failed'
def test_should_not_raise_exception_if_raise_set_to_false(app, caplog, mocker):
mock_logger = mocker.patch('flask.Flask.logger')
app.config['REDIS_ENABLED'] = True
redis_client = RedisClient()
redis_client.init_app(app)
redis_client.redis_store.get = Mock(side_effect=Exception())
redis_client.redis_store.set = Mock(side_effect=Exception())
redis_client.redis_store.incr = Mock(side_effect=Exception())
redis_client.redis_store.pipeline = Mock(side_effect=Exception())
redis_client.redis_store.expire = Mock(side_effect=Exception())
redis_client.redis_store.delete = Mock(side_effect=Exception())
assert redis_client.get('get_key') is None
assert redis_client.set('set_key', 'set_value') is None
assert redis_client.incr('incr_key') is None
assert redis_client.exceeded_rate_limit('rate_limit_key', 100,
100) is False
assert redis_client.expire('expire_key', 100) is None
assert redis_client.delete('delete_key') is None
assert redis_client.delete('a', 'b', 'c') is None
assert mock_logger.mock_calls == [
call.exception('Redis error performing get on get_key'),
call.exception('Redis error performing set on set_key'),
call.exception('Redis error performing incr on incr_key'),
call.exception(
'Redis error performing rate-limit-pipeline on rate_limit_key'),
call.exception('Redis error performing expire on expire_key'),
call.exception('Redis error performing delete on delete_key'),
call.exception('Redis error performing delete on a, b, c'),
]
def test_should_not_raise_exception_if_raise_set_to_false(app):
app.config['REDIS_ENABLED'] = True
redis_client = RedisClient()
redis_client.init_app(app)
redis_client.redis_store.get = Mock(side_effect=Exception())
redis_client.redis_store.set = Mock(side_effect=Exception())
redis_client.redis_store.incr = Mock(side_effect=Exception())
redis_client.redis_store.pipeline = Mock(side_effect=Exception())
assert redis_client.get('test') is None
assert redis_client.set('test', 'test') is None
assert redis_client.incr('test') is None
assert redis_client.exceeded_rate_limit('test', 100, 100) is False
def build_redis_client(app, mocked_redis_pipeline, mocker):
redis_client = RedisClient()
redis_client.init_app(app)
mocker.patch.object(redis_client.redis_store, 'get', return_value=100)
mocker.patch.object(redis_client.redis_store, 'set')
mocker.patch.object(redis_client.redis_store, 'hincrby')
mocker.patch.object(redis_client.redis_store, 'hgetall',
return_value={b'template-1111': b'8', b'template-2222': b'8'})
mocker.patch.object(redis_client.redis_store, 'hmset')
mocker.patch.object(redis_client.redis_store, 'expire')
mocker.patch.object(redis_client.redis_store, 'delete')
mocker.patch.object(redis_client.redis_store, 'pipeline', return_value=mocked_redis_pipeline)
return redis_client
def mocked_redis_client(app):
app.config['REDIS_ENABLED'] = True
redis_client = RedisClient()
redis_client.init_app(app)
return redis_client