def test_does_not_have_info(self):
info = {}
a = Packet(info, 1000)
with self.assertRaises(PacketWithoutInfoError):
a.id
with self.assertRaises(PacketWithoutInfoError):
a.qname
with self.assertRaises(PacketWithoutInfoError):
a.query
with self.assertRaises(PacketWithoutInfoError):
a.source
with self.assertRaises(PacketWithoutInfoError):
a.dest
with self.assertRaises(PacketWithoutInfoError):
a.dest
with self.assertRaises(PacketWithoutInfoError):
a.is_answer()
with self.assertRaises(PacketWithoutInfoError):
a.is_critical_type()
def test_methods(self):
queryProblematic = {'dest': 'encrypted(dnsip1)',
'source': 'encrypted(ip1)',
'flags': '0',
'queries': [
{'qname': 'www.ni_c.cl', 'qtype': '1'}]}
answer = {'dest': 'encrypted(ip3)',
'source': 'encrypted(dnsip3)',
'flags': '8000',
'queries': [
{'qname': 'www.jerry.cl', 'qtype': '5'}]}
q = Packet(queryProblematic)
a = Packet(answer)
self.assertTrue(q.is_critical_type())
self.assertTrue(a.is_answer())
self.assertFalse(a.is_critical_type())
self.assertFalse(q.is_answer())
