def __init__(self, channel: aio.Channel): self.UnaryCall = channel.unary_unary( '/grpc.testing.BenchmarkService/UnaryCall') self.StreamingFromServer = channel.unary_stream( '/grpc.testing.BenchmarkService/StreamingFromServer') self.StreamingCall = channel.stream_stream( '/grpc.testing.BenchmarkService/StreamingCall')
async def block_until_certain_state(channel: aio.Channel, expected_state: grpc.ChannelConnectivity): state = channel.get_state() while state != expected_state: await channel.wait_for_state_change(state) state = channel.get_state()