From fe229e9130697c96a7e774ee05287b3db57f3c8f Mon Sep 17 00:00:00 2001 From: aouinizied Date: Mon, 7 Mar 2022 15:52:15 +0100 Subject: [PATCH] Add test debug traces. --- tests.py | 29 +++++++++++++++++------------ 1 file changed, 17 insertions(+), 12 deletions(-) diff --git a/tests.py b/tests.py index 0515c6c..679ef2f 100644 --- a/tests.py +++ b/tests.py @@ -375,25 +375,30 @@ class TestMethods(unittest.TestCase): # ocs.pcap # quic-mvfst-22_decryption_error.pcap # http-crash-content-disposition.pcap - # doq.pcap pcap_files = get_files_list(os.path.join("tests", "pcaps")) result_files = get_files_list(os.path.join("tests", "results")) + failures = 0 print("\n----------------------------------------------------------------------") print(".Test nDPI integration on {} applications:".format(len(pcap_files))) for file_idx, test_file in enumerate(pcap_files): test_case_name = os.path.basename(test_file) - test = NFStreamer(source=test_file, - n_dissections=20, - n_meters=1).to_pandas()[["id", - "bidirectional_packets", - "bidirectional_bytes", - "application_name", - "application_category_name", - "application_is_guessed"]].to_dict() + try: + test = NFStreamer(source=test_file, + n_dissections=20, + n_meters=1).to_pandas()[["id", + "bidirectional_packets", + "bidirectional_bytes", + "application_name", + "application_category_name", + "application_is_guessed"]].to_dict() - true = pd.read_csv(result_files[file_idx]).to_dict() - self.assertEqual(test, true) - print("{}\t: {}".format(test_case_name.ljust(60, ' '), colored('OK', 'green'))) + true = pd.read_csv(result_files[file_idx]).to_dict() + self.assertEqual(test, true) + print("{}\t: {}".format(test_case_name.ljust(60, ' '), colored('OK', 'green'))) + except AssertionError: + failures += 1 + print("{}\t: {}".format(test_case_name.ljust(60, ' '), colored('KO', 'red'))) + self.assertEqual(failures, 0) def test_splt(self): print("\n----------------------------------------------------------------------")