Function test_long_response

test/mitmproxy/proxy/layers/http/test_http2.py:304–442

(tctx: Context, trailers)

@pytest.mark.parametrize("trailers", ["trailers", ""])
def test_long_response(tctx: Context, trailers):
    playbook, cff = start_h2_client(tctx)
    flow = Placeholder(HTTPFlow)
    server = Placeholder(Server)
    initial = Placeholder(bytes)
    assert (
        playbook
        >> DataReceived(
            tctx.client,
            cff.build_headers_frame(
                example_request_headers, flags=["END_STREAM"]
            ).serialize(),
        )
        << http.HttpRequestHeadersHook(flow)
        >> reply()
        << http.HttpRequestHook(flow)
        >> reply()
        << OpenConnection(server)
        >> reply(None, side_effect=make_h2)
        << SendData(server, initial)
    )
    frames = decode_frames(initial())
    assert [type(x) for x in frames] == [
        hyperframe.frame.SettingsFrame,
        hyperframe.frame.WindowUpdateFrame,
        hyperframe.frame.HeadersFrame,
    ]
    sff = FrameFactory()
    assert (
        playbook
        # a conforming h2 server would send settings first, we disregard this for now.
        >> DataReceived(
            server, sff.build_headers_frame(example_response_headers).serialize()
        )
        << http.HttpResponseHeadersHook(flow)
        >> reply()
        >> DataReceived(
            server, sff.build_data_frame(b"a" * 10000, flags=[]).serialize()
        )
        >> DataReceived(
            server,
            sff.build_data_frame(b"a" * 10000, flags=[]).serialize(),
        )
        >> DataReceived(
            server,
            sff.build_data_frame(b"a" * 10000, flags=[]).serialize(),
        )
        >> DataReceived(
            server,
            sff.build_data_frame(b"a" * 10000, flags=[]).serialize(),
        )
        >> DataReceived(
            server,
            sff.build_data_frame(b"a" * 10000, flags=[]).serialize(),
        )
        >> DataReceived(
            server,
            sff.build_data_frame(b"a" * 10000, flags=[]).serialize(),