Browse Source

Tests of new implemented named queries

Roey Regev 6 years ago
parent
commit
648c7e8417
1 changed files with 13 additions and 2 deletions
  1. 13 2
      code/Test/test_NamedQueries.py

+ 13 - 2
code/Test/test_NamedQueries.py

@@ -8,7 +8,11 @@ pcap = ROOT_DIR + "/../resources/test/reference_1998.pcap"
 controller = Ctrl.Controller(pcap_file_path=pcap, do_extra_tests=False, non_verbose=False)
 controller.load_pcap_statistics(flag_write_file=False, flag_recalculate_stats=True, flag_print_statistics=False)
 
-
+allWinSize = [0, 822, 1330, 5082, 8192, 9900, 27060, 35657, 39917, 47030, 50782, 51310, 52202, 52740, 55062, 56492,
+              58520, 59950, 59980,61380, 62788, 62810, 62811, 62906, 63056, 63076, 63086, 63151, 63261, 63350, 63370,
+              63400, 63409, 63456, 63516, 63547, 63552, 63572, 63603, 63628, 63655, 63663, 63675, 63686, 63706, 63839,
+              63842, 63886, 63893, 63917, 63954, 63963, 63982, 63991, 64000, 64005, 64088, 64110, 64148, 64165, 64177,
+              64189, 64194, 64198, 64209, 64230, 64240, 65535]
 class UnitTestPyparsing(unittest.TestCase):
     def test_named_query(self):
         self.assertEqual(controller.statistics.process_db_query('least_used(portnumber)'),
@@ -53,4 +57,11 @@ class UnitTestPyparsing(unittest.TestCase):
                           '72.247.178.67','93.184.220.29'])
         # undefined Query
         with self.assertRaises(pyparsing.ParseException):
-            controller.statistics.process_db_query('avg(notDefinedQuery)')
+            controller.statistics.process_db_query('avg(notDefinedQuery)')
+
+        self.assertEqual(controller.statistics.process_db_query('all(ipClass)'),
+                         ['A', 'A-private', 'B', 'C', 'C-private'])
+        self.assertEqual(controller.statistics.process_db_query('all(winSize)'),
+                         allWinSize)
+        self.assertEqual(controller.statistics.process_db_query('least_used(ipClass)'),
+                         ['A-private', 'C', 'C-private'])