def setUp(self):
self._protos = python_protos.Library.from_strings(TEST_PROTO_1)
self._client = client.Client.from_modules(
callback_client.Impl(), [client.Channel(1, self._handle_request)],
self._protos.modules())
self._last_request: packet_pb2.RpcPacket = None
self._next_packets: List[Tuple[bytes, Status]] = []
self._send_responses_on_request = True
def test_default_timeouts_set_for_all_rpcs(self):
rpc_client = client.Client.from_modules(callback_client.Impl(
99, 100), [client.Channel(1, lambda *a, **b: None)],
self._protos.modules())
rpcs = rpc_client.channel(1).rpcs
self.assertEqual(
rpcs.pw.test1.PublicService.SomeUnary.default_timeout_s, 99)
self.assertEqual(
rpcs.pw.test1.PublicService.SomeServerStreaming.default_timeout_s,
100)
def _test_setup(output=None):
protos = python_protos.Library.from_strings([TEST_PROTO_1, TEST_PROTO_2])
return protos, client.Client.from_modules(callback_client.Impl(),
[client.Channel(1, output)],
protos.modules())