|
|
|
@ -64,6 +64,24 @@ class FooUnitTest(unittest.TestCase): |
|
|
|
|
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): |
|
|
|
|