diff options
Diffstat (limited to 'tests/dnsviz_grok_options.py')
-rw-r--r-- | tests/dnsviz_grok_options.py | 51 |
1 files changed, 34 insertions, 17 deletions
diff --git a/tests/dnsviz_grok_options.py b/tests/dnsviz_grok_options.py index 8d6ebc5..346459e 100644 --- a/tests/dnsviz_grok_options.py +++ b/tests/dnsviz_grok_options.py @@ -93,34 +93,39 @@ class DNSVizGrokOptionsTestCase(unittest.TestCase): try: args = ['-r', example_auth_out.name] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.ingest_input() # Bad json args = ['-r', example_bad_json.name] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) with self.assertRaises(AnalysisInputError): arghelper.ingest_input() # No version args = ['-r', example_no_version.name] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) with self.assertRaises(AnalysisInputError): arghelper.ingest_input() # Invalid version args = ['-r', example_invalid_version_1.name] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) with self.assertRaises(AnalysisInputError): arghelper.ingest_input() # Invalid version args = ['-r', example_invalid_version_2.name] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) with self.assertRaises(AnalysisInputError): arghelper.ingest_input() @@ -132,7 +137,8 @@ class DNSVizGrokOptionsTestCase(unittest.TestCase): def test_ingest_names(self): args = ['example.com', 'example.net'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.ingest_names() self.assertEqual(list(arghelper.names), [dns.name.from_text('example.com'), dns.name.from_text('example.net')]) @@ -145,20 +151,23 @@ class DNSVizGrokOptionsTestCase(unittest.TestCase): try: args = ['-f', names_file.name] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.ingest_names() self.assertEqual(list(arghelper.names), [dns.name.from_text('example.com'), dns.name.from_text('example.net')]) args = ['-r', example_names_only.name] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.ingest_input() arghelper.ingest_names() self.assertEqual(list(arghelper.names), [dns.name.from_text('example.com'), dns.name.from_text('example.net'), dns.name.from_text('example.org')]) args = ['-r', example_names_only.name, 'example.com'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.ingest_input() arghelper.ingest_names() self.assertEqual(list(arghelper.names), [dns.name.from_text('example.com')]) @@ -176,7 +185,8 @@ class DNSVizGrokOptionsTestCase(unittest.TestCase): args = ['example.com'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.aggregate_trusted_key_info() self.assertEqual(arghelper.trusted_keys, None) arghelper.update_trusted_key_info() @@ -191,14 +201,16 @@ class DNSVizGrokOptionsTestCase(unittest.TestCase): try: args = ['-t', tk1_file.name, '-t', tk2_file.name, 'example.com'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.aggregate_trusted_key_info() arghelper.update_trusted_key_info() self.assertEqual(arghelper.trusted_keys, tk_explicit) args = ['-t', '/dev/null', 'example.com'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.aggregate_trusted_key_info() arghelper.update_trusted_key_info() self.assertEqual(arghelper.trusted_keys, []) @@ -212,7 +224,8 @@ class DNSVizGrokOptionsTestCase(unittest.TestCase): # Names file and command-line domain names are mutually exclusive args = ['-f', '/dev/null', 'example.com'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) with self.assertRaises(argparse.ArgumentTypeError): arghelper.check_args() @@ -221,25 +234,29 @@ class DNSVizGrokOptionsTestCase(unittest.TestCase): # Names file and command-line domain names are mutually exclusive args = [] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.set_kwargs() self.assertEqual(arghelper.log_level, logging.DEBUG) args = ['-l', 'info'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.set_kwargs() self.assertEqual(arghelper.log_level, logging.INFO) args = ['-l', 'warning'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.set_kwargs() self.assertEqual(arghelper.log_level, logging.WARNING) args = ['-l', 'error'] arghelper = GrokArgHelper(self.logger) - arghelper.build_parser('grok', args) + arghelper.build_parser('grok') + arghelper.parse_args(args) arghelper.set_kwargs() self.assertEqual(arghelper.log_level, logging.ERROR) |