added autobahn test server support

Lawouach · Lawouach · commit 133cafa7ce33 · 2012-03-23T21:54:57.000+01:00
diff --git a/test/autobahn_test_servers.py b/test/autobahn_test_servers.py
@@ -9,6 +9,7 @@ def run_cherrypy_server(host="127.0.0.1", port=9000):
     
     cherrypy.config.update({'server.socket_host': host,
                             'server.socket_port': port,
+                            'engine.autoreload_on': False, 
                             'log.screen': False})
     WebSocketPlugin(cherrypy.engine).subscribe()
     cherrypy.tools.websocket = WebSocketTool()
@@ -55,7 +56,26 @@ def on_message(self, message):
     logger.warning("Serving Tornado server on %s:%s" % (host, port))
     ioloop.IOLoop.instance().start()
 
+def run_autobahn_server(host="127.0.0.1", port=9003):
+    from autobahntestsuite import choosereactor
+    import autobahn
+    from autobahn.websocket import listenWS
+    from twisted.internet import reactor
+    from autobahn.websocket import WebSocketServerFactory, \
+         WebSocketServerProtocol
 
+    class ServerProtocol(WebSocketServerProtocol):
+        def onMessage(self, msg, binary):
+            self.sendMessage(msg, binary)
+
+    class ServerFactory(WebSocketServerFactory):
+        protocol = ServerProtocol
+
+    factory = ServerFactory("ws://%s:%d" % (host, port))
+    factory.setProtocolOptions(failByDrop=False)
+    logger.warning("Serving Autobahn server on %s:%s" % (host, port))
+    listenWS(factory, None)
+    reactor.run()
 
 if __name__ == '__main__':
     import argparse
@@ -74,12 +94,15 @@ def on_message(self, message):
                         help='Run the gevent server backend')
     parser.add_argument('--run-tornado-server', dest='run_tornado', action='store_true',
                         help='Run the Tornado server backend')
+    parser.add_argument('--run-autobahn-server', dest='run_autobahn', action='store_true',
+                        help='Run the Autobahn server backend')
     args = parser.parse_args()
 
     if args.run_all:
         args.run_cherrypy = True
         args.run_gevent = True
         args.run_tornado = True
+        args.run_autobahn = True
 
     procs = []
     logger.warning("CherryPy server: %s" % args.run_cherrypy)
@@ -100,6 +123,12 @@ def on_message(self, message):
         p2.daemon = True
         procs.append(p2)
 
+    logger.warning("Autobahn server: %s" % args.run_autobahn)
+    if args.run_autobahn:
+        p3 = Process(target=run_autobahn_server)
+        p3.daemon = True
+        procs.append(p3)
+
     for p in procs:
         p.start()
         logging.info("Starting process... %d" % p.pid)
