test_internalQueries.py 7.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148
  1. import unittest
  2. import random
  3. import ID2TLib.TestLibrary as Lib
  4. import Core.Controller as Ctrl
  5. controller = Ctrl.Controller(pcap_file_path=Lib.test_pcap, do_extra_tests=False, non_verbose=True)
  6. controller.load_pcap_statistics(flag_write_file=False, flag_recalculate_stats=True, flag_print_statistics=False)
  7. 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',
  8. '192.168.33.254', '204.79.197.200', '23.51.123.27', '35.161.3.50', '52.11.17.245', '52.34.37.177',
  9. '52.39.210.199', '52.41.250.141', '52.85.173.182', '54.149.74.139', '54.187.98.195', '54.192.44.108',
  10. '54.192.44.177', '72.247.178.113', '72.247.178.67', '93.184.220.29']
  11. class UnitTestInternalQueries(unittest.TestCase):
  12. # FILE METAINFORMATION TESTS
  13. def test_get_file_information(self):
  14. self.assertEqual(controller.statistics.get_file_information(),
  15. [('Pcap file path', Lib.test_pcap),
  16. ('Total packet count', 1998, 'packets'),
  17. ("Recognized packets", 1988, "packets"),
  18. ("Unrecognized packets", 10, "PDUs"), ("% Recognized packets", 99.49949949949949, "%"),
  19. ("% Unrecognized packets", 0.5005005005005005, "%"),
  20. ("Last unknown PDU", '1970-01-01 00:07:39.604899'),
  21. ('Capture duration', '384.454345703125', 'seconds'),
  22. ('Capture start', '\t1970-01-01 00:01:45.647675'),
  23. ('Capture end', '\t1970-01-01 00:08:10.102034')])
  24. def test_get_packet_count(self):
  25. self.assertEqual(controller.statistics.get_packet_count(), 1998)
  26. def test_get_capture_duration(self):
  27. self.assertEqual(controller.statistics.get_capture_duration(), '384.454345703125')
  28. def test_get_pcap_timestamp_start(self):
  29. self.assertEqual(controller.statistics.get_pcap_timestamp_start(), '1970-01-01 00:01:45.647675')
  30. def test_get_pcap_timestamp_end(self):
  31. self.assertEqual(controller.statistics.get_pcap_timestamp_end(), '1970-01-01 00:08:10.102034')
  32. # FIXME: This seems to be the only testcase where float values differ slightly between macOS and Linux
  33. def test_get_general_file_statistics(self):
  34. file_stats = controller.statistics.get_general_file_statistics()
  35. self.assertEqual(file_stats[0], ('Avg. packet rate', 5.196976184844971, 'packets/sec'))
  36. self.assertEqual(file_stats[1], ('Avg. packet size', 0.0, 'kbytes'))
  37. self.assertEqual(file_stats[2], ('Avg. packets sent', 90.0, 'packets'))
  38. self.assertEqual(file_stats[3][0], 'Avg. bandwidth in')
  39. self.assertAlmostEqual(file_stats[3][1], 0.6302894353866577, places=5)
  40. self.assertEqual(file_stats[3][2], 'kbit/s')
  41. self.assertEqual(file_stats[4][0], 'Avg. bandwidth out')
  42. self.assertAlmostEqual(file_stats[4][1], 0.6302894353866577, places=5)
  43. # INTERNAL QUERY TESTS
  44. def test_get_ip_address_count(self):
  45. self.assertEqual(controller.statistics.get_ip_address_count(), 22)
  46. def test_get_ip_addresses(self):
  47. self.assertEqual(controller.statistics.get_ip_addresses(), ipAddresses)
  48. def test_get_most_used_ip_address(self):
  49. self.assertEqual(controller.statistics.get_most_used_ip_address(), '10.0.2.15')
  50. def test_get_random_ip_address(self):
  51. random.seed(5)
  52. self.assertEqual(controller.statistics.get_random_ip_address(), '72.247.178.113')
  53. def test_get_random_ip_address_count_2(self):
  54. random.seed(5)
  55. self.assertEqual(controller.statistics.get_random_ip_address(2), ['72.247.178.113', '23.51.123.27'])
  56. def test_get_ip_address_from_mac(self):
  57. self.assertEqual(controller.statistics.get_ip_address_from_mac('08:00:27:a3:83:43'), '10.0.2.15')
  58. def test_get_mac_address_1(self):
  59. self.assertEqual(controller.statistics.get_mac_address(ip_address='72.247.178.67'), '52:54:00:12:35:02')
  60. def test_get_mac_address_2(self):
  61. self.assertEqual(controller.statistics.get_mac_address(ip_address='10.0.2.15'), '08:00:27:a3:83:43')
  62. def test_get_most_used_mss(self):
  63. self.assertEqual(controller.statistics.get_most_used_mss(ip_address='10.0.2.15'), 1460)
  64. def test_get_most_used_ttl(self):
  65. self.assertEqual(controller.statistics.get_most_used_ttl(ip_address='10.0.2.15'), 128)
  66. def test_get_pps_sent_1(self):
  67. self.assertEqual(controller.statistics.get_pps_sent(ip_address='72.247.178.67'), 0)
  68. def test_get_pps_sent_2(self):
  69. self.assertEqual(controller.statistics.get_pps_sent(ip_address='10.0.2.15'), 2)
  70. def test_get_pps_sent_wrong_input(self):
  71. # wrong input parameter
  72. with self.assertRaises(TypeError):
  73. self.assertEqual(controller.statistics.get_pps_sent('08:00:27:a3:83:43'), 32)
  74. def test_get_pps_received_1(self):
  75. self.assertEqual(controller.statistics.get_pps_received(ip_address='72.247.178.67'), 0)
  76. def test_get_pps_received_2(self):
  77. self.assertEqual(controller.statistics.get_pps_received(ip_address='10.0.2.15'), 3)
  78. def test_get_ttl_distribution_1(self):
  79. self.assertEqual(controller.statistics.get_ttl_distribution(ip_address='72.247.178.67'), {64: 5})
  80. def test_get_ttl_distribution_2(self):
  81. self.assertEqual(controller.statistics.get_ttl_distribution(ip_address='10.0.2.15'), {128: 817})
  82. def test_get_mss_distribution_1(self):
  83. self.assertEqual(controller.statistics.get_mss_distribution(ip_address='72.247.178.67'), {1460: 1})
  84. def test_get_mss_distribution_2(self):
  85. self.assertEqual(controller.statistics.get_mss_distribution(ip_address='10.0.2.15'), {1460: 36})
  86. def test_get_win_distribution_1(self):
  87. self.assertEqual(controller.statistics.get_win_distribution(ip_address='72.247.178.67'), {65535: 5})
  88. def test_get_tos_distribution_1(self):
  89. self.assertEqual(controller.statistics.get_tos_distribution(ip_address='72.247.178.67'), {0: 5})
  90. def test_get_tos_distribution_2(self):
  91. self.assertEqual(controller.statistics.get_tos_distribution(ip_address='10.0.2.15'), {0: 817})
  92. # INTERNAL HELPER-FUNCTION TESTS
  93. def test_calculate_standard_deviation(self):
  94. self.assertEqual(controller.statistics.calculate_standard_deviation([1, 1, 2, 3, 5, 8, 13, 21]),
  95. 6.609652033201143)
  96. def test_calculate_entropy(self):
  97. self.assertEqual(controller.statistics.calculate_entropy([1, 1, 2, 3, 5, 8, 13, 21]), 2.371389165297016)
  98. def test_calculate_entropy_normalized(self):
  99. self.assertEqual(controller.statistics.calculate_entropy([1, 1, 2, 3, 5, 8, 13, 21], normalized=True),
  100. (2.371389165297016, 0.7904630550990053))
  101. def test_calculate_complement_packet_rates_1(self):
  102. cpr = controller.statistics.calculate_complement_packet_rates(0)[0:9]
  103. self.assertEqual(cpr, [(186.418564, 0), (186.418824, 0), (186.419346, 0), (186.445361, 0),
  104. (186.46954399999998, 0), (186.476234, 0), (186.477304, 0), (186.48606999999998, 0),
  105. (186.486761, 0)])
  106. def test_calculate_complement_packet_rates_2(self):
  107. cpr = controller.statistics.calculate_complement_packet_rates(42)[0:9]
  108. self.assertEqual(cpr, [(186.418564, 41), (186.418824, 42), (186.419346, 42), (186.445361, 42),
  109. (186.46954399999998, 42), (186.476234, 42), (186.477304, 42),
  110. (186.48606999999998, 42),
  111. (186.486761, 42)])