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