def test_check_full_happy_path(self):
sys.argv = [
'dagda.py', 'vuln', '--product', 'openldap', '--product_version',
'2.2.20'
]
parsed_args = VulnCLIParser()
self.assertEqual(parsed_args.get_product(), 'openldap')
self.assertEqual(parsed_args.get_product_version(), '2.2.20')
def __init__(self):
super(DagdaCLIParser, self).__init__()
self.parser = DagdaGlobalParser(prog='dagda.py',
usage=dagda_global_parser_text,
add_help=False)
self.parser.add_argument('command',
choices=[
'vuln', 'check', 'history', 'start',
'monitor', 'docker', 'agent'
])
self.parser.add_argument('-h', '--help', action=_HelpAction)
self.parser.add_argument('-v',
'--version',
action='version',
version='%(prog)s 0.9.0')
self.args, self.unknown = self.parser.parse_known_args()
if self.get_command() == 'vuln':
self.extra_args = VulnCLIParser()
elif self.get_command() == 'check':
self.extra_args = CheckCLIParser()
elif self.get_command() == 'history':
self.extra_args = HistoryCLIParser()
elif self.get_command() == 'start':
self.extra_args = StartCLIParser()
elif self.get_command() == 'monitor':
self.extra_args = MonitorCLIParser()
elif self.get_command() == 'docker':
self.extra_args = DockerCLIParser()
elif self.get_command() == 'agent':
self.extra_args = AgentCLIParser()
def test_bad_rhsa_info(self):
args = generate_args(False, False, None, None, None, None, None, None, None, None, None, None, None, 'RHSA-201222:002')
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 24)
def test_not_only_init_status(self):
args = generate_args(False, True, None, None, 12345, None, None, None,
None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 3)
def test_bad_exploit_db(self):
args = generate_args(False, False, None, None, -12345, None, None)
status = VulnCLIParser.verify_args("dagda.py vuln", args)
self.assertEqual(status, 9)
def test_ok(self):
args = generate_args(False, False, None, None, None, 'openldap',
'2.2.20')
status = VulnCLIParser.verify_args("dagda.py vuln", args)
self.assertEqual(status, 0)
def test_not_only_cve(self):
args = generate_args(False, False, 'CVE-2002-1562', 12345, None, None,
None)
status = VulnCLIParser.verify_args("dagda.py vuln", args)
self.assertEqual(status, 4)
def test_not_only_bid(self):
args = generate_args(False, False, None, 12345, 12345, None, None)
status = VulnCLIParser.verify_args("dagda.py vuln", args)
self.assertEqual(status, 6)
def test_empty_args(self):
empty_args = generate_args(False, False, None, None, None, None, None, None, None, None, None, None, None, None)
status = VulnCLIParser.verify_args(empty_args)
self.assertEqual(status, 1)
def test_not_only_bid_info(self):
args = generate_args(False, False, None, None, None, 12345, 'openldap',
None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 10)
def test_get_exploit_db_info_id(self):
sys.argv = ['dagda.py', 'vuln', '--exploit_db_info', '15']
parsed_args = VulnCLIParser()
self.assertEqual(parsed_args.get_exploit_db_info_id(), 15)
def test_check_full_happy_path(self):
sys.argv = ['dagda.py', 'vuln', '--product', 'openldap', '--product_version', '2.2.20']
parsed_args = VulnCLIParser()
self.assertEqual(parsed_args.get_product(), 'openldap')
self.assertEqual(parsed_args.get_product_version(), '2.2.20')
def test_get_bid_info(self):
sys.argv = ['dagda.py', 'vuln', '--bid_info', '15']
parsed_args = VulnCLIParser()
self.assertEqual(parsed_args.get_bid_info(), 15)
def test_get_cve_info(self):
sys.argv = ['dagda.py', 'vuln', '--cve_info', 'CVE-2002-2002']
parsed_args = VulnCLIParser()
self.assertEqual(parsed_args.get_cve_info(), 'CVE-2002-2002')
def test_ok(self):
args = generate_args(False, False, None, None, None, None, None, None, 'openldap', '2.2.20', None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 0)
def test_not_only_cveinfo(self):
args = generate_args(False, False, None, 'CVE-2002-1562', None, 12345,
None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 6)
def test_not_only_init_status(self):
args = generate_args(False, True, None, None, 12345, None, None, None, None, None, None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 3)
def test_bad_cveinfo(self):
args = generate_args(False, False, None, 'CVE-62', None, None, None,
None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 7)
def test_not_only_cveinfo(self):
args = generate_args(False, False, None,'CVE-2002-1562', None, 12345, None, None, None, None, None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 6)
def test_bad_bid_info(self):
args = generate_args(False, False, None, None, None, -12345, None,
None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 11)
def test_bad_cveinfo(self):
args = generate_args(False, False, None, 'CVE-62', None, None, None, None, None, None, None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 7)
def test_bad_cve(self):
args = generate_args(False, False, 'CVE-62', None, None, None, None)
status = VulnCLIParser.verify_args("dagda.py vuln", args)
self.assertEqual(status, 5)
def test_not_only_bid(self):
args = generate_args(False, False, None, None, 12345, None, 'openldap', None, None, None, None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 8)
def test_not_only_exploit_db(self):
args = generate_args(False, False, None, None, 12345, 'openldap', None)
status = VulnCLIParser.verify_args("dagda.py vuln", args)
self.assertEqual(status, 8)
def test_empty_args(self):
empty_args = generate_args(False, False, None, None, None, None, None)
status = VulnCLIParser.verify_args("dagda.py vuln", empty_args)
self.assertEqual(status, 1)
def test_only_product_version(self):
args = generate_args(False, False, None, None, None, None, '2.30')
status = VulnCLIParser.verify_args("dagda.py vuln", args)
self.assertEqual(status, 10)
def test_bad_rhba_info(self):
args = generate_args(False, False, None, None, None, None, None, None, None, None, None, 'RHBA--2012:002', None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 20)
def test_bad_exploit_db(self):
args = generate_args(False, False, None, None, None, None, -12345, None, None, None, None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 13)
def test_not_only_rhsa_info(self):
args = generate_args(False, False, None, None, None, None, None, None, 'openldap', None, None, None, None, 'RHSA-2012:002')
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 23)
def test_not_only_exploit_db_info(self):
args = generate_args(False, False, None, None, None, None, None, 12345, 'openldap', None, None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 14)
def test_only_product_version(self):
args = generate_args(False, False, None, None, None, None, None, None, None, '2.30', None, None, None, None)
status = VulnCLIParser.verify_args(args)
self.assertEqual(status, 16)
