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_check_falco_rules_file(self):
sys.argv = [
'dagda.py', 'start', '--falco_rules_file',
'./tests/mock_files/falco_rules.yaml'
]
args = StartCLIParser()
self.assertEqual(args.get_falco_rules_filename(),
'./tests/mock_files/falco_rules.yaml')
def test_start_full_happy_path(self):
sys.argv = ['dagda.py', 'start', '-s', '127.0.0.1', '-p', '5000', '-m', '127.0.0.1', '-mp', '27017']
parsed_args = StartCLIParser()
self.assertEqual(parsed_args.get_server_host(), '127.0.0.1')
self.assertEqual(parsed_args.get_server_port(), 5000)
self.assertEqual(parsed_args.get_mongodb_host(), '127.0.0.1')
self.assertEqual(parsed_args.get_mongodb_port(), 27017)
self.assertFalse(parsed_args.is_mongodb_ssl_enabled())
self.assertIsNone(parsed_args.get_mongodb_user())
self.assertIsNone(parsed_args.get_mongodb_pass())
self.assertIsNone(parsed_args.get_falco_rules_filename())
def test_check_exit_1(self):
sys.argv = ['dagda.py', 'start', '-p', '-1']
with self.assertRaises(SystemExit) as cm:
StartCLIParser()
self.assertEqual(cm.exception.code, 1)
def test_check_none_external_falco(self):
sys.argv = ['dagda.py', 'start']
args = StartCLIParser()
self.assertEqual(args.get_external_falco_output_filename(), None)