import unittest import random import ID2TLib.TestLibrary as Lib import Core.Controller as Ctrl controller = Ctrl.Controller(pcap_file_path=Lib.test_pcap, do_extra_tests=False, non_verbose=True) controller.load_pcap_statistics(flag_write_file=False, flag_recalculate_stats=True, flag_print_statistics=False) ipAddresses = ['10.0.2.15', '104.83.103.45', '13.107.21.200', '131.253.61.100', '172.217.23.142', '172.217.23.174', '192.168.33.254', '204.79.197.200', '23.51.123.27', '35.161.3.50', '52.11.17.245', '52.34.37.177', '52.39.210.199', '52.41.250.141', '52.85.173.182', '54.149.74.139', '54.187.98.195', '54.192.44.108', '54.192.44.177', '72.247.178.113', '72.247.178.67', '93.184.220.29'] class UnitTestInternalQueries(unittest.TestCase): # FILE METAINFORMATION TESTS def test_get_file_information(self): self.assertEqual(controller.statistics.get_file_information(), [('Pcap file path', Lib.test_pcap), ('Total packet count', 1998, 'packets'), ("Recognized packets", 1988, "packets"), ("Unrecognized packets", 10, "PDUs"), ("% Recognized packets", 99.49949949949949, "%"), ("% Unrecognized packets", 0.5005005005005005, "%"), ("Last unknown PDU", '1970-01-01 01:07:39.604899'), ('Capture duration', '25.4294414520264', 'seconds'), ('Capture start', '\t1970-01-01 01:01:45.647675'), ('Capture end', '\t1970-01-01 01:08:10.102034')]) def test_get_packet_count(self): self.assertEqual(controller.statistics.get_packet_count(), 1998) def test_get_capture_duration(self): self.assertEqual(controller.statistics.get_capture_duration(), '25.4294414520264') def test_get_pcap_timestamp_start(self): self.assertEqual(controller.statistics.get_pcap_timestamp_start(), '1970-01-01 01:01:45.647675') def test_get_pcap_timestamp_end(self): self.assertEqual(controller.statistics.get_pcap_timestamp_end(), '1970-01-01 01:08:10.102034') # FIXME: This seems to be the only testcase where float values differ slightly between macOS and Linux def test_get_general_file_statistics(self): file_stats = controller.statistics.get_general_file_statistics() self.assertEqual(file_stats[0], ('Avg. packet rate', 78.57034301757812, 'packets/sec')) self.assertEqual(file_stats[1], ('Avg. packet size', 0.0, 'kbytes')) self.assertEqual(file_stats[2], ('Avg. packets sent', 90.0, 'packets')) self.assertEqual(file_stats[3][0], 'Avg. bandwidth in') self.assertAlmostEqual(file_stats[3][1], 9.529013633728027, places=5) self.assertEqual(file_stats[3][2], 'kbit/s') self.assertEqual(file_stats[4][0], 'Avg. bandwidth out') self.assertAlmostEqual(file_stats[4][1], 9.529013633728027, places=5) # INTERNAL QUERY TESTS def test_get_ip_address_count(self): self.assertEqual(controller.statistics.get_ip_address_count(), 22) def test_get_ip_addresses(self): self.assertEqual(controller.statistics.get_ip_addresses(), ipAddresses) def test_get_most_used_ip_address(self): self.assertEqual(controller.statistics.get_most_used_ip_address(), '10.0.2.15') def test_get_random_ip_address(self): random.seed(5) self.assertEqual(controller.statistics.get_random_ip_address(), '72.247.178.113') def test_get_random_ip_address_count_2(self): random.seed(5) self.assertEqual(controller.statistics.get_random_ip_address(2), ['72.247.178.113', '23.51.123.27']) def test_get_ip_address_from_mac(self): self.assertEqual(controller.statistics.get_ip_address_from_mac('08:00:27:a3:83:43'), '10.0.2.15') def test_get_mac_address_1(self): self.assertEqual(controller.statistics.get_mac_address(ip_address='72.247.178.67'), '52:54:00:12:35:02') def test_get_mac_address_2(self): self.assertEqual(controller.statistics.get_mac_address(ip_address='10.0.2.15'), '08:00:27:a3:83:43') def test_get_most_used_mss(self): self.assertEqual(controller.statistics.get_most_used_mss(ip_address='10.0.2.15'), 1460) def test_get_most_used_ttl(self): self.assertEqual(controller.statistics.get_most_used_ttl(ip_address='10.0.2.15'), 128) def test_get_pps_sent_1(self): self.assertEqual(controller.statistics.get_pps_sent(ip_address='72.247.178.67'), 0) def test_get_pps_sent_2(self): self.assertEqual(controller.statistics.get_pps_sent(ip_address='10.0.2.15'), 32) def test_get_pps_sent_wrong_input(self): # wrong input parameter with self.assertRaises(TypeError): self.assertEqual(controller.statistics.get_pps_sent('08:00:27:a3:83:43'), 32) def test_get_pps_received_1(self): self.assertEqual(controller.statistics.get_pps_received(ip_address='72.247.178.67'), 0) def test_get_pps_received_2(self): self.assertEqual(controller.statistics.get_pps_received(ip_address='10.0.2.15'), 46) def test_get_ttl_distribution_1(self): self.assertEqual(controller.statistics.get_ttl_distribution(ip_address='72.247.178.67'), {64: 5}) def test_get_ttl_distribution_2(self): self.assertEqual(controller.statistics.get_ttl_distribution(ip_address='10.0.2.15'), {128: 817}) def test_get_mss_distribution_1(self): self.assertEqual(controller.statistics.get_mss_distribution(ip_address='72.247.178.67'), {1460: 1}) def test_get_mss_distribution_2(self): self.assertEqual(controller.statistics.get_mss_distribution(ip_address='10.0.2.15'), {1460: 36}) def test_get_win_distribution_1(self): self.assertEqual(controller.statistics.get_win_distribution(ip_address='72.247.178.67'), {65535: 5}) def test_get_tos_distribution_1(self): self.assertEqual(controller.statistics.get_tos_distribution(ip_address='72.247.178.67'), {0: 5}) def test_get_tos_distribution_2(self): self.assertEqual(controller.statistics.get_tos_distribution(ip_address='10.0.2.15'), {0: 817}) # INTERNAL HELPER-FUNCTION TESTS def test_calculate_standard_deviation(self): self.assertEqual(controller.statistics.calculate_standard_deviation([1, 1, 2, 3, 5, 8, 13, 21]), 6.609652033201143) def test_calculate_entropy(self): self.assertEqual(controller.statistics.calculate_entropy([1, 1, 2, 3, 5, 8, 13, 21]), 2.371389165297016) def test_calculate_entropy_normalized(self): self.assertEqual(controller.statistics.calculate_entropy([1, 1, 2, 3, 5, 8, 13, 21], normalized=True), (2.371389165297016, 0.7904630550990053)) def test_calculate_complement_packet_rates_1(self): cpr = controller.statistics.calculate_complement_packet_rates(0)[0:9] self.assertEqual(cpr, [(186.418564, 0), (186.418824, 0), (186.419346, 0), (186.445361, 0), (186.46954399999998, 0), (186.476234, 0), (186.477304, 0), (186.48606999999998, 0), (186.486761, 0)]) def test_calculate_complement_packet_rates_2(self): cpr = controller.statistics.calculate_complement_packet_rates(42)[0:9] self.assertEqual(cpr, [(186.418564, 41), (186.418824, 42), (186.419346, 42), (186.445361, 42), (186.46954399999998, 42), (186.476234, 42), (186.477304, 42), (186.48606999999998, 42), (186.486761, 42)])