def test_insert_mongo(self): mongoHandler = MongoDBHandler() mongoHandler.handle(self.capture) self.capture.file.open() pcapReader = dpkt.pcap.Reader(self.capture.file) packetList = [] dbList = [] db = mongodb.db packets = db.packets for timestamp, packet in pcapReader: jsonObj = generate_json(self.capture, packet, timestamp) packetList.append(jsonObj) for packet in packets.find(): dbList.append(packet) # check whether inserted packets exist & correspond to the inserted ones self.assertEqual(len(packetList), len(dbList)) self.assertEqual(packetList[0]["mac_address_src"], dbList[0]["mac_address_src"]) self.assertEqual(packetList[1]["mac_address_src"], dbList[1]["mac_address_src"]) self.assertEqual(packetList[2]["mac_address_src"], dbList[2]["mac_address_src"]) self.assertEqual(packetList[0]["tags"], dbList[0]["tags"])
def test_unpack_pcap(self): pcapReader = dpkt.pcap.Reader(self.capture.file) packetList = [] for timestamp, packet in pcapReader: jsonObj = generate_json(self.capture, packet, timestamp) packetList.append(jsonObj) self.assertEqual(packetList[0]["capture_uuid"], self.capture.uuid) self.assertEqual(packetList[0]["mac_address_src"], "cc08e06156a1") self.assertEqual(packetList[0]["ssid"], "") self.assertEqual(packetList[0]["signal_strength"], -83) self.assertEqual(packetList[0]["mac_address_dst"], "ffffffffffff") self.assertEqual(packetList[1]["capture_uuid"], self.capture.uuid) self.assertEqual(packetList[1]["mac_address_src"], "0024d6799d8e") self.assertEqual(packetList[1]["ssid"], "public") self.assertEqual(packetList[1]["signal_strength"], -62) self.assertEqual(packetList[1]["mac_address_dst"], "ffffffffffff") self.assertEqual(packetList[2]["capture_uuid"], self.capture.uuid) self.assertEqual(packetList[2]["mac_address_src"], "e8150e460364") self.assertEqual(packetList[2]["ssid"], "eduroam") self.assertEqual(packetList[2]["signal_strength"], -49) self.assertEqual(packetList[1]["mac_address_dst"], "ffffffffffff")