Function test_modify_message

test/mitmproxy/proxy/layers/test_websocket.py:227–241

(ws_testdata)

def test_modify_message(ws_testdata):
    tctx, playbook, flow = ws_testdata
    assert (
        playbook
        << websocket.WebsocketStartHook(flow)
        >> reply()
        >> DataReceived(tctx.server, b"\x81\x03foo")
        << websocket.WebsocketMessageHook(flow)
    )
    flow.websocket.messages[-1].content = flow.websocket.messages[-1].content.replace(
        b"foo", b"foobar"
    )
    playbook >> reply()
    playbook << SendData(tctx.client, b"\x81\x06foobar")
    assert playbook


def test_empty_message(ws_testdata):