Merge pull request #3 from takano32/master

rewrite test as an example

Author: methane

example/echoserver.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+"""Echo service.
+This server using msgpackrpc.Server.
+"""
+
+import msgpackrpc
+
+class EchoHandler(object):
+
+    def echo(self, msg):
+        return msg
+
+def serve_background(server, daemon=False):
+    def _start_server(server):
+        server.start()
+        server.close()
+
+    import threading
+    t = threading.Thread(target=_start_server, args = (server,))
+    t.setDaemon(daemon)
+    t.start()
+    return t
+
+def serve(daemon=False):
+    """Serve echo server in background on localhost.
+    This returns (server, port). port is number in integer.
+
+    To stop, use ``server.shutdown()``
+    """
+    for port in xrange(9000, 10000):
+        try:
+            addr = msgpackrpc.Address('localhost', port)
+            server = msgpackrpc.Server(EchoHandler())
+            print server
+            server.listen(addr)
+            thread = serve_background(server, daemon)
+            return (addr, server, thread)
+        except Exception as err:
+            print err
+            pass
+
+if __name__ == '__main__':
+    port = serve(False)
+    print "Serving on localhost:%d\n" % port[1]
+

example/test_client.py

@@ -0,0 +1,40 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+import msgpackrpc
+import echoserver
+
+ADDR = SERVER = THREAD = None
+
+
+def setup():
+    global ADDR, SERVER, THREAD
+    (ADDR, SERVER, THREAD) = echoserver.serve()
+
+
+def teardown():
+    global SERVER, THREAD
+    SERVER.stop()
+    THREAD.join()
+
+
+def test_client():
+    global ADDR
+    client = msgpackrpc.Client(ADDR, unpack_encoding = 'utf-8')
+
+    f1 = client.call('echo', 'foo')
+    f2 = client.call('echo', 'bar')
+    f3 = client.call('echo', 'baz')
+
+    assert f2 == 'bar'
+    assert f1 == 'foo'
+    assert f3 == 'baz'
+
+    print "EchoHandler#echo via msgpackrpc"
+
+
+if __name__ == '__main__':
+    setup()
+    test_client()
+    teardown()
+