def test_stop_shuts_down_the_ws_server(self, make_server, wsgi_app): # Given controller = WebSocketController() controller.start(8080) # When controller.stop() # Then make_server.return_value.shutdown.assert_called_once_with()
def test_stop_closes_the_websocket(self, make_server, wsgi_app): # Given controller = WebSocketController() controller.start(8080) ws = Mock() # When controller.on_web_socket_opened(ws) controller.stop() # Then ws.close.assert_called_once_with()
class Server(object): def __init__(self): self.web_server = HTTPServerController() self.web_socket = WebSocketController() def start(self, quiet=False): self.web_server.start() self.web_socket.start(self.web_server.http_port + 1) def stop(self): self.web_socket.stop() self.web_server.stop() def send_message(self, msg): self.web_socket.send_async(msg) @property def http_port(self): return self.web_server.http_port