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
