def dns_qd_question_ipExist(self): src_ref = '181.149.152.176' dst_ref = '125.195.213.93' src_rev = '176.152.149.181.in-addr.arpa' ref_pkt = dns.DNS(qdcount=1, qd=[DNSQR(qtype=12, qname='79.255.233.124.in-addr.arpa')]) mac_pkt = dns.DNS(qdcount=1, qd=[DNSQR(qtype=12, qname=src_rev)]) data = lib.build_mock_dict() pp.dns_change_ips(mac_pkt, data) self.assertTrue( lib.compare_mac_pkts(ref_pkt, mac_pkt), msg= (str(ref_pkt) + ' == ' + str(mac_pkt)) )
def dns_qd_questions_ipNotExist(self): src_ref = '83.78.233.252' dst_ref = '125.195.213.93' src_rev = '252.233.78.83.in-addr.arpa' src_rev = '93.213.195.125.in-addr.arpa' ref_pkt = dns.DNS(qdcount=2, qd=[DNSQR(qtype=12, qname=src_rev), DNSQR(qtype=12, qname=src_rev) ]) mac_pkt = dns.DNS(qdcount=2, qd=[DNSQR(qtype=12, qname=src_rev), DNSQR(qtype=12, qname=src_rev) ]) data = lib.build_mock_dict() pp.dns_change_ips(mac_pkt, data) self.assertTrue( lib.compare_mac_pkts(ref_pkt, mac_pkt), msg= (str(ref_pkt) + ' == ' + str(mac_pkt)) )