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}"