def client_db_disconnected() -> Generator:
server = FakeServer()
server.connected = False
db = FakeRedis(server=server)
app.dependency_overrides[get_db] = lambda: db
with TestClient(app) as client:
yield client
app.dependency_overrides = {}
def setUp(self):
server = FakeServer()
server.connected = True
self.store = RedisStore(prefix='prefix:',
redis_class=FakeStrictRedis,
server=server)
self.redis = FakeStrictRedis(server=server)
super(RedisStoreTestCase, self).setUp()
def test_init_connect_exc():
server = FakeServer()
server.connected = False
fake_cache_url = 'redis://random:123'
with pytest.raises(ConnectionError) as exc:
RedisAdapter(default_name='name',
cache_url=fake_cache_url,
server=server)
assert str(
exc.value) == f'Could not connect to Redis with URL: {fake_cache_url}'
def mock_flask_server(create_server):
redis_server = FakeServer()
mock_db = FakeRedis(server=redis_server)
server = create_server(mock_db)
server.redis_server = redis_server
server.app.config['TESTING'] = True
server.client = server.app.test_client()
return server
def mock_work_server(create_server):
redis_server = FakeServer()
mock_db = FakeRedis(server=redis_server)
server = create_server(mock_db)
server.redis_server = redis_server
server.app.config['TESTING'] = True
server.client = server.app.test_client()
server.data = MockDataStorage()
return server
def mock_dashboard_server(create_server):
redis_server = FakeServer()
mock_db = FakeRedis(server=redis_server)
mock_docker = MagicMock()
server = create_server(mock_db, mock_docker)
server.redis_server = redis_server
server.app.config['TESTING'] = True
server.client = server.app.test_client()
server.data = MockDataStorage()
return server
def test_create_server_not_connected():
redis_server = FakeServer()
mock_db = FakeRedis(server=redis_server)
redis_server.connected = False
assert create_server(mock_db) is None
import pytest
from fakeredis import FakeRedis, FakeServer
from fastapi.testclient import TestClient
from app.app import app, redis_connection
server = FakeServer()
def fake_redis_connection():
r = FakeRedis(server=server, encoding='utf-8')
return r
app.dependency_overrides[redis_connection] = fake_redis_connection
@pytest.fixture()
def client():
c = TestClient(app)
return c
def fake_redis_server():
from fakeredis import FakeServer
server = FakeServer()
return server
def setUp(self):
self.fake_redis_server = FakeServer()
self.fake_redis = FakeStrictRedisWithConnection(
server=self.fake_redis_server)
self.connection, self.channel = self.create_connection_and_channel(
self.fake_redis, transport=Transport)
def get_r(self):
r_server = server.get(self.alias)
if not r_server:
r_server = FakeRedis(server=FakeServer())
server[self.alias] = r_server
return r_server