def test_pcap(self):
with open(os.path.join("tests", "pcaps", self.pcapfile), "rb") as f:
reader = PcapReader(f)
reader.tcp = TCPPacketStreamer(reader, self.handlers)
reader.raise_exceptions = self.use_exceptions
output = [self.format(*stream) for stream in reader.process()]
assert self.expected_output == output
def get_output(self, pcap):
reader = PcapReader(pcap)
reader.raise_exceptions = self.use_exceptions
list(reader.process())
key, exception = reader.exceptions.popitem()
output = [
exception["exception"],
os.path.basename(os.path.normpath(exception["trace"][-1][0]))
]
return output
def test_pcap(self):
with open(os.path.join("tests", "pcaps", self.pcapfile), "rb") as f:
reader = PcapReader(f)
reader.tcp = TCPPacketStreamer(reader, self.handlers)
reader.tlsinfo = self.tlsinfo
reader.raise_exceptions = self.use_exceptions
output = [
self.format(*stream) for stream in reader.process()
]
assert len(self.expected_output) == len(output) and sorted(self.expected_output,key=lambda x: "" if x is None else str(x)) == sorted(output,key=lambda x: "" if x is None else str(x))
