def test_invalid_query_get_server(self):
try:
self._channelz_stub.GetServer(
channelz_pb2.GetServerRequest(server_id=10000))
except BaseException as e:
self.assertIn('StatusCode.NOT_FOUND', str(e))
else:
self.fail('Invalid query not detected')
def test_get_one_server(self):
self._pairs = _generate_channel_server_pairs(1)
gss_resp = self._channelz_stub.GetServers(
channelz_pb2.GetServersRequest(start_server_id=0))
self.assertEqual(len(gss_resp.server), 1)
gs_resp = self._channelz_stub.GetServer(
channelz_pb2.GetServerRequest(
server_id=gss_resp.server[0].ref.server_id))
self.assertEqual(gss_resp.server[0].ref.server_id,
gs_resp.server.ref.server_id)
async def test_invalid_query_get_server(self):
with self.assertRaises(aio.AioRpcError) as exception_context:
await self._channelz_stub.GetServer(
channelz_pb2.GetServerRequest(server_id=_LARGE_UNASSIGNED_ID))
self.assertEqual(grpc.StatusCode.NOT_FOUND,
exception_context.exception.code())