Fix tests

Author: Stefan Machmeier

heifip/layers/dns.py

@@ -11,7 +11,8 @@
 class DNSPacket(TransportPacket):
     def __init__(self, packet: Packet, address_mapping={}, layer_map={}) -> None:
         TransportPacket.__init__(self, packet, address_mapping, layer_map)
-        self.hash = hashlib.md5(f"{self.packet[DNS].qr}".encode('utf-8')).hexdigest()
+        # if DNS in self.layer_map:
+        #     self.hash = hashlib.md5(f"{self.packet[DNS].qr}".encode('utf-8')).hexdigest()
 
     def header_preprocessing(self):
         # TODO: Fix issue with DNS processing

heifip/layers/http.py

@@ -19,8 +19,7 @@ def header_preprocessing(self):
 class HTTPRequestPacket(HTTPPacket):
     def __init__(self, packet: Packet, address_mapping={}, layer_map={}):
         HTTPPacket.__init__(self, packet, address_mapping, layer_map)
-        # self.hash = hashlib.md5(f"{self.packet[HTTPRequest].Path},{self.packet[HTTPRequest].Method},{self.packet[HTTPRequest].Accept}".encode('utf-8')).hexdigest()
-        self.hash = hashlib.md5(f"{self.packet[HTTPRequest].Method},{self.packet[HTTPRequest].Accept}".encode('utf-8')).hexdigest()
+        self.hash = hashlib.md5(f"{self.packet[HTTPRequest].Path},{self.packet[HTTPRequest].Method},{self.packet[HTTPRequest].Accept}".encode('utf-8')).hexdigest()
         if Raw in self.layer_map:
             self.packet[HTTPRequest].remove_payload()
 
@@ -49,8 +48,7 @@ def header_preprocessing(self):
 class HTTPResponsePacket(HTTPPacket):
     def __init__(self, packet: Packet, address_mapping={}, layer_map={}):
         HTTPPacket.__init__(self, packet, address_mapping, layer_map)
-        # self.hash = hashlib.md5(f"{self.packet[HTTPResponse].Server},{self.packet[HTTPResponse].Status_Code},{self.packet[HTTPResponse].Connection}".encode('utf-8')).hexdigest()
-        self.hash = hashlib.md5(f"{self.packet[HTTPResponse].Status_Code},{self.packet[HTTPResponse].Connection}".encode('utf-8')).hexdigest()
+        self.hash = hashlib.md5(f"{self.packet[HTTPResponse].Server},{self.packet[HTTPResponse].Status_Code},{self.packet[HTTPResponse].Connection}".encode('utf-8')).hexdigest()
         if Raw in self.layer_map:
             self.packet[HTTPResponse].remove_payload()