Python UnitTest-Websocket服务器

Question

今天，我正在为我的应用程序创建一些单元测试：一个websocket客户端。

在现实世界中，ws服务器是家庭网络中的嵌入式PC。

现在，对于我的单元测试，我想创建一个假的ws服务器并将其用于测试客户端。 您能否建议我一些ws-server即插即用功能，可以在我的unittest设置中调用它并将其用于测试？

我试图使用Autobahn ws服务器，但是它不是即插即用的。它应该可以工作，但是我无法在单独的线程中正确处理它。

我的目标是测试客户端不要开发虚拟服务器。

您能为我提供一些简单易用的东西吗？

在此先感谢，Salvo

Answer 1

我编写的最小代码是为了避免阻塞命令（serve_forever）

我用ws4py作为websocket库。

from wsgiref.simple_server import make_server
from ws4py.websocket import WebSocket
from ws4py.server.wsgirefserver import WSGIServer, WebSocketWSGIRequestHandler
from ws4py.server.wsgiutils import WebSocketWSGIApplication
import threading


class TestWebSocket(WebSocket):
    def received_message(self, message):
        self.send("+OK", False)


class TestServer:
    def __init__(self, hostname='127.0.0.1', port=8080):
        self.server = make_server(hostname,\
                             port,\
                             server_class=WSGIServer,\
                             handler_class=WebSocketWSGIRequestHandler,\
                             app=WebSocketWSGIApplication(handler_cls=TestWebSocket)\
                             )
        self.server.initialize_websockets_manager()
        self.thread = threading.Thread(target=self.server.serve_forever)
        self.thread.start()
        print("Server started for {}:{}".format(hostname, str(port)))

    def shutdown(self):
        self.server.shutdown()