Add more test cases

Author: repeatedly

msgpackrpc/server.py

@@ -44,7 +44,7 @@ def dispatch(self, method, param, responder):
             if inPy3k:
                 method = method.decode("utf-8")
             if not hasattr(self._dispatcher, method):
-                raise error.NoMethodError("{0} not found".format(method))
+                raise error.NoMethodError("'{0}' method not found".format(method))
             responder.set_result(getattr(self._dispatcher, method)(*param))
         except Exception as e:
             responder.set_error(str(e))

test/test_msgpackrpc.py

@@ -4,6 +4,7 @@
 import helper
 import msgpackrpc
 from msgpackrpc import inPy3k
+from msgpackrpc import error
 
 
 class TestMessagePackRPC(unittest.TestCase):
@@ -28,9 +29,6 @@ def _start_server(server):
         self._thread = threading.Thread(target=_start_server, args=(self._server,))
         self._thread.start()
 
-        import time
-        time.sleep(1)
-
         self._client = msgpackrpc.Client(self._address)
         return self._client;
 
@@ -39,17 +37,49 @@ def tearDown(self):
         self._server.stop();
         self._thread.join();
 
-    def test_hello(self):
+    def test_call(self):
         client = self.setup_env();
         result = client.call('hello')
         if inPy3k:
             result = result.decode("utf-8")
-        self.assertEqual(result, "world", "hello result is incorrect")
+        self.assertEqual(result, "world", "'hello' result is incorrect")
 
-    def test_add(self):
-        client = self.setup_env();
         result = client.call('sum', 1, 2)
-        self.assertEqual(result, 3, "sum result is incorrect")
+        self.assertEqual(result, 3, "'sum' result is incorrect")
+
+    def test_call_async(self):
+        client = self.setup_env();
+
+        feture1 = client.call_async('hello')
+        feture2 = client.call_async('sum', 1, 2)
+        feture1.join()
+        feture2.join()
+
+        if inPy3k:
+            result = feture1.result.decode("utf-8")
+        else:
+            result = feture1.result
+        self.assertEqual(result, "world", "'hello' result is incorrect in call_async")
+        self.assertEqual(feture2.result, 3, "'sum' result is incorrect in call_async")
+
+    def test_notify(self):
+        client = self.setup_env();
+        result = True
+        try:
+            client.notify('hello')
+            client.notify('sum', 1, 2)
+        except:
+            result = False
+        self.assert_(result)
+
+    def test_unknown_method(self):
+        client = self.setup_env();
+        self.assertRaises(error.RPCError, lambda: client.call('unknown', True))
+        try:
+            client.call('unknown', True)
+            self.assert_(False)
+        except error.RPCError as e:
+            self.assertEqual(e.message, "'unknown' method not found", "Error message mismatched")
 
 
 if __name__ == '__main__':