Function test_inject_message

test/mitmproxy/proxy/layers/test_websocket.py:472–486

(ws_testdata)

def test_inject_message(ws_testdata):
    tctx, playbook, flow = ws_testdata
    assert (
        playbook
        << websocket.WebsocketStartHook(flow)
        >> reply()
        >> WebSocketMessageInjected(
            flow, WebSocketMessage(Opcode.TEXT, False, b"hello")
        )
        << websocket.WebsocketMessageHook(flow)
    )
    assert flow.websocket.messages[-1].content == b"hello"
    assert flow.websocket.messages[-1].from_client is False
    assert flow.websocket.messages[-1].injected is True
    assert playbook >> reply() << SendData(tctx.client, b"\x81\x05hello")