mirror of https://github.com/grpc/grpc.git
Merge pull request #9145 from nathanielmanistaatgoogle/v1.0.x-upmerge
v1.0.x manual upmergepull/9165/head
Nicolas Noble
8 years ago
committed by
GitHub
commit
f29de26194
36 changed files with 608 additions and 124 deletions
@ -0,0 +1,175 @@ |
||||
# Copyright 2016, Google Inc. |
||||
# All rights reserved. |
||||
# |
||||
# Redistribution and use in source and binary forms, with or without |
||||
# modification, are permitted provided that the following conditions are |
||||
# met: |
||||
# |
||||
# * Redistributions of source code must retain the above copyright |
||||
# notice, this list of conditions and the following disclaimer. |
||||
# * Redistributions in binary form must reproduce the above |
||||
# copyright notice, this list of conditions and the following disclaimer |
||||
# in the documentation and/or other materials provided with the |
||||
# distribution. |
||||
# * Neither the name of Google Inc. nor the names of its |
||||
# contributors may be used to endorse or promote products derived from |
||||
# this software without specific prior written permission. |
||||
# |
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
"""Test of RPCs made against gRPC Python's application-layer API.""" |
||||
|
||||
import unittest |
||||
|
||||
import grpc |
||||
|
||||
from tests.unit.framework.common import test_constants |
||||
|
||||
_SERIALIZE_REQUEST = lambda bytestring: bytestring * 2 |
||||
_DESERIALIZE_REQUEST = lambda bytestring: bytestring[len(bytestring) // 2:] |
||||
_SERIALIZE_RESPONSE = lambda bytestring: bytestring * 3 |
||||
_DESERIALIZE_RESPONSE = lambda bytestring: bytestring[:len(bytestring) // 3] |
||||
|
||||
_UNARY_UNARY = '/test/UnaryUnary' |
||||
_UNARY_STREAM = '/test/UnaryStream' |
||||
_STREAM_UNARY = '/test/StreamUnary' |
||||
_STREAM_STREAM = '/test/StreamStream' |
||||
|
||||
|
||||
def _unary_unary_multi_callable(channel): |
||||
return channel.unary_unary(_UNARY_UNARY) |
||||
|
||||
|
||||
def _unary_stream_multi_callable(channel): |
||||
return channel.unary_stream( |
||||
_UNARY_STREAM, |
||||
request_serializer=_SERIALIZE_REQUEST, |
||||
response_deserializer=_DESERIALIZE_RESPONSE) |
||||
|
||||
|
||||
def _stream_unary_multi_callable(channel): |
||||
return channel.stream_unary( |
||||
_STREAM_UNARY, |
||||
request_serializer=_SERIALIZE_REQUEST, |
||||
response_deserializer=_DESERIALIZE_RESPONSE) |
||||
|
||||
|
||||
def _stream_stream_multi_callable(channel): |
||||
return channel.stream_stream(_STREAM_STREAM) |
||||
|
||||
|
||||
class InvalidMetadataTest(unittest.TestCase): |
||||
|
||||
def setUp(self): |
||||
self._channel = grpc.insecure_channel('localhost:8080') |
||||
self._unary_unary = _unary_unary_multi_callable(self._channel) |
||||
self._unary_stream = _unary_stream_multi_callable(self._channel) |
||||
self._stream_unary = _stream_unary_multi_callable(self._channel) |
||||
self._stream_stream = _stream_stream_multi_callable(self._channel) |
||||
|
||||
def testUnaryRequestBlockingUnaryResponse(self): |
||||
request = b'\x07\x08' |
||||
metadata = (('InVaLiD', 'UnaryRequestBlockingUnaryResponse'),) |
||||
expected_error_details = "metadata was invalid: %s" % metadata |
||||
with self.assertRaises(ValueError) as exception_context: |
||||
self._unary_unary(request, metadata=metadata) |
||||
self.assertIn(expected_error_details, str(exception_context.exception)) |
||||
|
||||
def testUnaryRequestBlockingUnaryResponseWithCall(self): |
||||
request = b'\x07\x08' |
||||
metadata = (('InVaLiD', 'UnaryRequestBlockingUnaryResponseWithCall'),) |
||||
expected_error_details = "metadata was invalid: %s" % metadata |
||||
with self.assertRaises(ValueError) as exception_context: |
||||
self._unary_unary.with_call(request, metadata=metadata) |
||||
self.assertIn(expected_error_details, str(exception_context.exception)) |
||||
|
||||
def testUnaryRequestFutureUnaryResponse(self): |
||||
request = b'\x07\x08' |
||||
metadata = (('InVaLiD', 'UnaryRequestFutureUnaryResponse'),) |
||||
expected_error_details = "metadata was invalid: %s" % metadata |
||||
response_future = self._unary_unary.future(request, metadata=metadata) |
||||
with self.assertRaises(grpc.RpcError) as exception_context: |
||||
response_future.result() |
||||
self.assertEqual( |
||||
exception_context.exception.details(), expected_error_details) |
||||
self.assertEqual( |
||||
exception_context.exception.code(), grpc.StatusCode.INTERNAL) |
||||
self.assertEqual(response_future.details(), expected_error_details) |
||||
self.assertEqual(response_future.code(), grpc.StatusCode.INTERNAL) |
||||
|
||||
def testUnaryRequestStreamResponse(self): |
||||
request = b'\x37\x58' |
||||
metadata = (('InVaLiD', 'UnaryRequestStreamResponse'),) |
||||
expected_error_details = "metadata was invalid: %s" % metadata |
||||
response_iterator = self._unary_stream(request, metadata=metadata) |
||||
with self.assertRaises(grpc.RpcError) as exception_context: |
||||
next(response_iterator) |
||||
self.assertEqual( |
||||
exception_context.exception.details(), expected_error_details) |
||||
self.assertEqual( |
||||
exception_context.exception.code(), grpc.StatusCode.INTERNAL) |
||||
self.assertEqual(response_iterator.details(), expected_error_details) |
||||
self.assertEqual(response_iterator.code(), grpc.StatusCode.INTERNAL) |
||||
|
||||
def testStreamRequestBlockingUnaryResponse(self): |
||||
request_iterator = (b'\x07\x08' for _ in range(test_constants.STREAM_LENGTH)) |
||||
metadata = (('InVaLiD', 'StreamRequestBlockingUnaryResponse'),) |
||||
expected_error_details = "metadata was invalid: %s" % metadata |
||||
with self.assertRaises(ValueError) as exception_context: |
||||
self._stream_unary(request_iterator, metadata=metadata) |
||||
self.assertIn(expected_error_details, str(exception_context.exception)) |
||||
|
||||
def testStreamRequestBlockingUnaryResponseWithCall(self): |
||||
request_iterator = ( |
||||
b'\x07\x08' for _ in range(test_constants.STREAM_LENGTH)) |
||||
metadata = (('InVaLiD', 'StreamRequestBlockingUnaryResponseWithCall'),) |
||||
expected_error_details = "metadata was invalid: %s" % metadata |
||||
multi_callable = _stream_unary_multi_callable(self._channel) |
||||
with self.assertRaises(ValueError) as exception_context: |
||||
multi_callable.with_call(request_iterator, metadata=metadata) |
||||
self.assertIn(expected_error_details, str(exception_context.exception)) |
||||
|
||||
def testStreamRequestFutureUnaryResponse(self): |
||||
request_iterator = ( |
||||
b'\x07\x08' for _ in range(test_constants.STREAM_LENGTH)) |
||||
metadata = (('InVaLiD', 'StreamRequestFutureUnaryResponse'),) |
||||
expected_error_details = "metadata was invalid: %s" % metadata |
||||
response_future = self._stream_unary.future( |
||||
request_iterator, metadata=metadata) |
||||
with self.assertRaises(grpc.RpcError) as exception_context: |
||||
response_future.result() |
||||
self.assertEqual( |
||||
exception_context.exception.details(), expected_error_details) |
||||
self.assertEqual( |
||||
exception_context.exception.code(), grpc.StatusCode.INTERNAL) |
||||
self.assertEqual(response_future.details(), expected_error_details) |
||||
self.assertEqual(response_future.code(), grpc.StatusCode.INTERNAL) |
||||
|
||||
def testStreamRequestStreamResponse(self): |
||||
request_iterator = ( |
||||
b'\x07\x08' for _ in range(test_constants.STREAM_LENGTH)) |
||||
metadata = (('InVaLiD', 'StreamRequestStreamResponse'),) |
||||
expected_error_details = "metadata was invalid: %s" % metadata |
||||
response_iterator = self._stream_stream(request_iterator, metadata=metadata) |
||||
with self.assertRaises(grpc.RpcError) as exception_context: |
||||
next(response_iterator) |
||||
self.assertEqual( |
||||
exception_context.exception.details(), expected_error_details) |
||||
self.assertEqual( |
||||
exception_context.exception.code(), grpc.StatusCode.INTERNAL) |
||||
self.assertEqual(response_iterator.details(), expected_error_details) |
||||
self.assertEqual(response_iterator.code(), grpc.StatusCode.INTERNAL) |
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
unittest.main(verbosity=2) |
||||
@ -0,0 +1,64 @@ |
||||
# Copyright 2015, Google Inc. |
||||
# All rights reserved. |
||||
# |
||||
# Redistribution and use in source and binary forms, with or without |
||||
# modification, are permitted provided that the following conditions are |
||||
# met: |
||||
# |
||||
# * Redistributions of source code must retain the above copyright |
||||
# notice, this list of conditions and the following disclaimer. |
||||
# * Redistributions in binary form must reproduce the above |
||||
# copyright notice, this list of conditions and the following disclaimer |
||||
# in the documentation and/or other materials provided with the |
||||
# distribution. |
||||
# * Neither the name of Google Inc. nor the names of its |
||||
# contributors may be used to endorse or promote products derived from |
||||
# this software without specific prior written permission. |
||||
# |
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
require 'grpc' |
||||
|
||||
StatusCodes = GRPC::Core::StatusCodes |
||||
|
||||
describe StatusCodes do |
||||
# convert upper snake-case to camel case. |
||||
# e.g., DEADLINE_EXCEEDED -> DeadlineExceeded |
||||
def upper_snake_to_camel(name) |
||||
name.to_s.split('_').map(&:downcase).map(&:capitalize).join('') |
||||
end |
||||
|
||||
StatusCodes.constants.each do |status_name| |
||||
it 'there is a subclass of BadStatus corresponding to StatusCode: ' \ |
||||
"#{status_name} that has code: #{StatusCodes.const_get(status_name)}" do |
||||
camel_case = upper_snake_to_camel(status_name) |
||||
error_class = GRPC.const_get(camel_case) |
||||
# expect the error class to be a subclass of BadStatus |
||||
expect(error_class < GRPC::BadStatus) |
||||
|
||||
error_object = error_class.new |
||||
# check that the code matches the int value of the error's constant |
||||
status_code = StatusCodes.const_get(status_name) |
||||
expect(error_object.code).to eq(status_code) |
||||
|
||||
# check default parameters |
||||
expect(error_object.details).to eq('unknown cause') |
||||
expect(error_object.metadata).to eq({}) |
||||
|
||||
# check that the BadStatus factory for creates the correct |
||||
# exception too |
||||
from_factory = GRPC::BadStatus.new_status_exception(status_code) |
||||
expect(from_factory.is_a?(error_class)).to be(true) |
||||
end |
||||
end |
||||
end |
||||
Loading…
Reference in new issue