Gamercat
/
geneva
mirror of https://github.com/Kkevsterrr/geneva
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #53 from Mitochondrion2/patch-2 

added test_compression_strategy from dev
This commit is contained in:
Kevin Bock 2023-01-22 23:17:19 -05:00 committed by GitHub
parent 57e19c2d4c f759e9c938
commit 68f7b1ebdc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_compress.py
View File

@ -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]))