Mirror/masscanned
1
0
Fork 0
mirror of https://github.com/ivre/masscanned.git synced 2025-10-02 06:38:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tests: add values to assert messages

Browse source
This commit is contained in:
Pierre Lalet 2024-10-14 02:15:23 +02:00
parent f5d40dd1a8
commit c852680930
1 changed files with 15 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

20
test/src/tests/dns.py
View file

@ -44,8 +44,12 @@ def test_ipv4_udp_dns_in_a():
check_ip_checksum(resp) check_ip_checksum(resp)
assert UDP in resp, "no UDP layer found" assert UDP in resp, "no UDP layer found"
udp = resp[UDP] udp = resp[UDP]
assert udp.sport == dport, "unexpected UDP sport: {}".format(udp.sport) assert (
assert udp.dport == sport, "unexpected UDP dport: {}".format(udp.dport) udp.sport == dport
), f"unexpected UDP sport: {udp.sport!r} ({domain})"
assert (
udp.dport == sport
), f"unexpected UDP dport: {udp.dport!r} ({domain})"
if DNS not in udp: if DNS not in udp:
try: try:
dns_rep = DNS(udp.load) dns_rep = DNS(udp.load)
@ -53,7 +57,9 @@ def test_ipv4_udp_dns_in_a():
raise AssertionError("no DNS layer found") raise AssertionError("no DNS layer found")
else: else:
dns_rep = udp[DNS] dns_rep = udp[DNS]
assert dns_rep.id == 1234, f"unexpected id value: {dns_rep.id}" assert (
dns_rep.id == 1234
), f"unexpected id value: {dns_rep.id!r} ({domain})"
assert dns_rep.qr, "unexpected qr value" assert dns_rep.qr, "unexpected qr value"
assert dns_rep.opcode == 0, "unexpected opcode value" assert dns_rep.opcode == 0, "unexpected opcode value"
assert dns_rep.aa, "unexpected aa value" assert dns_rep.aa, "unexpected aa value"
@ -62,7 +68,9 @@ def test_ipv4_udp_dns_in_a():
assert not dns_rep.ra, "unexpected ra value" assert not dns_rep.ra, "unexpected ra value"
assert dns_rep.z == 0, "unexpected z value" assert dns_rep.z == 0, "unexpected z value"
assert dns_rep.rcode == 0, "unexpected rcode value" assert dns_rep.rcode == 0, "unexpected rcode value"
assert dns_rep.qdcount == 1, "unexpected qdcount value" assert (
dns_rep.qdcount == 1
), f"unexpected qdcount value: {dns_rep.qdcount!r} vs 1 ({domain})"
assert dns_rep.ancount == 1, "unexpected ancount value" assert dns_rep.ancount == 1, "unexpected ancount value"
assert dns_rep.nscount == 0, "unexpected nscount value" assert dns_rep.nscount == 0, "unexpected nscount value"
assert dns_rep.arcount == 0, "unexpected arcount value" assert dns_rep.arcount == 0, "unexpected arcount value"
@ -125,7 +133,9 @@ def test_ipv4_udp_dns_in_a_multiple_queries():
assert not dns_rep.ra, "unexpected ra value" assert not dns_rep.ra, "unexpected ra value"
assert dns_rep.z == 0, "unexpected z value" assert dns_rep.z == 0, "unexpected z value"
assert dns_rep.rcode == 0, "unexpected rcode value" assert dns_rep.rcode == 0, "unexpected rcode value"
assert dns_rep.qdcount == 3, "unexpected qdcount value" assert (
dns_rep.qdcount == 3
), f"unexpected qdcount value: {dns_rep.qdcount} vs 3"
assert dns_rep.ancount == 3, "unexpected ancount value" assert dns_rep.ancount == 3, "unexpected ancount value"
assert dns_rep.nscount == 0, "unexpected nscount value" assert dns_rep.nscount == 0, "unexpected nscount value"
assert dns_rep.arcount == 0, "unexpected arcount value" assert dns_rep.arcount == 0, "unexpected arcount value"