Function test_regular_mode_no_hook

test/mitmproxy/proxy/layers/test_dns.py:105–144

(tctx, transport_protocol)

@pytest.mark.parametrize("transport_protocol", ["tcp", "udp"])
def test_regular_mode_no_hook(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 = None

    req = tdnsreq()

    def no_resolve(flow: DNSFlow):
        nonlocal req
        assert flow.request
        req.timestamp = flow.request.timestamp
        assert flow.request == req

    assert (
        Playbook(layer)
        >> DataReceived(
            tctx.client, dns.pack_message(req, tctx.client.transport_protocol)
        )
        << dns.DnsRequestHook(f)
        >> reply(side_effect=no_resolve)
        << dns.DnsErrorHook(f)
        >> reply()
        << SendData(
            tctx.client,
            dns.pack_message(
                req.fail(response_codes.SERVFAIL), tctx.client.transport_protocol
            ),
        )
        >> ConnectionClosed(tctx.client)
        << None
    )
    assert f().request == req
    assert not f().response
    assert not f().live
    assert (
        f().error.msg == "No hook has set a response and there is no upstream server."
    )


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