3
0

test_Queries.py 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232
  1. import random
  2. import unittest
  3. import Core.Controller as Ctrl
  4. import ID2TLib.TestLibrary as Test
  5. # TODO: improve coverage
  6. controller = Ctrl.Controller(pcap_file_path=Test.test_pcap, do_extra_tests=False)
  7. controller.load_pcap_statistics(flag_write_file=False, flag_recalculate_stats=True, flag_print_statistics=False)
  8. file_information = [('Pcap file', Test.test_pcap),
  9. ('Packets', 1998, 'packets'), ('Capture length', '25.4294414520264', 'seconds'),
  10. ('Capture start', '1970-01-01 01:01:45.647675'), ('Capture end', '1970-01-01 01:08:10.102034')]
  11. file_statistics = [('Avg. packet rate', 78.57034301757812, 'packets/sec'), ('Avg. packet size', 0.0, 'kbytes'),
  12. ('Avg. packets sent', 90.0, 'packets'), ('Avg. bandwidth in', 9.5290, 'kbit/s'),
  13. ('Avg. bandwidth out', 9.5290, 'kbit/s')]
  14. ip_addresses = ["10.0.2.15", "104.83.103.45", "13.107.21.200", "131.253.61.100", "172.217.23.142",
  15. "172.217.23.174", "192.168.33.254", "204.79.197.200", "23.51.123.27", "35.161.3.50",
  16. "52.11.17.245", "52.34.37.177", "52.39.210.199", "52.41.250.141", "52.85.173.182",
  17. "54.149.74.139", "54.187.98.195", "54.192.44.108", "54.192.44.177", "72.247.178.113",
  18. "72.247.178.67", "93.184.220.29"]
  19. ports = [53, 80, 443, 49157, 49160, 49163, 49164, 49165, 49166, 49167, 49168, 49169, 49170, 49171, 49172, 49173, 49174,
  20. 49175, 49176, 49177, 49178, 49179, 49180, 49181, 49182, 49183, 49184, 49185, 49186, 49187, 49188, 49189, 49190,
  21. 49191, 49192, 49193, 49194, 49195, 49196, 49197, 49247, 49323, 49470, 49636, 49695, 49798, 49927, 49935, 49945,
  22. 50262, 50836, 50968, 51143, 51166, 51350, 51451, 51669, 51713, 52033, 52135, 52399, 52520, 52644, 52697, 52743,
  23. 52786, 52964, 52981, 53059, 53234, 53461, 53691, 53708, 53745, 53836, 54049, 54446, 54593, 54598, 54652, 54663,
  24. 54717, 54853, 54930, 55004, 55018, 55119, 55125, 55299, 55310, 55463, 55650, 55667, 55752, 55843, 55851, 56146,
  25. 56325, 56567, 56589, 56750, 57049, 57179, 57275, 57520, 57653, 57840, 57957, 57991, 58401, 58440, 58645, 58797,
  26. 58814, 58905, 58913, 58943, 59380, 59408, 59461, 59467, 59652, 59660, 59718, 59746, 59844, 60006, 60209, 60414,
  27. 60422, 60659, 60696, 60708, 60756, 60827, 60840, 61181, 61300, 61592, 61718, 61738, 61769, 61807, 62412, 62428,
  28. 62447, 62490, 62625, 62626, 62664, 63425, 64096, 64121, 64137, 64252, 64334, 64337, 64479, 64509, 64637, 64807,
  29. 64811, 65448, 65487]
  30. class TestQueries(unittest.TestCase):
  31. def test_get_file_information(self):
  32. self.assertEqual(controller.statistics.get_file_information(), file_information)
  33. def test_get_general_file_statistics(self):
  34. file_stats = controller.statistics.get_general_file_statistics()
  35. file_stats[3] = ('Avg. bandwidth in', round(file_stats[3][1], 4), 'kbit/s')
  36. file_stats[4] = ('Avg. bandwidth out', round(file_stats[4][1], 4), 'kbit/s')
  37. self.assertEqual(file_stats, file_statistics)
  38. def test_get_capture_duration(self):
  39. self.assertEqual(controller.statistics.get_capture_duration(), '25.4294414520264')
  40. def test_get_pcap_timestamp_start(self):
  41. self.assertEqual(controller.statistics.get_pcap_timestamp_start(), '1970-01-01 01:01:45.647675')
  42. def test_get_pcap_timestamp_end(self):
  43. self.assertEqual(controller.statistics.get_pcap_timestamp_end(), '1970-01-01 01:08:10.102034')
  44. def test_get_pps_sent_1(self):
  45. self.assertEqual(controller.statistics.get_pps_sent(ip_address='72.247.178.67'), 0)
  46. def test_get_pps_sent_2(self):
  47. self.assertEqual(controller.statistics.get_pps_sent(ip_address='10.0.2.15'), 32)
  48. def test_get_pps_received_1(self):
  49. self.assertEqual(controller.statistics.get_pps_received(ip_address='72.247.178.67'), 0)
  50. def test_get_pps_received_2(self):
  51. self.assertEqual(controller.statistics.get_pps_received(ip_address='10.0.2.15'), 46)
  52. def test_get_packet_count(self):
  53. self.assertEqual(controller.statistics.get_packet_count(), 1998)
  54. def test_get_most_used_ip_address(self):
  55. self.assertEqual(controller.statistics.get_most_used_ip_address(), '10.0.2.15')
  56. def test_get_ttl_distribution_1(self):
  57. self.assertEqual(controller.statistics.get_ttl_distribution(ip_address='72.247.178.67'), {64: 5})
  58. def test_get_ttl_distribution_2(self):
  59. self.assertEqual(controller.statistics.get_ttl_distribution(ip_address='10.0.2.15'), {128: 817})
  60. def test_get_mss_distribution_1(self):
  61. self.assertEqual(controller.statistics.get_mss_distribution(ip_address='72.247.178.67'), {1460: 1})
  62. def test_get_mss_distribution_2(self):
  63. self.assertEqual(controller.statistics.get_mss_distribution(ip_address='10.0.2.15'), {1460: 36})
  64. def test_get_win_distribution_1(self):
  65. self.assertEqual(controller.statistics.get_win_distribution(ip_address='72.247.178.67'), {65535: 5})
  66. def test_get_tos_distribution_1(self):
  67. self.assertEqual(controller.statistics.get_tos_distribution(ip_address='72.247.178.67'), {0: 5})
  68. def test_get_tos_distribution_2(self):
  69. self.assertEqual(controller.statistics.get_tos_distribution(ip_address='10.0.2.15'), {0: 817})
  70. def test_get_ip_address_count(self):
  71. self.assertEqual(controller.statistics.get_ip_address_count(), 22)
  72. def test_get_ip_addresses(self):
  73. self.assertEqual(controller.statistics.get_ip_addresses(), ip_addresses)
  74. def test_get_random_ip_address(self):
  75. random.seed(5)
  76. self.assertEqual(controller.statistics.get_random_ip_address(), '72.247.178.113')
  77. def test_get_random_ip_address_count_2(self):
  78. random.seed(5)
  79. self.assertEqual(controller.statistics.get_random_ip_address(2), ['72.247.178.113', '23.51.123.27'])
  80. def test_get_mac_address_1(self):
  81. self.assertEqual(controller.statistics.get_mac_address(ip_address='72.247.178.67'), '52:54:00:12:35:02')
  82. def test_get_mac_address_2(self):
  83. self.assertEqual(controller.statistics.get_mac_address(ip_address='10.0.2.15'), '08:00:27:a3:83:43')
  84. def test_get_most_used_mss(self):
  85. self.assertEqual(controller.statistics.get_most_used_mss(ip_address='10.0.2.15'), 1460)
  86. def test_get_most_used_ttl(self):
  87. self.assertEqual(controller.statistics.get_most_used_ttl(ip_address='10.0.2.15'), 128)
  88. def test_is_query_no_string(self):
  89. self.assertFalse(controller.statistics.is_query(42))
  90. def test_is_query_named_query(self):
  91. self.assertTrue(controller.statistics.is_query('least_used(ipaddress)'))
  92. def test_is_query_standard_query(self):
  93. self.assertTrue(controller.statistics.is_query('SELECT * from ip_statistics'))
  94. def test_calculate_standard_deviation(self):
  95. self.assertEqual(controller.statistics.calculate_standard_deviation([1, 1, 2, 3, 5, 8, 13, 21]),
  96. 6.609652033201143)
  97. def test_calculate_entropy(self):
  98. self.assertEqual(controller.statistics.calculate_entropy([1, 1, 2, 3, 5, 8, 13, 21]), 2.371389165297016)
  99. def test_calculate_entropy_normalized(self):
  100. self.assertEqual(controller.statistics.calculate_entropy([1, 1, 2, 3, 5, 8, 13, 21], normalized=True),
  101. (2.371389165297016, 0.7904630550990053))
  102. def test_calculate_complement_packet_rates_1(self):
  103. cpr = controller.statistics.calculate_complement_packet_rates(0)[0:9]
  104. self.assertEqual(cpr, [(186.418564, 0), (186.418824, 0), (186.419346, 0), (186.445361, 0),
  105. (186.46954399999998, 0), (186.476234, 0), (186.477304, 0), (186.48606999999998, 0),
  106. (186.486761, 0)])
  107. def test_calculate_complement_packet_rates_2(self):
  108. cpr = controller.statistics.calculate_complement_packet_rates(42)[0:9]
  109. self.assertEqual(cpr, [(186.418564, 41), (186.418824, 42), (186.419346, 42), (186.445361, 42),
  110. (186.46954399999998, 42), (186.476234, 42), (186.477304, 42), (186.48606999999998, 42),
  111. (186.486761, 42)])
  112. # NAMED QUERY TESTS
  113. def test_most_used_ipaddress(self):
  114. self.assertEqual(controller.statistics.process_db_query('most_used(ipaddress)'), '10.0.2.15')
  115. def test_most_used_macaddress(self):
  116. self.assertEqual(controller.statistics.process_db_query('most_used(macaddress)'), '52:54:00:12:35:02')
  117. def test_most_used_portnumber(self):
  118. self.assertEqual(controller.statistics.process_db_query('most_used(portnumber)'), 443)
  119. def test_most_used_protocolname(self):
  120. self.assertEqual(controller.statistics.process_db_query('most_used(protocolname)'), 'IPv4')
  121. def test_most_used_ttlvalue(self):
  122. self.assertEqual(controller.statistics.process_db_query('most_used(ttlvalue)'), 64)
  123. def test_most_used_mssvalue(self):
  124. self.assertEqual(controller.statistics.process_db_query('most_used(mssvalue)'), 1460)
  125. def test_most_used_winsize(self):
  126. self.assertEqual(controller.statistics.process_db_query('most_used(winsize)'), 65535)
  127. def test_most_used_ipclass(self):
  128. self.assertEqual(controller.statistics.process_db_query('most_used(ipclass)'), 'A')
  129. def test_least_used_ipaddress(self):
  130. self.assertEqual(controller.statistics.process_db_query('least_used(ipaddress)'), '72.247.178.113')
  131. def test_least_used_macaddress(self):
  132. self.assertEqual(controller.statistics.process_db_query('least_used(macaddress)'), '08:00:27:a3:83:43')
  133. def test_least_used_portnumber(self):
  134. self.assertEqual(controller.statistics.process_db_query('least_used(portnumber)'), [58645, 59844])
  135. def test_least_used_protocolname(self):
  136. self.assertEqual(controller.statistics.process_db_query('least_used(protocolname)'), 'UDP')
  137. def test_least_used_ttlvalue(self):
  138. self.assertEqual(controller.statistics.process_db_query('least_used(ttlvalue)'), 255)
  139. def test_avg_pktsreceived(self):
  140. self.assertEqual(controller.statistics.process_db_query('avg(pktsreceived)'), 90.36363636363636)
  141. def test_avg_pktssent(self):
  142. self.assertEqual(controller.statistics.process_db_query('avg(pktssent)'), 90.36363636363636)
  143. def test_avg_kbytesreceived(self):
  144. self.assertEqual(controller.statistics.process_db_query('avg(kbytesreceived)'), 30.289683948863637)
  145. def test_avg_kbytessent(self):
  146. self.assertEqual(controller.statistics.process_db_query('avg(kbytessent)'), 30.289683948863637)
  147. def test_avg_ttlvalue(self):
  148. self.assertEqual(controller.statistics.process_db_query('avg(ttlvalue)'), 75.08695652173913)
  149. def test_avg_mss(self):
  150. self.assertEqual(controller.statistics.process_db_query('avg(mss)'), 1460.0)
  151. def test_all_ipaddress(self):
  152. self.assertEqual(controller.statistics.process_db_query('all(ipaddress)'), ip_addresses)
  153. def test_all_ttlvalue(self):
  154. self.assertEqual(controller.statistics.process_db_query('all(ttlvalue)'), [64, 128, 255])
  155. def test_all_mss(self):
  156. self.assertEqual(controller.statistics.process_db_query('all(mss)'), 1460)
  157. def test_all_macaddress(self):
  158. self.assertEqual(controller.statistics.process_db_query('all(macaddress)'), ['08:00:27:a3:83:43',
  159. '52:54:00:12:35:02'])
  160. def test_all_portnumber(self):
  161. self.assertEqual(controller.statistics.process_db_query('all(portnumber)'), ports)
  162. def test_all_protocolname(self):
  163. self.assertEqual(controller.statistics.process_db_query('all(protocolname)'), ['IPv4', 'TCP', 'UDP'])
  164. def test_nested_query(self):
  165. self.assertEqual(controller.statistics.process_db_query('macaddress(ipaddress=most_used(ipaddress))'), '08:00:27:a3:83:43')