def test_grpc_ssl_with_flow_and_client(cert_pem, key_pem, error_log_level):
with Flow(
protocol='grpc',
ssl_certfile=cert_pem,
ssl_keyfile=key_pem,
) as flow:
with open(cert_pem, 'rb') as f:
creds = f.read()
GrpcConnectionPool.send_health_check_sync(
target=f'localhost:{flow.port}',
root_certificates=creds,
tls=True,
timeout=1.0,
)
def is_ready(ctrl_address: str, **kwargs) -> bool:
"""
Check if status is ready.
:param ctrl_address: the address where the control request needs to be sent
:param kwargs: extra keyword arguments
:return: True if status is ready else False.
"""
try:
from grpc_health.v1 import health_pb2, health_pb2_grpc
response = GrpcConnectionPool.send_health_check_sync(
ctrl_address, timeout=1.0
)
# TODO: Get the proper value of the ServingStatus SERVING KEY
return response.status == 1
except RpcError:
return False