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"