def test_load_config(unused_tcp_port): fixed_port = range(unused_tcp_port, unused_tcp_port) serv_config = EvaluatorServerConfig(custom_port_range=fixed_port) expected_host = port_handler._get_ip_address() expected_port = unused_tcp_port expected_url = f"wss://{expected_host}:{expected_port}" expected_client_uri = f"{expected_url}/client" expected_dispatch_uri = f"{expected_url}/dispatch" assert serv_config.host == expected_host assert serv_config.port == expected_port assert serv_config.url == expected_url assert serv_config.client_uri == expected_client_uri assert serv_config.dispatch_uri == expected_dispatch_uri assert serv_config.token is not None assert serv_config.cert is not None sock = serv_config.get_socket() assert sock is not None assert not sock._closed sock.close() ee_config = EvaluatorServerConfig(custom_port_range=range(1024, 65535)) sock = ee_config.get_socket() assert sock is not None assert not sock._closed sock.close()
def test_load_config(unused_tcp_port): serv_config = EvaluatorServerConfig(unused_tcp_port) expected_host = _get_ip_address() expected_port = unused_tcp_port expected_url = f"wss://{expected_host}:{expected_port}" expected_client_uri = f"{expected_url}/client" expected_dispatch_uri = f"{expected_url}/dispatch" assert serv_config.host == expected_host assert serv_config.port == expected_port assert serv_config.url == expected_url assert serv_config.client_uri == expected_client_uri assert serv_config.dispatch_uri == expected_dispatch_uri assert serv_config.token is not None assert serv_config.cert is not None sock = serv_config.get_socket() assert sock is not None assert not sock._closed sock.close() ee_config = EvaluatorServerConfig() assert ee_config.port in range(51820, 51840) sock = ee_config.get_socket() assert sock is not None assert not sock._closed sock.close()
def test_secure_echo(ws): custom_port_range = range(1024, 65535) config = EvaluatorServerConfig(custom_port_range=custom_port_range) async def handler(websocket, path): msg = await websocket.recv() await websocket.send(msg) ws( config.host, config.port, handler, ssl=config.get_server_ssl_context(), sock=config.get_socket(), ) with ExitStack() as stack: duplexer = SyncWebsocketDuplexer( f"wss://{config.host}:{config.port}", f"wss://{config.host}:{config.port}", cert=config.cert, token=None, ) stack.callback(duplexer.stop) duplexer.send("Hello Secure World") assert next(duplexer.receive()) == "Hello Secure World"