ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

PATH: usr / lib / python2.7 / site-packages / google / protobuf / internal

#
# Protocol Buffers - Google's data interchange format
# Copyright 2008 Google Inc.  All rights reserved.
# https://developers.google.com/protocol-buffers/
#
# 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.

"""Tests for google.protobuf.internal.service_reflection."""

__author__ = 'petar@google.com (Petar Petrov)'

try:
  import unittest2 as unittest  #PY26
except ImportError:
  import unittest

from google.protobuf import unittest_pb2
from google.protobuf import service_reflection
from google.protobuf import service

class FooUnitTest(unittest.TestCase):

def testService(self):
    class MockRpcChannel(service.RpcChannel):
      def CallMethod(self, method, controller, request, response, callback):
        self.method = method
        self.controller = controller
        self.request = request
        callback(response)

class MockRpcController(service.RpcController):
      def SetFailed(self, msg):
        self.failure_message = msg

self.callback_response = None

class MyService(unittest_pb2.TestService):
      pass

self.callback_response = None

def MyCallback(response):
      self.callback_response = response

rpc_controller = MockRpcController()
    channel = MockRpcChannel()
    srvc = MyService()
    srvc.Foo(rpc_controller, unittest_pb2.FooRequest(), MyCallback)
    self.assertEqual('Method Foo not implemented.',
                     rpc_controller.failure_message)
    self.assertEqual(None, self.callback_response)

rpc_controller.failure_message = None

service_descriptor = unittest_pb2.TestService.GetDescriptor()
    srvc.CallMethod(service_descriptor.methods[1], rpc_controller,
                    unittest_pb2.BarRequest(), MyCallback)
    self.assertTrue(srvc.GetRequestClass(service_descriptor.methods[1]) is
                    unittest_pb2.BarRequest)
    self.assertTrue(srvc.GetResponseClass(service_descriptor.methods[1]) is
                    unittest_pb2.BarResponse)
    self.assertEqual('Method Bar not implemented.',
                     rpc_controller.failure_message)
    self.assertEqual(None, self.callback_response)

class MyServiceImpl(unittest_pb2.TestService):
      def Foo(self, rpc_controller, request, done):
        self.foo_called = True
      def Bar(self, rpc_controller, request, done):
        self.bar_called = True

srvc = MyServiceImpl()
    rpc_controller.failure_message = None
    srvc.Foo(rpc_controller, unittest_pb2.FooRequest(), MyCallback)
    self.assertEqual(None, rpc_controller.failure_message)
    self.assertEqual(True, srvc.foo_called)

rpc_controller.failure_message = None
    srvc.CallMethod(service_descriptor.methods[1], rpc_controller,
                    unittest_pb2.BarRequest(), MyCallback)
    self.assertEqual(None, rpc_controller.failure_message)
    self.assertEqual(True, srvc.bar_called)

def testServiceStub(self):
    class MockRpcChannel(service.RpcChannel):
      def CallMethod(self, method, controller, request,
                     response_class, callback):
        self.method = method
        self.controller = controller
        self.request = request
        callback(response_class())

self.callback_response = None

def MyCallback(response):
      self.callback_response = response

channel = MockRpcChannel()
    stub = unittest_pb2.TestService_Stub(channel)
    rpc_controller = 'controller'
    request = 'request'

# GetDescriptor now static, still works as instance method for compatibility
    self.assertEqual(unittest_pb2.TestService_Stub.GetDescriptor(),
                     stub.GetDescriptor())

# Invoke method.
    stub.Foo(rpc_controller, request, MyCallback)

self.assertIsInstance(self.callback_response, unittest_pb2.FooResponse)
    self.assertEqual(request, channel.request)
    self.assertEqual(rpc_controller, channel.controller)
    self.assertEqual(stub.GetDescriptor().methods[0], channel.method)

if __name__ == '__main__':
  unittest.main()

