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_fail_external_falco(self):
temporary_dir = tempfile.mkdtemp()
filename = temporary_dir + '/fail_falco_rules'
with open(filename, 'a+') as f:
f.write('{}$##')
f.flush()
args = generate_args(None, None, None, None, False, None, None, open(filename, 'rb+'), open(filename, 'rb+'))
status = StartCLIParser.verify_args(args)
os.remove(filename)
shutil.rmtree(temporary_dir)
self.assertEqual(status, 6)
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_fail_mongodb_port(self):
args = generate_args(None, None, None, 65536, False, None, None, None, None)
status = StartCLIParser.verify_args(args)
self.assertEqual(status, 2)
def test_fail_only_mongodb_pass(self):
args = generate_args(None, None, None, None, False, None, '1234', None, None)
status = StartCLIParser.verify_args(args)
self.assertEqual(status, 4)
def test_ok_server_ports(self):
args = generate_args(None, 5555, None, 27017, False, None, None, None, None)
status = StartCLIParser.verify_args(args)
self.assertEqual(status, 0)
def test_ok_empty_args(self):
args = generate_args(None, None, None, None, False, None, None, None, None)
status = StartCLIParser.verify_args(args)
self.assertEqual(status, 0)
def test_check_none_external_falco(self):
sys.argv = ['dagda.py', 'start']
args = StartCLIParser()
self.assertEqual(args.get_external_falco_output_filename(), None)
def test_fail_mongodb_port(self):
args = generate_args(None, None, None, 65536)
status = StartCLIParser.verify_args("dagda.py start", args)
self.assertEqual(status, 2)
def test_ok_server_ports(self):
args = generate_args(None, 5555, None, 27017)
status = StartCLIParser.verify_args("dagda.py start", args)
self.assertEqual(status, 0)
