def _Cb(msg):
(method, args, version) = client.ParseRequest(msg)
self.assertEqual(method, "fn5")
self.assertEqual(args, "xyz")
return client.FormatResponse(True,
"F",
version=self.MY_LUXI_VERSION * 2)
def testParseRequest(self):
msg = serializer.DumpJson({
client.KEY_METHOD: "foo",
client.KEY_ARGS: ("bar", "baz", 123),
})
self.assertEqualValues(client.ParseRequest(msg),
("foo", ["bar", "baz", 123], None))
self.assertRaises(client.ProtocolError, client.ParseRequest,
"this\"is {invalid, ]json data")
# No dict
self.assertRaises(client.ProtocolError, client.ParseRequest,
serializer.DumpJson(123))
# Empty dict
self.assertRaises(client.ProtocolError, client.ParseRequest,
serializer.DumpJson({ }))
# No arguments
self.assertRaises(client.ProtocolError, client.ParseRequest,
serializer.DumpJson({ client.KEY_METHOD: "foo", }))
# No method
self.assertRaises(client.ProtocolError, client.ParseRequest,
serializer.DumpJson({ client.KEY_ARGS: [], }))
# No method or arguments
self.assertRaises(client.ProtocolError, client.ParseRequest,
serializer.DumpJson({ client.KEY_VERSION: 1, }))
def _Cb(msg):
(method, args, version) = client.ParseRequest(msg)
self.assertEqual(method, "fn9")
self.assertEqual(args, [])
err = errors.OpPrereqError("TestVer")
return client.FormatResponse(False, errors.EncodeException(err),
version=self.MY_LUXI_VERSION)
def testParseRequestWithVersion(self):
msg = serializer.DumpJson({
client.KEY_METHOD: "version",
client.KEY_ARGS: (["some"], "args", 0, "here"),
client.KEY_VERSION: 20100101,
})
self.assertEqualValues(client.ParseRequest(msg),
("version", [["some"], "args", 0, "here"], 20100101))
def RunTask(self, server, message, client):
"""Process the request.
"""
client_ops = ClientOps(server)
try:
(method, args, ver) = rpccl.ParseRequest(message)
except rpcerr.ProtocolError, err:
logging.error("Protocol Error: %s", err)
client.close_log()
return
def Call(self, data):
"""Calls LUXI method.
In this test class the method is not actually called, but added to a list
of called methods and then an exception (L{errors.RapiTestResult}) is
raised. There is no return value.
"""
(method, _, _) = rpccl.ParseRequest(data)
# Take a note of called method
self._record_fn(method)
# Everything went fine until here, so let's abort the test
raise errors.RapiTestResult
def _Cb(msg):
(method, args, version) = client.ParseRequest(msg)
self.assertEqual(method, "fn1")
self.assertEqual(args, "Hello World")
return client.FormatResponse(True, "x")
