From 81b2e935c95887bae6f7467965e7ccd20cca86ac Mon Sep 17 00:00:00 2001 From: Kkevsterrr Date: Fri, 13 Mar 2020 08:44:04 -0400 Subject: [PATCH] Added timeout to sr1 calls --- tests/test_tamper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_tamper.py b/tests/test_tamper.py index e5fbeee..e4acb84 100644 --- a/tests/test_tamper.py +++ b/tests/test_tamper.py @@ -207,7 +207,7 @@ def test_decompress(): original = packet.copy() tamper.tamper(packet, logger) assert bytes(packet["DNS"]) == b'\x00\x00\x01\x00\x00\x02\x00\x00\x00\x00\x00\x00\x07minghui\xc0\x1a\x00\x01\x00\x01\x02ca\x00\x00\x01\x00\x01' - resp = sr1(packet.packet) + resp = sr1(packet.packet, timeout=5) assert resp["DNS"] assert resp["DNS"].rcode != 1 assert resp["DNSQR"] @@ -219,7 +219,7 @@ def test_decompress(): original = packet.copy() tamper.tamper(packet, logger) assert bytes(packet["DNS"]) == b'\x00\x00\x01\x00\x00\x02\x00\x00\x00\x00\x00\x00\x04maps\xc0\x17\x00\x01\x00\x01\x06google\x03com\x00\x00\x01\x00\x01' - resp = sr1(packet.packet) + resp = sr1(packet.packet, timeout=5) assert resp["DNS"] assert resp["DNS"].rcode != 1 assert resp["DNSQR"]