Python ServerState Examples

Example #1

File: test_auto_detection.py Project: vpistis/uvicorn

def test_http_auto():
    config = Config(app=None)
    server_state = ServerState()
    protocol = AutoHTTPProtocol(config=config, server_state=server_state)
    if httptools is None:
        assert type(protocol).__name__ == "H11Protocol"
    else:
        assert type(protocol).__name__ == "HttpToolsProtocol"

Example #2

File: test_auto_detection.py Project: vpistis/uvicorn

def test_websocket_auto():
    config = Config(app=None)
    server_state = ServerState()
    protocol = AutoWebSocketsProtocol(config=config, server_state=server_state)
    if websockets is None:
        assert type(protocol).__name__ == "WSProtocol"
    else:
        assert type(protocol).__name__ == "WebSocketProtocol"

Example #3

def get_connected_protocol(app, protocol_cls, **kwargs):
    loop = MockLoop()
    transport = MockTransport()
    config = Config(app=app, **kwargs)
    server_state = ServerState()
    protocol = protocol_cls(config=config, server_state=server_state, _loop=loop)
    protocol.connection_made(transport)
    return protocol

Example #4

File: test_http.py Project: stiiin/uvicorn

def get_connected_protocol(app, protocol_cls, event_loop, **kwargs):
    loop = MockLoop(event_loop)
    transport = MockTransport()
    config = Config(app=app, **kwargs)
    server_state = ServerState()
    protocol = protocol_cls(config=config, server_state=server_state, _loop=loop)
    protocol.connection_made(transport)
    yield protocol
    protocol.loop.close()

Example #5

def get_connected_protocol(app, protocol_cls, event_loop, **kwargs):
    loop = MockLoop(event_loop)
    asyncio._set_running_loop(loop)
    transport = MockTransport()
    config = Config(app=app, **kwargs)
    server_state = ServerState()
    protocol = protocol_cls(config=config, server_state=server_state, _loop=loop)
    protocol.connection_made(transport)
    try:
        yield protocol
    finally:
        protocol.loop.close()
        asyncio._set_running_loop(None)

Example #6

File: test_websocket.py Project: taoru/uvicorn

def run_server(app, protocol_cls, path="/"):
    asyncio.set_event_loop(None)
    loop = asyncio.new_event_loop()
    config = Config(app=app, ws=protocol_cls)
    server_state = ServerState()
    protocol = functools.partial(H11Protocol, config=config, server_state=server_state)
    create_server_task = loop.create_server(protocol, host="127.0.0.1")
    server = loop.run_until_complete(create_server_task)
    port = server.sockets[0].getsockname()[1]
    url = "ws://127.0.0.1:{port}{path}".format(port=port, path=path)
    try:
        # Run the event loop in a new thread.
        thread = threading.Thread(target=run_loop, args=[loop])
        thread.start()
        # Return the contextmanager state.
        yield url
    finally:
        # Close the loop from our main thread.
        while server_state.tasks:
            time.sleep(0.01)
        loop.call_soon_threadsafe(loop.stop)
        thread.join()

Example #7

async def test_websocket_auto():
    config = Config(app=app)
    server_state = ServerState()
    protocol = AutoWebSocketsProtocol(config=config, server_state=server_state)
    expected_websockets = "WSProtocol" if websockets is None else "WebSocketProtocol"
    assert type(protocol).__name__ == expected_websockets

Example #8

async def test_http_auto():
    config = Config(app=app)
    server_state = ServerState()
    protocol = AutoHTTPProtocol(config=config, server_state=server_state)
    expected_http = "H11Protocol" if httptools is None else "HttpToolsProtocol"
    assert type(protocol).__name__ == expected_http

Example #9

def test_http_auto():
    config = Config(app=None)
    server_state = ServerState()
    protocol = AutoHTTPProtocol(config=config, server_state=server_state)
    assert isinstance(protocol, HttpToolsProtocol)