1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556 |
- import unittest, pyparsing
- from definitions import ROOT_DIR
- import Core.Controller as Ctrl
- 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)
- class UnitTestPyparsing(unittest.TestCase):
- def test_named_query(self):
- self.assertEqual(controller.statistics.process_db_query('least_used(portnumber)'),
- [58645, 59844])
- self.assertEqual(controller.statistics.process_db_query('most_used(portnumber)'),
- 443)
- self.assertEqual(controller.statistics.process_db_query('most_used(protocolname)'),
- 'IPv4')
- self.assertEqual(controller.statistics.process_db_query('least_used(protocolname)'),
- 'UDP')
- self.assertEqual(controller.statistics.process_db_query('most_used(ttlvalue)'),
- 64)
- self.assertEqual(controller.statistics.process_db_query('least_used(ttlvalue)'),
- 255)
- self.assertEqual(controller.statistics.process_db_query('most_used(mssvalue)'),
- 1460)
- self.assertEqual(controller.statistics.process_db_query('least_used(mssvalue)'),
- 1460)
- self.assertEqual(controller.statistics.process_db_query('most_used(winsize)'),
- 65535)
- self.assertEqual(controller.statistics.process_db_query('most_used(ipclass)'),
- 'A')
- self.assertEqual(controller.statistics.process_db_query('least_used(ipclass)'),
- ['A-private', 'C', 'C-private'])
- self.assertEqual(controller.statistics.process_db_query('avg(pktsreceived)'),
- 90.36363636363636)
- self.assertEqual(controller.statistics.process_db_query('avg(pktssent)'),
- 90.36363636363636)
- self.assertEqual(controller.statistics.process_db_query('avg(kbytesreceived)'),
- 30.289683948863637)
- self.assertEqual(controller.statistics.process_db_query('avg(kbytessent)'),
- 30.289683948863637)
- self.assertEqual(controller.statistics.process_db_query('avg(ttlvalue)'),
- 75.08695652173913)
- self.assertEqual(controller.statistics.process_db_query('avg(mss)'),
- 1460.0)
- self.assertEqual(controller.statistics.process_db_query('all(ipaddress)'),
- ['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'])
- # undefined Query
- with self.assertRaises(pyparsing.ParseException):
- controller.statistics.process_db_query('avg(notDefinedQuery)')
|