def raw_thrift_client(endpoint):
pool = ThriftConnectionPool(endpoint)
with pool.connection() as client_protocol:
yield TestService.Client(client_protocol)
def check_thrift_service(endpoint):
pool = ThriftConnectionPool(endpoint, size=1, timeout=TIMEOUT)
with pool.connection() as protocol:
client = BaseplateService.Client(protocol)
assert client.is_healthy(), "service indicated unhealthiness"
def check_thrift_service(endpoint):
pool = ThriftConnectionPool(endpoint, size=1, timeout=TIMEOUT)
with pool.connection() as protocol:
client = BaseplateService.Client(protocol)
assert client.is_healthy(), "service indicated unhealthiness"