Initial commit of Ccnet server.

python/ccnet/test-client.py

@@ -0,0 +1,62 @@
+import sys
+import os
+from ccnet import ClientPool, RpcClientBase
+from pysearpc import searpc_func
+import threading
+import logging
+
+RPC_SERVICE_NAME = 'test-rpcserver'
+CCNET_CONF_DIR = os.path.expanduser('~/.ccnet')
+
+class TestRpcClient(RpcClientBase):
+    def __init__(self, client_pool, *args, **kwargs):
+        RpcClientBase.__init__(self, client_pool, RPC_SERVICE_NAME, *args, **kwargs)
+
+    @searpc_func('string', ['string', 'int'])
+    def str_mul(self, s, n):
+        pass
+
+
+class Worker(threading.Thread):
+    def __init__(self, rpc):
+        threading.Thread.__init__(self)
+        self.rpc = rpc
+
+    def run(self):
+        s = 'abcdef'
+        n = 100
+        assert self.rpc.str_mul(s, n) == s * n
+
+def test(n):
+    rpcclient = TestRpcClient(ClientPool(CCNET_CONF_DIR, CCNET_CONF_DIR))
+
+    workers = []
+    for i in xrange(n):
+        t = Worker(rpcclient)
+        t.start()
+        workers.append(t)
+
+    for t in workers:
+        t.join()
+
+def setup_logging():
+    kw = {
+        'format': '[%(asctime)s][%(module)s]: %(message)s',
+        'datefmt': '%m/%d/%Y %H:%M:%S',
+        'level': logging.DEBUG,
+        'stream': sys.stdout,
+    }
+
+    logging.basicConfig(**kw)
+
+def main():
+    setup_logging()
+    if len(sys.argv) > 1:
+        test(int(sys.argv[1]))
+    else:
+        test(100)
+
+    print 'test passed'
+
+if __name__ == '__main__':
+    main()