def testAsync_WithRequest(self):
stub = BasicService.Stub(self.transport)
request = SimpleRequest()
request.param1 = 'val1'
request.param2 = 'val2'
response = SimpleResponse()
rpc = transport.Rpc(request)
self.transport.send_rpc(BasicService.remote_method.remote,
request).AndReturn(rpc)
self.mox.ReplayAll()
self.assertEquals(rpc, stub.proto_async.remote_method(request))
self.mox.VerifyAll()
def testSync_WithKwargs(self):
stub = BasicService.Stub(self.transport)
request = SimpleRequest()
request.param1 = 'val1'
request.param2 = 'val2'
response = SimpleResponse()
rpc = transport.Rpc(request)
rpc.set_response(response)
self.transport.send_rpc(BasicService.remote_method.remote,
request).AndReturn(rpc)
self.mox.ReplayAll()
self.assertEquals(SimpleResponse(),
stub.remote_method(param1='val1', param2='val2'))
self.mox.VerifyAll()
