def test_parse_args_no_req_main(self):
argv = ['aws_okta_keyman.py', '-D']
config = Config(argv)
config.parse_args(main_required=False)
# Should succeed without throwing due to missing args
self.assertEqual(config.debug, True)
def test_parse_args_req_main_missing(self):
argv = ['aws_okta_keyman.py', '-D']
config = Config(argv)
# Main required but not passed, should raise
with self.assertRaises(SystemExit):
config.parse_args(main_required=True)
def test_parse_args_req_main_present(self):
argv = [
'aws_okta_keyman.py', '-a', 'app/id', '-o', 'foobar', '-u', 'test'
]
config = Config(argv)
config.parse_args(main_required=True)
# Should succeed without throwing due to missing args
self.assertEqual(config.appid, 'app/id')
self.assertEqual(config.org, 'foobar')
self.assertEqual(config.username, 'test')
def test_parse_args_req_main_present(self):
argv = [
"aws_okta_keyman.py",
"-a",
"app/id",
"-o",
"foobar",
"-u",
"test",
]
config = Config(argv)
config.parse_args(main_required=True)
# Should succeed without throwing due to missing args
self.assertEqual(config.appid, "app/id")
self.assertEqual(config.org, "foobar")
self.assertEqual(config.username, "test")
def test_parse_args_verify_all_parsed_full(self):
argv = [
'aws_okta_keyman.py', '--appid', 'app/id', '--org', 'foobar',
'--username', 'test', '--name', 'profilename', '--config',
'config_file_path', '--writepath', 'write_file_path', '--debug',
'--reup'
]
config = Config(argv)
config.parse_args(main_required=True)
self.assertEquals(config.appid, 'app/id')
self.assertEquals(config.org, 'foobar')
self.assertEquals(config.username, 'test')
self.assertEquals(config.name, 'profilename')
self.assertEquals(config.config, 'config_file_path')
self.assertEquals(config.writepath, 'write_file_path')
self.assertEquals(config.debug, True)
self.assertEquals(config.reup, True)
def test_parse_args_verify_all_parsed_short(self):
argv = [
'aws_okta_keyman.py', '-a', 'app/id', '-o', 'foobar', '-u', 'test',
'-n', 'profilename', '-c', 'config_file_path', '-w',
'write_file_path', '-D', '-r', '-p'
]
config = Config(argv)
config.parse_args(main_required=True)
self.assertEquals(config.appid, 'app/id')
self.assertEquals(config.org, 'foobar')
self.assertEquals(config.username, 'test')
self.assertEquals(config.name, 'profilename')
self.assertEquals(config.config, 'config_file_path')
self.assertEquals(config.writepath, 'write_file_path')
self.assertEquals(config.debug, True)
self.assertEquals(config.reup, True)
self.assertEquals(config.oktapreview, True)
def test_parse_args_verify_all_parsed_short(self):
argv = [
"aws_okta_keyman.py",
"-a",
"app/id",
"-ac",
"accountname",
"-o",
"foobar",
"-u",
"test",
"-n",
"profilename",
"-ro",
"rolename",
"-c",
"config_file_path",
"-w",
"write_file_path",
"-d",
"push",
"-D",
"-r",
"-p",
]
config = Config(argv)
config.parse_args(main_required=True)
self.assertEqual(config.appid, "app/id")
self.assertEqual(config.account, "accountname")
self.assertEqual(config.org, "foobar")
self.assertEqual(config.username, "test")
self.assertEqual(config.name, "profilename")
self.assertEqual(config.role, "rolename")
self.assertEqual(config.config, "config_file_path")
self.assertEqual(config.writepath, "write_file_path")
self.assertEqual(config.duo_factor, "push")
self.assertEqual(config.debug, True)
self.assertEqual(config.reup, True)
self.assertEqual(config.oktapreview, True)
def test_parse_args_verify_all_parsed_full(self):
argv = [
"aws_okta_keyman.py",
"--account",
"accountname",
"--appid",
"app/id",
"--org",
"foobar",
"--username",
"test",
"--name",
"profilename",
"--role",
"rolename",
"--config",
"config_file_path",
"--writepath",
"write_file_path",
"--duo_factor",
"push",
"--debug",
"--reup",
]
config = Config(argv)
config.parse_args(main_required=True)
self.assertEqual(config.account, "accountname")
self.assertEqual(config.appid, "app/id")
self.assertEqual(config.org, "foobar")
self.assertEqual(config.username, "test")
self.assertEqual(config.name, "profilename")
self.assertEqual(config.role, "rolename")
self.assertEqual(config.config, "config_file_path")
self.assertEqual(config.writepath, "write_file_path")
self.assertEqual(config.duo_factor, "push")
self.assertEqual(config.debug, True)
self.assertEqual(config.reup, True)
