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()