Function test_reverse_fail_connection

test/mitmproxy/proxy/layers/test_dns.py:250–283

(tctx, transport_protocol)

@pytest.mark.parametrize("transport_protocol", ["tcp", "udp"])
def test_reverse_fail_connection(tctx, transport_protocol):
    tctx.client.transport_protocol = transport_protocol
    tctx.server.transport_protocol = transport_protocol

    f = Placeholder(DNSFlow)
    layer = dns.DNSLayer(tctx)
    layer.context.server.address = ("8.8.8.8", 53)

    req = tdnsreq()

    assert (
        Playbook(layer)
        >> DataReceived(
            tctx.client, dns.pack_message(req, tctx.client.transport_protocol)
        )
        << dns.DnsRequestHook(f)
        >> reply()
        << OpenConnection(tctx.server)
        >> reply("UDP no likey today.")
        << dns.DnsErrorHook(f)
        >> reply()
        << SendData(
            tctx.client,
            dns.pack_message(
                req.fail(response_codes.SERVFAIL), tctx.client.transport_protocol
            ),
        )
        << None
    )
    assert f().request
    assert not f().response
    assert f().error.msg == "UDP no likey today."
    req.timestamp = f().request.timestamp
    assert f().request == req


@pytest.mark.parametrize("transport_protocol", ["tcp", "udp"])