diff options
author | Casey Deccio <casey@deccio.net> | 2020-10-08 18:59:26 +0300 |
---|---|---|
committer | Casey Deccio <casey@deccio.net> | 2020-10-08 18:59:26 +0300 |
commit | b4fb6a0be6dce62a000a0937a78f79ac677d33ed (patch) | |
tree | 6e8302fe0930db763c4a354434249b2790088c98 /tests | |
parent | ea689b63ff33efe8d4920f06cae3b7f07f5a0115 (diff) |
Allow blank argument for RR types
Diffstat (limited to 'tests')
-rw-r--r-- | tests/dnsviz_probe_tests.py (renamed from tests/command_line_tests.py) | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tests/command_line_tests.py b/tests/dnsviz_probe_tests.py index d8ea477..7c3afb6 100644 --- a/tests/command_line_tests.py +++ b/tests/dnsviz_probe_tests.py @@ -409,12 +409,17 @@ class CommandLineTestCase(unittest.TestCase): arg1_with_spaces = ' A , AAAA , MX , CNAME ' arg2 = 'A' arg3 = 'A,BLAH' + arg4_empty = '' + arg4_empty_spaces = ' ' type_list1 = [dns.rdatatype.A, dns.rdatatype.AAAA, dns.rdatatype.MX, dns.rdatatype.CNAME] type_list2 = [dns.rdatatype.A] + empty_list = [] self.assertEqual(ArgHelper.comma_separated_dns_types(arg1), type_list1) self.assertEqual(ArgHelper.comma_separated_dns_types(arg1_with_spaces), type_list1) + self.assertEqual(ArgHelper.comma_separated_dns_types(arg4_empty), empty_list) + self.assertEqual(ArgHelper.comma_separated_dns_types(arg4_empty_spaces), empty_list) # invalid schema with self.assertRaises(argparse.ArgumentTypeError): |