async def test_is_ready(test_client, loop, test_name, mock_return,
expected_code, expected_msg):
"""Assert that the readiness check works as expected."""
# setup
class Service():
def __init__(self, mock_return):
self.mock_return = mock_return
@property
async def is_ready(self):
return self.mock_return
if mock_return is None:
mock_service = None
else:
mock_service = [Service(mock_return)]
probe = Probes(loop=loop, components=mock_service)
client = await test_client(probe.get_app())
# test it
resp = await client.get('/readyz')
# verify
assert resp.status == expected_code
assert expected_msg == await resp.json()
def test_get_app(loop):
"""Assert that get_app returns the same instance everytime."""
probe = Probes(loop=loop, port=unused_port())
app1 = probe.get_app()
app2 = probe.get_app()
assert app1 == app2