diff options
author | 2018-01-22 00:47:41 +0000 | |
---|---|---|
committer | 2018-01-26 07:09:28 +0000 | |
commit | c73758acb02d5c21f77e5144976fd5603c24f72e (patch) | |
tree | 1dbf6a4e110aa501c287e23ea6ccd6159d4aef3d /src/python/grpcio_tests/tests/unit/_metadata_test.py | |
parent | 31ddbff8cfed36d77ad34ec892af812dff614ce6 (diff) |
Elide cygrpc.ChannelArg and cygrpc.ChannelArgs
Diffstat (limited to 'src/python/grpcio_tests/tests/unit/_metadata_test.py')
-rw-r--r-- | src/python/grpcio_tests/tests/unit/_metadata_test.py | 16 |
1 files changed, 7 insertions, 9 deletions
diff --git a/src/python/grpcio_tests/tests/unit/_metadata_test.py b/src/python/grpcio_tests/tests/unit/_metadata_test.py index a918066ea4..5908421011 100644 --- a/src/python/grpcio_tests/tests/unit/_metadata_test.py +++ b/src/python/grpcio_tests/tests/unit/_metadata_test.py @@ -80,7 +80,7 @@ _TRAILING_METADATA = ( _EXPECTED_TRAILING_METADATA = _TRAILING_METADATA -def user_agent(metadata): +def _user_agent(metadata): for key, val in metadata: if key == 'user-agent': return val @@ -88,16 +88,14 @@ def user_agent(metadata): def validate_client_metadata(test, servicer_context): + invocation_metadata = servicer_context.invocation_metadata() test.assertTrue( - test_common.metadata_transmitted( - _EXPECTED_INVOCATION_METADATA, - servicer_context.invocation_metadata())) + test_common.metadata_transmitted(_EXPECTED_INVOCATION_METADATA, + invocation_metadata)) + user_agent = _user_agent(invocation_metadata) test.assertTrue( - user_agent(servicer_context.invocation_metadata()) - .startswith('primary-agent ' + _channel._USER_AGENT)) - test.assertTrue( - user_agent(servicer_context.invocation_metadata()) - .endswith('secondary-agent')) + user_agent.startswith('primary-agent ' + _channel._USER_AGENT)) + test.assertTrue(user_agent.endswith('secondary-agent')) def handle_unary_unary(test, request, servicer_context): |