def test_init_okta(self, okta_mock, _config_mock):
okta_mock.OktaSaml = mock.MagicMock()
keyman = Keyman(['foo', '-o', 'foo', '-u', 'bar', '-a', 'baz'])
keyman.init_okta('troz')
okta_mock.OktaSaml.assert_has_calls(
[mock.call(mock.ANY, mock.ANY, 'troz', duo_factor=mock.ANY)])
def test_init_okta_with_empty_input(self, okta_mock, _config_mock):
okta_mock.EmptyInput = BaseException
okta_mock.OktaSaml = mock.MagicMock()
okta_mock.OktaSaml.side_effect = okta.EmptyInput
keyman = Keyman(['foo', '-o', 'foo', '-u', 'bar', '-a', 'baz'])
with self.assertRaises(SystemExit):
keyman.init_okta('troz')
def test_init_okta_with_oktapreview(self, okta_mock, _config_mock):
okta_mock.OktaSaml = mock.MagicMock()
keyman = Keyman(['foo', '-o', 'foo', '-u', 'bar', '-a', 'baz'])
keyman.config.oktapreview = True
keyman.init_okta('troz')
okta_mock.OktaSaml.assert_has_calls(
[mock.call(mock.ANY, mock.ANY, 'troz', oktapreview=True)])
def test_init_okta_with_empty_input(self, okta_mock, _config_mock):
okta_mock.EmptyInput = BaseException
okta_mock.OktaSaml = mock.MagicMock()
okta_mock.OktaSaml.side_effect = okta.EmptyInput
keyman = Keyman(["foo", "-o", "foo", "-u", "bar", "-a", "baz"])
with self.assertRaises(SystemExit):
keyman.init_okta("troz")
def test_init_okta(self, okta_mock, _config_mock):
okta_mock.OktaSaml = mock.MagicMock()
keyman = Keyman(["foo", "-o", "foo", "-u", "bar", "-a", "baz"])
keyman.init_okta("troz")
okta_mock.OktaSaml.assert_has_calls(
[
mock.call(mock.ANY, mock.ANY, "troz", duo_factor=mock.ANY),
],
)
def test_init_okta_with_oktapreview(self, okta_mock, _config_mock):
okta_mock.OktaSaml = mock.MagicMock()
keyman = Keyman(["foo", "-o", "foo", "-u", "bar", "-a", "baz"])
keyman.config.oktapreview = True
keyman.init_okta("troz")
okta_mock.OktaSaml.assert_has_calls(
[
mock.call(
mock.ANY,
mock.ANY,
"troz",
mock.ANY,
oktapreview=True,
),
],
)
def test_main_aws_auth_error(self, _config_mock):
keyman = Keyman(["foo", "-o", "foo", "-u", "bar", "-a", "baz"])
keyman.handle_appid_selection = mock.MagicMock()
keyman.handle_appid_selection.side_effect = NoAWSAccounts()
keyman.user_password = mock.MagicMock()
keyman.user_password.return_value = "foo"
keyman.init_okta = mock.MagicMock()
keyman.auth_okta = mock.MagicMock()
keyman.aws_auth_loop = mock.MagicMock()
with self.assertRaises(SystemExit):
keyman.main()
def test_main_no_aws_accounts(self, _config_mock):
keyman = Keyman(['foo', '-o', 'foo', '-u', 'bar', '-a', 'baz'])
keyman.handle_appid_selection = mock.MagicMock()
keyman.user_password = mock.MagicMock()
keyman.user_password.return_value = 'foo'
keyman.init_okta = mock.MagicMock()
keyman.auth_okta = mock.MagicMock()
keyman.aws_auth_loop = mock.MagicMock()
keyman.aws_auth_loop.return_value = 1
with self.assertRaises(SystemExit):
keyman.main()
def test_main_update(self, config_mock):
config_mock().update = True
keyman = Keyman(["foo", "-o", "foo", "-u", "bar", "-a", "baz"])
keyman.update = mock.MagicMock()
keyman.handle_appid_selection = mock.MagicMock()
keyman.user_password = mock.MagicMock()
keyman.user_password.return_value = "foo"
keyman.init_okta = mock.MagicMock()
keyman.auth_okta = mock.MagicMock()
keyman.aws_auth_loop = mock.MagicMock()
with self.assertRaises(SystemExit):
keyman.main()
keyman.update.assert_has_calls([mock.call(__version__)])
def test_main_post_okta_appid_selection(self, _config_mock):
keyman = Keyman(['foo', '-o', 'foo', '-u', 'bar'])
keyman.handle_appid_selection = mock.MagicMock()
keyman.config.appid = None
keyman.user_password = mock.MagicMock()
keyman.user_password.return_value = 'foo'
keyman.init_okta = mock.MagicMock()
keyman.auth_okta = mock.MagicMock()
keyman.aws_auth_loop = mock.MagicMock()
keyman.aws_auth_loop.return_value = None
keyman.handle_appid_selection = mock.MagicMock()
keyman.main()
keyman.handle_appid_selection.assert_has_calls(
[mock.call(), mock.call(okta_ready=True)])
def test_main(self, _config_mock):
keyman = Keyman(['foo', '-o', 'foo', '-u', 'bar', '-a', 'baz'])
keyman.handle_appid_selection = mock.MagicMock()
keyman.user_password = mock.MagicMock()
keyman.user_password.return_value = 'foo'
keyman.init_okta = mock.MagicMock()
keyman.auth_okta = mock.MagicMock()
keyman.aws_auth_loop = mock.MagicMock()
keyman.main()
assert keyman.handle_appid_selection.called
assert keyman.user_password.called
keyman.init_okta.assert_called_with('foo')
assert keyman.auth_okta.called
assert keyman.aws_auth_loop.called
def test_main(self, _config_mock):
keyman = Keyman(["foo", "-o", "foo", "-u", "bar", "-a", "baz"])
keyman.handle_appid_selection = mock.MagicMock()
keyman.user_password = mock.MagicMock()
keyman.user_password.return_value = "foo"
keyman.init_okta = mock.MagicMock()
keyman.auth_okta = mock.MagicMock()
keyman.aws_auth_loop = mock.MagicMock()
keyman.aws_auth_loop.return_value = None
keyman.main()
assert keyman.handle_appid_selection.called
assert keyman.user_password.called
keyman.init_okta.assert_called_with("foo")
assert keyman.auth_okta.called
assert keyman.aws_auth_loop.called
