tornado/maint/test/websocket/client.py

33 lines
972 B
Python
Raw Normal View History

#!/usr/bin/env python
import sys
from tornado.options import options, define, parse_command_line
from twisted.python import log
from twisted.internet import reactor
from autobahn.fuzzing import FuzzingClientFactory
define('servers', type=str, multiple=True,
default=['Tornado=ws://localhost:9000'])
define('cases', type=str, multiple=True,
default=["*"])
define('exclude', type=str, multiple=True,
default=["9.*"])
if __name__ == '__main__':
parse_command_line()
log.startLogging(sys.stdout)
servers = []
for server in options.servers:
name, _, url = server.partition('=')
servers.append({"agent": name, "url": url, "options": {"version": 17}})
spec = {
"options": {"failByDrop": False},
"enable-ssl": False,
"servers": servers,
"cases": options.cases,
"exclude-cases": options.exclude,
"exclude-agent-cases": {},
}
fuzzer = FuzzingClientFactory(spec)
reactor.run()