Pull out _LARGE_UNASSIGNED_ID

pull/22667/head
Lidi Zheng 5 years ago
parent aaa0f328c9
commit da428e9020
  1. 18
      src/python/grpcio_tests/tests_aio/channelz/channelz_servicer_test.py

@ -27,8 +27,6 @@ from grpc_channelz.v1 import channelz_pb2_grpc
from tests.unit.framework.common import test_constants from tests.unit.framework.common import test_constants
from tests_aio.unit._test_base import AioTestBase from tests_aio.unit._test_base import AioTestBase
aio.shutdown_grpc_aio()
_SUCCESSFUL_UNARY_UNARY = '/test/SuccessfulUnaryUnary' _SUCCESSFUL_UNARY_UNARY = '/test/SuccessfulUnaryUnary'
_FAILED_UNARY_UNARY = '/test/FailedUnaryUnary' _FAILED_UNARY_UNARY = '/test/FailedUnaryUnary'
_SUCCESSFUL_STREAM_STREAM = '/test/SuccessfulStreamStream' _SUCCESSFUL_STREAM_STREAM = '/test/SuccessfulStreamStream'
@ -40,6 +38,8 @@ _DISABLE_REUSE_PORT = (('grpc.so_reuseport', 0),)
_ENABLE_CHANNELZ = (('grpc.enable_channelz', 1),) _ENABLE_CHANNELZ = (('grpc.enable_channelz', 1),)
_DISABLE_CHANNELZ = (('grpc.enable_channelz', 0),) _DISABLE_CHANNELZ = (('grpc.enable_channelz', 0),)
_LARGE_UNASSIGNED_ID = 10000
async def _successful_unary_unary(request, servicer_context): async def _successful_unary_unary(request, servicer_context):
return _RESPONSE return _RESPONSE
@ -167,7 +167,8 @@ class ChannelzServicerTest(AioTestBase):
async def test_get_top_channels_high_start_id(self): async def test_get_top_channels_high_start_id(self):
await _generate_channel_server_pairs(1) await _generate_channel_server_pairs(1)
resp = await self._channelz_stub.GetTopChannels( resp = await self._channelz_stub.GetTopChannels(
channelz_pb2.GetTopChannelsRequest(start_channel_id=10000)) channelz_pb2.GetTopChannelsRequest(
start_channel_id=_LARGE_UNASSIGNED_ID))
self.assertEqual(len(resp.channel), 0) self.assertEqual(len(resp.channel), 0)
self.assertEqual(resp.end, True) self.assertEqual(resp.end, True)
@ -286,28 +287,29 @@ class ChannelzServicerTest(AioTestBase):
async def test_invalid_query_get_server(self): async def test_invalid_query_get_server(self):
with self.assertRaises(aio.AioRpcError) as exception_context: with self.assertRaises(aio.AioRpcError) as exception_context:
await self._channelz_stub.GetServer( await self._channelz_stub.GetServer(
channelz_pb2.GetServerRequest(server_id=10000)) channelz_pb2.GetServerRequest(server_id=_LARGE_UNASSIGNED_ID))
self.assertEqual(grpc.StatusCode.NOT_FOUND, self.assertEqual(grpc.StatusCode.NOT_FOUND,
exception_context.exception.code()) exception_context.exception.code())
async def test_invalid_query_get_channel(self): async def test_invalid_query_get_channel(self):
with self.assertRaises(aio.AioRpcError) as exception_context: with self.assertRaises(aio.AioRpcError) as exception_context:
await self._channelz_stub.GetChannel( await self._channelz_stub.GetChannel(
channelz_pb2.GetChannelRequest(channel_id=10000)) channelz_pb2.GetChannelRequest(channel_id=_LARGE_UNASSIGNED_ID))
self.assertEqual(grpc.StatusCode.NOT_FOUND, self.assertEqual(grpc.StatusCode.NOT_FOUND,
exception_context.exception.code()) exception_context.exception.code())
async def test_invalid_query_get_subchannel(self): async def test_invalid_query_get_subchannel(self):
with self.assertRaises(aio.AioRpcError) as exception_context: with self.assertRaises(aio.AioRpcError) as exception_context:
await self._channelz_stub.GetSubchannel( await self._channelz_stub.GetSubchannel(
channelz_pb2.GetSubchannelRequest(subchannel_id=10000)) channelz_pb2.GetSubchannelRequest(
subchannel_id=_LARGE_UNASSIGNED_ID))
self.assertEqual(grpc.StatusCode.NOT_FOUND, self.assertEqual(grpc.StatusCode.NOT_FOUND,
exception_context.exception.code()) exception_context.exception.code())
async def test_invalid_query_get_socket(self): async def test_invalid_query_get_socket(self):
with self.assertRaises(aio.AioRpcError) as exception_context: with self.assertRaises(aio.AioRpcError) as exception_context:
await self._channelz_stub.GetSocket( await self._channelz_stub.GetSocket(
channelz_pb2.GetSocketRequest(socket_id=10000)) channelz_pb2.GetSocketRequest(socket_id=_LARGE_UNASSIGNED_ID))
self.assertEqual(grpc.StatusCode.NOT_FOUND, self.assertEqual(grpc.StatusCode.NOT_FOUND,
exception_context.exception.code()) exception_context.exception.code())
@ -315,7 +317,7 @@ class ChannelzServicerTest(AioTestBase):
with self.assertRaises(aio.AioRpcError) as exception_context: with self.assertRaises(aio.AioRpcError) as exception_context:
await self._channelz_stub.GetServerSockets( await self._channelz_stub.GetServerSockets(
channelz_pb2.GetServerSocketsRequest( channelz_pb2.GetServerSocketsRequest(
server_id=10000, server_id=_LARGE_UNASSIGNED_ID,
start_socket_id=0, start_socket_id=0,
)) ))
self.assertEqual(grpc.StatusCode.NOT_FOUND, self.assertEqual(grpc.StatusCode.NOT_FOUND,

Loading…
Cancel
Save