def test_sort_pcap():
f = lambda x: os.path.join("tests", "files", "pcap", x)
h = lambda x: hashlib.sha1(open(x, "rb").read()).hexdigest()
filepath = tempfile.mktemp()
sort_pcap(f("smtp.pcap"), filepath)
assert h(filepath) == "c99b74eaca15d792049e8f75a4bfe6c1c416c26b"
filepath = tempfile.mktemp()
sort_pcap(f("duplicate-dns-requests.pcap"), filepath)
assert h(filepath) == "4859334accbee12858621cca39564fdbce9ae605"
filepath = tempfile.mktemp()
sort_pcap(f("mixed-traffic.pcap"), filepath)
assert h(filepath) == "8d92880f9f356d5bdfd04e24541f614f275b02e0"
filepath = tempfile.mktemp()
sort_pcap(f("not-http.pcap"), filepath)
assert h(filepath) == "340e6c442619de912253d956b499e428164e8cdd"
filepath = tempfile.mktemp()
sort_pcap(f("status-code.pcap"), filepath)
assert h(filepath) == "106dc235ef82ff844b7025339c2713aad752037e"
filepath = tempfile.mktemp()
sort_pcap(f("used_dns_server.pcap"), filepath)
assert h(filepath) == "782b766b99998fd8cbbe400b7c419abfe645b50d"
def test_sort_pcap_rm_temp(p):
filepath = tempfile.mktemp()
sort_pcap("tests/files/pcap/smtp.pcap", filepath)
p.assert_called_once()
assert p.call_args[0][0].startswith(tempfile.gettempdir())