[+] ..
[-] more_extensions_pb2.pyc [edit]
[-] enum_type_wrapper.pyc [edit]
[-] unknown_fields_test.py [edit]
[-] well_known_types_test.pyo [edit]
[-] encoder.pyc [edit]
[-] message_listener.pyc [edit]
[-] message_factory_test.py [edit]
[-] test_util.pyc [edit]
[-] more_messages_pb2.pyo [edit]
[-] python_message.pyo [edit]
[-] message_factory_test.pyc [edit]
[-] reflection_test.pyc [edit]
[-] wire_format.pyc [edit]
[-] descriptor_pool_test2_pb2.pyo [edit]
[-] descriptor_pool_test1_pb2.pyc [edit]
[-] reflection_test.pyo [edit]
[-] decoder.pyc [edit]
[-] test_util.pyo [edit]
[-] api_implementation.pyo [edit]
[-] packed_field_test_pb2.pyc [edit]
[-] service_reflection_test.pyo [edit]
[-] descriptor_test.py [edit]
[-] descriptor_pool_test.pyc [edit]
[-] factory_test1_pb2.py [edit]
[-] factory_test2_pb2.pyc [edit]
[-] descriptor_pool_test.py [edit]
[-] factory_test1_pb2.pyo [edit]
[-] __init__.py [edit]
[-] more_extensions_dynamic_pb2.pyc [edit]
[-] well_known_types_test.py [edit]
[-] symbol_database_test.py [edit]
[-] message_set_extensions_pb2.py [edit]
[-] unknown_fields_test.pyc [edit]
[-] symbol_database_test.pyo [edit]
[-] any_test_pb2.pyc [edit]
[-] proto_builder_test.pyc [edit]
[-] descriptor_database_test.py [edit]
[-] text_format_test.py [edit]
[-] python_message.py [edit]
[-] __init__.pyo [edit]
[-] __init__.pyc [edit]
[-] testing_refleaks.py [edit]
[-] text_format_test.pyc [edit]
[-] _parameterized.py [edit]
[-] well_known_types.pyo [edit]
[-] containers.py [edit]
[-] api_implementation.pyc [edit]
[-] generator_test.pyc [edit]
[-] descriptor_pool_test2_pb2.pyc [edit]
[-] descriptor_test.pyo [edit]
[-] test_bad_identifiers_pb2.pyo [edit]
[-] well_known_types.pyc [edit]
[-] more_extensions_dynamic_pb2.pyo [edit]
[-] packed_field_test_pb2.pyo [edit]
[-] missing_enum_values_pb2.pyo [edit]
[-] json_format_test.pyo [edit]
[-] message_test.pyo [edit]
[-] proto_builder_test.py [edit]
[-] wire_format.py [edit]
[-] containers.pyo [edit]
[-] text_encoding_test.pyc [edit]
[-] python_message.pyc [edit]
[-] message_listener.pyo [edit]
[-] enum_type_wrapper.pyo [edit]
[+] import_test_package
[-] more_extensions_dynamic_pb2.py [edit]
[-] missing_enum_values_pb2.pyc [edit]
[-] descriptor_pool_test1_pb2.pyo [edit]
[-] message_listener.py [edit]
[-] descriptor_pool_test2_pb2.py [edit]
[-] service_reflection_test.py [edit]
[-] any_test_pb2.pyo [edit]
[-] generator_test.pyo [edit]
[-] testing_refleaks.pyc [edit]
[-] descriptor_test.pyc [edit]
[-] type_checkers.pyo [edit]
[-] reflection_test.py [edit]
[-] service_reflection_test.pyc [edit]
[-] text_encoding_test.py [edit]
[-] test_util.py [edit]
[-] well_known_types_test.pyc [edit]
[-] factory_test2_pb2.py [edit]
[-] message_set_extensions_pb2.pyc [edit]
[-] decoder.pyo [edit]
[-] file_options_test_pb2.pyc [edit]
[-] unknown_fields_test.pyo [edit]
[-] api_implementation.py [edit]
[-] descriptor_pool_test1_pb2.py [edit]
[-] more_messages_pb2.pyc [edit]
[-] enum_type_wrapper.py [edit]
[-] test_bad_identifiers_pb2.py [edit]
[-] more_messages_pb2.py [edit]
[-] wire_format_test.pyo [edit]
[-] message_factory_test.pyo [edit]
[-] text_encoding_test.pyo [edit]
[-] descriptor_database_test.pyo [edit]
[-] wire_format.pyo [edit]
[-] decoder.py [edit]
[-] symbol_database_test.pyc [edit]
[-] any_test_pb2.py [edit]
[-] descriptor_pool_test.pyo [edit]
[-] file_options_test_pb2.pyo [edit]
[-] factory_test2_pb2.pyo [edit]
[-] more_extensions_pb2.py [edit]
[-] packed_field_test_pb2.py [edit]
[-] type_checkers.pyc [edit]
[-] more_extensions_pb2.pyo [edit]
[-] file_options_test_pb2.py [edit]
[-] well_known_types.py [edit]
[-] proto_builder_test.pyo [edit]
[-] wire_format_test.py [edit]
[-] text_format_test.pyo [edit]
[-] json_format_test.py [edit]
[-] descriptor_database_test.pyc [edit]
[-] message_test.pyc [edit]
[-] json_format_test.pyc [edit]
[-] _parameterized.pyo [edit]
[-] type_checkers.py [edit]
[-] testing_refleaks.pyo [edit]
[-] _parameterized.pyc [edit]
[-] missing_enum_values_pb2.py [edit]
[-] containers.pyc [edit]
[-] wire_format_test.pyc [edit]
[-] message_test.py [edit]
[-] encoder.pyo [edit]
[-] factory_test1_pb2.pyc [edit]
[-] generator_test.py [edit]
[-] test_bad_identifiers_pb2.pyc [edit]
[-] encoder.py [edit]
[-] message_set_extensions_pb2.pyo [edit]