Function ws_testdata

test/mitmproxy/proxy/layers/test_websocket.py:203–224

(tctx)

@pytest.fixture()
def ws_testdata(tctx):
    tctx.server.address = ("example.com", 80)
    tctx.server.state = ConnectionState.OPEN
    flow = HTTPFlow(tctx.client, tctx.server)
    flow.request = Request.make(
        "GET",
        "http://example.com/",
        headers={
            "Connection": "upgrade",
            "Upgrade": "websocket",
            "Sec-WebSocket-Version": "13",
        },
    )
    flow.response = Response.make(
        101,
        headers={
            "Connection": "upgrade",
            "Upgrade": "websocket",
        },
    )
    flow.websocket = WebSocketData()
    return tctx, Playbook(websocket.WebsocketLayer(tctx, flow)), flow


def test_modify_message(ws_testdata):