Как сказал @Vladimir, вы все равно можете использовать AsyncHTTPTestCase
для создания / управления тестовым экземпляром веб-сервера, но вы можете по-прежнему тестировать WebSockets так же, как и обычные запросы HTTP - есть только нет синтаксического сахара, чтобы помочь вам.
Tornado также имеет свой собственный клиент WebSocket, поэтому нет необходимости (насколько я видел) использовать сторонний клиент - возможно, это недавнее дополнение. Так что попробуйте что-то вроде:
import tornado
class TestWebSockets(tornado.testing.AsyncHTTPTestCase):
def get_app(self):
# Required override for AsyncHTTPTestCase, sets up a dummy
# webserver for this test.
app = tornado.web.Application([
(r'/path/to/websocket', MyWebSocketHandler)
])
return app
@tornado.testing.gen_test
def test_websocket(self):
# self.get_http_port() gives us the port of the running test server.
ws_url = "ws://localhost:" + str(self.get_http_port()) + "/path/to/websocket"
# We need ws_url so we can feed it into our WebSocket client.
# ws_url will read (eg) "ws://localhost:56436/path/to/websocket
ws_client = yield tornado.websocket.websocket_connect(ws_url)
# Now we can run a test on the WebSocket.
ws_client.write_message("Hi, I'm sending a message to the server.")
response = yield ws_client.read_message()
self.assertEqual(response, "Hi client! This is a response from the server.")
# ...etc
Надеюсь, это хорошая отправная точка в любом случае.