Rooba
/
mitmproxy
mirror of https://github.com/mitmproxy/mitmproxy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[dns] eventsequence test

This commit is contained in:
Manuel Meitinger 2022-03-29 16:25:34 +02:00
parent fc0bdc143c
commit c504bc6eb1
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
test/mitmproxy/test_eventsequence.py
View File

@ -59,6 +59,22 @@ def test_tcp_flow(err):
assert isinstance(next(i), layers.tcp.TcpEndHook) 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(): def test_invalid():
with pytest.raises(TypeError): with pytest.raises(TypeError):
next(eventsequence.iterate(42)) next(eventsequence.iterate(42))