123456789101112131415161718192021222324252627282930313233343536373839404142434445464748 |
- import unittest, sqlite3, 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_nested_query(self):
- self.assertEqual(controller.statistics.process_db_query('macaddress(ipaddress in most_used(ipaddress))'),
- '08:00:27:a3:83:43')
- self.assertEqual(controller.statistics.process_db_query('macaddress(ipaddress in least_used(ipaddress))'),
- '52:54:00:12:35:02')
- self.assertEqual(controller.statistics.process_db_query('ipaddress(macaddress in least_used(macaddress))'),
- '10.0.2.15')
- self.assertEqual(controller.statistics.process_db_query('ipaddress(macaddress in most_used(macaddress))'),
- ['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'])
- # semantically incorrect query
- with self.assertRaises(sqlite3.OperationalError):
- controller.statistics.process_db_query('ipaddress(ipaddress in most_used(macaddress))')
- # syntactically incorrect query
- with self.assertRaises(pyparsing.ParseException):
- controller.statistics.process_db_query('ipaddress(macaddress in most_used(macaddress(ipaddress in'
- ' least_used(ipaddress))))')
|