def check_thrift_service(endpoint: EndpointConfiguration) -> None:
pool = ThriftConnectionPool(endpoint, size=1, timeout=TIMEOUT)
with pool.connection() as protocol:
client = BaseplateService.Client(protocol)
assert client.is_healthy(), "service indicated unhealthiness"
def raw_thrift_client(endpoint):
pool = ThriftConnectionPool(endpoint)
with pool.connection() as client_protocol:
yield TestService.Client(client_protocol)
def raw_thrift_client(endpoint, client_spec):
pool = ThriftConnectionPool(endpoint)
with pool.connection() as client_protocol:
yield client_spec.Client(client_protocol)
def check_thrift_service(endpoint: EndpointConfiguration, probe: int) -> None:
pool = ThriftConnectionPool(endpoint, size=1, timeout=TIMEOUT)
with pool.connection() as protocol:
client = BaseplateService.Client(protocol)
assert client.is_healthy(request=IsHealthyRequest(
probe=probe), ), "service indicated unhealthiness in probe {probe}"