def test__patch_callable_name_no_op(): callable = mock.Mock(spec=["__name__"]) callable.__name__ = "test_callable" grpc_helpers._patch_callable_name(callable) assert callable.__name__ == "test_callable"
def test__patch_callable_name_no_op(): callable = mock.Mock(spec=['__name__']) callable.__name__ = 'test_callable' grpc_helpers._patch_callable_name(callable) assert callable.__name__ == 'test_callable'
def test__patch_callable_name(): callable = mock.Mock(spec=["__class__"]) callable.__class__ = mock.Mock(spec=["__name__"]) callable.__class__.__name__ = "TestCallable" grpc_helpers._patch_callable_name(callable) assert callable.__name__ == "TestCallable"
def test__patch_callable_name(): callable = mock.Mock(spec=['__class__']) callable.__class__ = mock.Mock(spec=['__name__']) callable.__class__.__name__ = 'TestCallable' grpc_helpers._patch_callable_name(callable) assert callable.__name__ == 'TestCallable'
def _wrap_unary_errors(callable_): """Map errors for Unary-Unary async callables.""" grpc_helpers._patch_callable_name(callable_) @functools.wraps(callable_) def error_remapped_callable(*args, **kwargs): call = callable_(*args, **kwargs) return _WrappedUnaryUnaryCall().with_call(call) return error_remapped_callable
def _wrap_stream_errors(callable_): """Map errors for streaming RPC async callables.""" grpc_helpers._patch_callable_name(callable_) @functools.wraps(callable_) async def error_remapped_callable(*args, **kwargs): call = callable_(*args, **kwargs) if isinstance(call, aio.UnaryStreamCall): call = _WrappedUnaryStreamCall().with_call(call) elif isinstance(call, aio.StreamUnaryCall): call = _WrappedStreamUnaryCall().with_call(call) elif isinstance(call, aio.StreamStreamCall): call = _WrappedStreamStreamCall().with_call(call) else: raise TypeError('Unexpected type of call %s' % type(call)) await call.wait_for_connection() return call return error_remapped_callable