Function test_server_aborts

test/mitmproxy/proxy/layers/http/test_http.py:781–808

Test the scenario where the server doesn't serve a response

(tctx, data)

    ],
)
def test_server_aborts(tctx, data):
    """Test the scenario where the server doesn't serve a response"""
    server = Placeholder(Server)
    flow = Placeholder(HTTPFlow)
    playbook = Playbook(http.HttpLayer(tctx, HTTPMode.regular), hooks=False)
    assert (
        playbook
        >> DataReceived(
            tctx.client,
            b"GET http://example.com/ HTTP/1.1\r\nHost: example.com\r\n\r\n",
        )
        << OpenConnection(server)
        >> reply(None)
        << SendData(server, b"GET / HTTP/1.1\r\nHost: example.com\r\n\r\n")
    )
    if data:
        playbook >> DataReceived(server, data)
    assert (
        playbook
        >> ConnectionClosed(server)
        << CloseConnection(server)
        << http.HttpErrorHook(flow)
        >> reply()
        << SendData(tctx.client, BytesMatching(b"502 Bad Gateway"))
        << CloseConnection(tctx.client)
    )
    assert flow().error
    assert not flow().live


@pytest.mark.parametrize("redirect", ["", "change-destination", "change-proxy"])