From c504bc6eb1b132cdbdefe8be94138492080706cd Mon Sep 17 00:00:00 2001 From: Manuel Meitinger Date: Tue, 29 Mar 2022 16:25:34 +0200 Subject: [PATCH] [dns] eventsequence test --- test/mitmproxy/test_eventsequence.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/test/mitmproxy/test_eventsequence.py b/test/mitmproxy/test_eventsequence.py index 03ad5da57..f6c2dc191 100644 --- a/test/mitmproxy/test_eventsequence.py +++ b/test/mitmproxy/test_eventsequence.py @@ -59,6 +59,22 @@ def test_tcp_flow(err): assert isinstance(next(i), layers.tcp.TcpEndHook) +@pytest.mark.parametrize("resp, err", [ + (False, False), + (True, False), + (False, True), + (True, True), +]) +def test_dns(resp, err): + f = tflow.tdnsflow(resp=resp, err=err) + i = eventsequence.iterate(f) + assert isinstance(next(i), layers.dns.DnsRequestHook) + if resp: + assert isinstance(next(i), layers.dns.DnsResponseHook) + if err: + assert isinstance(next(i), layers.dns.DnsErrorHook) + + def test_invalid(): with pytest.raises(TypeError): next(eventsequence.iterate(42))