test_NamedQueries.py 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. import unittest
  2. import pyparsing
  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. ip_addresses = ["10.0.2.15", "104.83.103.45", "13.107.21.200", "131.253.61.100", "172.217.23.142",
  8. "172.217.23.174", "192.168.33.254", "204.79.197.200", "23.51.123.27", "35.161.3.50",
  9. "52.11.17.245", "52.34.37.177", "52.39.210.199", "52.41.250.141", "52.85.173.182",
  10. "54.149.74.139", "54.187.98.195", "54.192.44.108", "54.192.44.177", "72.247.178.113",
  11. "72.247.178.67", "93.184.220.29"]
  12. allWinSize = [0, 822, 1330, 5082, 8192, 9900, 27060, 35657, 39917, 47030, 50782, 51310, 52202, 52740, 55062,
  13. 56492, 58520, 59950, 59980, 61380, 62788, 62810, 62811, 62906, 63056, 63076, 63086, 63151, 63261, 63350,
  14. 63370, 63400, 63409, 63456, 63516, 63547, 63552, 63572, 63603, 63628, 63655, 63663, 63675, 63686, 63706,
  15. 63839, 63842, 63886, 63893, 63917, 63954, 63963, 63982, 63991, 64000, 64005, 64088, 64110, 64148, 64165,
  16. 64177, 64189, 64194, 64198, 64209, 64230, 64240, 65535]
  17. leastUsedWinASize = [822, 1330, 5082, 9900, 27060, 35657, 39917, 47030, 50782, 51310, 52202, 52740, 55062, 56492,
  18. 58520, 59950, 59980, 61380, 63056, 63963, 63982, 64000, 64005, 64198, 64230]
  19. allPort = [53, 80, 443, 49157, 49160, 49163, 49164, 49165, 49166, 49167, 49168, 49169, 49170, 49171, 49172,
  20. 49173, 49174, 49175, 49176, 49177, 49178, 49179, 49180, 49181, 49182, 49183, 49184, 49185, 49186, 49187,
  21. 49188, 49189, 49190, 49191, 49192, 49193, 49194, 49195, 49196, 49197, 49247, 49323, 49470, 49636, 49695,
  22. 49798, 49927, 49935, 49945, 50262, 50836, 50968, 51143, 51166, 51350, 51451, 51669, 51713, 52033, 52135,
  23. 52399, 52520, 52644, 52697, 52743, 52786, 52964, 52981, 53059, 53234, 53461, 53691, 53708, 53745, 53836,
  24. 54049, 54446, 54593, 54598, 54652, 54663, 54717, 54853, 54930, 55004, 55018, 55119, 55125, 55299, 55310,
  25. 55463, 55650, 55667, 55752, 55843, 55851, 56146, 56325, 56567, 56589, 56750, 57049, 57179, 57275, 57520,
  26. 57653, 57840, 57957, 57991, 58401, 58440, 58645, 58797, 58814, 58905, 58913, 58943, 59380, 59408, 59461,
  27. 59467, 59652, 59660, 59718, 59746, 59844, 60006, 60209, 60414, 60422, 60659, 60696, 60708, 60756, 60827,
  28. 60840, 61181, 61300, 61592, 61718, 61738, 61769, 61807, 62412, 62428, 62447, 62490, 62625, 62626, 62664,
  29. 63425, 64096, 64121, 64137, 64252, 64334, 64337, 64479, 64509, 64637, 64807, 64811, 65448, 65487]
  30. class UnitTestNamedQueries(unittest.TestCase):
  31. def test_most_used_ipaddress(self):
  32. self.assertEqual(controller.statistics.process_db_query('most_used(ipaddress)'), '10.0.2.15')
  33. def test_most_used_macaddress(self):
  34. self.assertEqual(controller.statistics.process_db_query('most_used(macaddress)'), '52:54:00:12:35:02')
  35. def test_most_used_portnumber(self):
  36. self.assertEqual(controller.statistics.process_db_query('most_used(portnumber)'), 443)
  37. def test_most_used_protocolname(self):
  38. self.assertEqual(controller.statistics.process_db_query('most_used(protocolname)'), 'IPv4')
  39. def test_most_used_ttlvalue(self):
  40. self.assertEqual(controller.statistics.process_db_query('most_used(ttlvalue)'), 64)
  41. def test_most_used_mssvalue(self):
  42. self.assertEqual(controller.statistics.process_db_query('most_used(mssvalue)'), 1460)
  43. def test_most_used_winsize(self):
  44. self.assertEqual(controller.statistics.process_db_query('most_used(winsize)'), 65535)
  45. def test_most_used_ipclass(self):
  46. self.assertEqual(controller.statistics.process_db_query('most_used(ipclass)'), 'A')
  47. def test_least_used_ipaddress(self):
  48. self.assertEqual(controller.statistics.process_db_query('least_used(ipaddress)'), '72.247.178.113')
  49. def test_least_used_macaddress(self):
  50. self.assertEqual(controller.statistics.process_db_query('least_used(macaddress)'), '08:00:27:a3:83:43')
  51. def test_least_used_portnumber(self):
  52. self.assertEqual(controller.statistics.process_db_query('least_used(portnumber)'), [58645, 59844])
  53. def test_least_used_protocolname(self):
  54. self.assertEqual(controller.statistics.process_db_query('least_used(protocolname)'), 'UDP')
  55. def test_least_used_ttlvalue(self):
  56. self.assertEqual(controller.statistics.process_db_query('least_used(ttlvalue)'), 255)
  57. def test_least_used_mssvalue(self):
  58. self.assertEqual(controller.statistics.process_db_query('least_used(mssvalue)'), 1460)
  59. def test_least_used_winsize(self):
  60. self.assertEqual(controller.statistics.process_db_query('least_used(winsize)'), leastUsedWinASize)
  61. def test_least_used_ipclass(self):
  62. self.assertEqual(controller.statistics.process_db_query('least_used(ipclass)'), ['A-private', 'C', 'C-private'])
  63. def test_avg_pktsreceived(self):
  64. self.assertEqual(controller.statistics.process_db_query('avg(pktsreceived)'), 90.36363636363636)
  65. def test_avg_pktssent(self):
  66. self.assertEqual(controller.statistics.process_db_query('avg(pktssent)'), 90.36363636363636)
  67. def test_avg_kbytesreceived(self):
  68. self.assertEqual(controller.statistics.process_db_query('avg(kbytesreceived)'), 30.289683948863637)
  69. def test_avg_kbytessent(self):
  70. self.assertEqual(controller.statistics.process_db_query('avg(kbytessent)'), 30.289683948863637)
  71. def test_avg_ttlvalue(self):
  72. self.assertEqual(controller.statistics.process_db_query('avg(ttlvalue)'), 75.08695652173913)
  73. def test_avg_mss(self):
  74. self.assertEqual(controller.statistics.process_db_query('avg(mss)'), 1460.0)
  75. def test_avg_ipaddress(self):
  76. with self.assertRaises(pyparsing.ParseException):
  77. controller.statistics.process_db_query('avg(ipAddress)')
  78. def test_all_ipaddress(self):
  79. self.assertEqual(controller.statistics.process_db_query('all(ipaddress)'), ip_addresses)
  80. def test_all_ttlvalue(self):
  81. self.assertEqual(controller.statistics.process_db_query('all(ttlvalue)'), [64, 128, 255])
  82. def test_all_mss(self):
  83. self.assertEqual(controller.statistics.process_db_query('all(mss)'), 1460)
  84. def test_all_macaddress(self):
  85. self.assertEqual(controller.statistics.process_db_query('all(macaddress)'), ['08:00:27:a3:83:43',
  86. '52:54:00:12:35:02'])
  87. def test_all_portnumber(self):
  88. self.assertEqual(controller.statistics.process_db_query('all(portnumber)'), allPort)
  89. def test_all_protocolname(self):
  90. self.assertEqual(controller.statistics.process_db_query('all(protocolname)'), ['IPv4', 'TCP', 'UDP'])
  91. def test_all_winsize(self):
  92. self.assertEqual(controller.statistics.process_db_query('all(winSize)'),
  93. allWinSize)
  94. def test_all_ipclass(self):
  95. self.assertEqual(controller.statistics.process_db_query('all(ipClass)'),
  96. ['A', 'A-private', 'B', 'C', 'C-private'])
  97. def test_is_query_named_query(self):
  98. self.assertTrue(controller.statistics.is_query('least_used(ipaddress)'))
  99. def test_is_query_no_string(self):
  100. self.assertFalse(controller.statistics.is_query(42))