Merge pull request #8 from kuenishi/master

add default serialization method for idl-defined message classes

Author: repeatedly

msgpackrpc/transport/tcp.py

@@ -11,7 +11,7 @@
 class BaseSocket(object):
     def __init__(self, stream, encodings):
         self._stream = stream
-        self._packer = msgpack.Packer(encoding=encodings[0])
+        self._packer = msgpack.Packer(encoding=encodings[0], default=lambda x: x.to_msgpack())
         self._unpacker = msgpack.Unpacker(encoding=encodings[1])
 
     def close(self):

test/test_msgpackrpc.py

@@ -8,6 +8,30 @@
 
 
 class TestMessagePackRPC(unittest.TestCase):
+
+    class TestArg:
+        ''' this class must know completely how to deserialize '''
+        def __init__(self, a, b, c):
+            self.a = a
+            self.b = b
+            self.c = c
+
+        def to_msgpack(self):
+            return (self.a, self.b, self.c)
+
+        def add(self, rhs):
+            self.a += rhs.a
+            self.b -= rhs.b
+            self.c *= rhs.c
+            return self
+
+        def __eq__(self, rhs):
+            return (self.a == rhs.a and self.b == rhs.b and self.c == rhs.c)
+
+        @staticmethod
+        def from_msgpack(arg):
+            return TestMessagePackRPC.TestArg(arg[0], arg[1], arg[2])
+
     class TestServer(object):
         def hello(self):
             return "world"
@@ -18,6 +42,11 @@ def sum(self, x, y):
         def nil(self):
             return None
 
+        def add_arg(self, arg0, arg1):
+            lhs = TestMessagePackRPC.TestArg.from_msgpack(arg0)
+            rhs = TestMessagePackRPC.TestArg.from_msgpack(arg1)
+            return lhs.add(rhs)
+
         def raise_error(self):
             raise Exception('error')
 
@@ -54,6 +83,21 @@ def test_call(self):
         self.assertEqual(result2, 3, "'sum' result is incorrect")
         self.assertIsNone(result3, "'nil' result is incorrect")
 
+    def test_call_userdefined_arg(self):
+        client = self.setup_env();
+
+        arg = TestMessagePackRPC.TestArg(0, 1, 2)
+        arg2 = TestMessagePackRPC.TestArg(23, 3, -23)
+
+        result1 = TestMessagePackRPC.TestArg.from_msgpack(client.call('add_arg',arg, arg2))
+        self.assertEqual(result1, arg.add(arg2))
+
+        result2 = TestMessagePackRPC.TestArg.from_msgpack(client.call('add_arg',arg2, arg))
+        self.assertEqual(result2, arg2.add(arg))
+
+        result3 = TestMessagePackRPC.TestArg.from_msgpack(client.call('add_arg',result1, result2))
+        self.assertEqual(result3, result1.add(result2))
+
     def test_call_async(self):
         client = self.setup_env();