From f759e9c938e87303b2b19713e955b3ee467a9c87 Mon Sep 17 00:00:00 2001 From: Frederick Sell <51059186+Mitochondrion2@users.noreply.github.com> Date: Mon, 16 Jan 2023 11:03:22 -0500 Subject: [PATCH] added test_compression_strategy from dev --- tests/test_compress.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/test_compress.py b/tests/test_compress.py index 7cddaf3..af08f27 100644 --- a/tests/test_compress.py +++ b/tests/test_compress.py @@ -3,6 +3,7 @@ import os import pytest import tempfile +import dns.resolver import actions.tree import actions.drop import actions.tamper @@ -24,3 +25,17 @@ def test_compression_strategy(logger): with engine.Engine(53, "[UDP:dport:53]-tamper{DNS:qd:compress}-|", server_side=False, environment_id="compress_test", output_directory=actions.utils.RUN_DIRECTORY, log_level=actions.utils.CONSOLE_LOG_LEVEL): os.system("dig @8.8.8.8 google.com") +def test_compression_strategy(logger): + """ + Tests dns compression strategy. + """ + with engine.Engine(53, "[UDP:dport:53]-tamper{DNS:qd:compress}-|", server_side=False, environment_id="compress_test", output_directory=actions.utils.RUN_DIRECTORY, log_level=actions.utils.CONSOLE_LOG_LEVEL): + resolver = dns.resolver.Resolver() + resolver.nameservers = ["8.8.8.8"] + resolver.lifetime = 3 + answer = resolver.query("google.com", "A") + + assert answer, "Resolver query returned None" + assert len(answer) > 0, "No answers" + assert answer[0], "No answer to DNS query" + print("Got answer: %s" % str(answer[0]))