Function test_regular_hook_err

test/mitmproxy/proxy/layers/test_dns.py:179–206

(tctx)

def test_regular_hook_err(tctx):
    f = Placeholder(DNSFlow)

    req = tdnsreq()

    def err(flow: DNSFlow):
        flow.error = terr()

    assert (
        Playbook(dns.DNSLayer(tctx))
        >> DataReceived(
            tctx.client, dns.pack_message(req, tctx.client.transport_protocol)
        )
        << dns.DnsRequestHook(f)
        >> reply(side_effect=err)
        << 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().error
    assert not f().live


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