mirror of https://github.com/grpc/grpc.git
This introduces grpc_test.test_common for gRPC-specific test code and fixes issue 2570.pull/2686/head
parent
11af974715
commit
dde9071f9b
2 changed files with 77 additions and 6 deletions
@ -0,0 +1,71 @@ |
||||
# 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. |
||||
|
||||
"""Common code used throughout tests of gRPC.""" |
||||
|
||||
import collections |
||||
|
||||
|
||||
def metadata_transmitted(original_metadata, transmitted_metadata): |
||||
"""Judges whether or not metadata was acceptably transmitted. |
||||
|
||||
gRPC is allowed to insert key-value pairs into the metadata values given by |
||||
applications and to reorder key-value pairs with different keys but it is not |
||||
allowed to alter existing key-value pairs or to reorder key-value pairs with |
||||
the same key. |
||||
|
||||
Args: |
||||
original_metadata: A metadata value used in a test of gRPC. |
||||
transmitted_metadata: A metadata value corresponding to original_metadata |
||||
after having been transmitted via gRPC. |
||||
|
||||
Returns: |
||||
A boolean indicating whether transmitted_metadata accurately reflects |
||||
original_metadata after having been transmitted via gRPC. |
||||
""" |
||||
original = collections.defaultdict(list) |
||||
for key, value in original_metadata: |
||||
original[key].append(value) |
||||
transmitted = collections.defaultdict(list) |
||||
for key, value in transmitted_metadata: |
||||
transmitted[key].append(value) |
||||
|
||||
for key, values in original.iteritems(): |
||||
transmitted_values = transmitted[key] |
||||
transmitted_iterator = iter(transmitted_values) |
||||
try: |
||||
for value in values: |
||||
while True: |
||||
transmitted_value = next(transmitted_iterator) |
||||
if value == transmitted_value: |
||||
break |
||||
except StopIteration: |
||||
return False |
||||
else: |
||||
return True |
Loading…
Reference in new issue