示例#1
0
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()
示例#2
0
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()
示例#3
0
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"