def test_mfa_is_passed_to_sts(self, mock_sts):
stub_token = Struct({'credentials': None})
mock_conn = MagicMock()
mock_conn.assume_role.return_value = stub_token
mock_sts.connect_to_region.return_value = mock_conn
arn = 'arn:role/developer'
session_name = 'dev-session'
Actions.user_token('un-south-1',
arn, session_name,
mfa_serial_number='arn:11111',
mfa_token='123456')
mock_conn.assume_role.assert_called_with(arn, session_name,
mfa_serial_number='arn:11111',
mfa_token='123456')
def test_mfa_is_passed_to_sts(self, mock_sts):
stub_token = Struct({'credentials': None})
mock_conn = MagicMock()
mock_conn.assume_role.return_value = stub_token
mock_sts.connect_to_region.return_value = mock_conn
arn = 'arn:role/developer'
session_name = 'dev-session'
Actions.user_token('un-south-1',
arn,
session_name,
mfa_serial_number='arn:11111',
mfa_token='123456')
mock_conn.assume_role.assert_called_with(arn,
session_name,
mfa_serial_number='arn:11111',
mfa_token='123456')
def test_credentials_are_generated_from_user(self, mock_sts):
stub_token = Struct({'credentials': None})
mock_conn = MagicMock()
mock_conn.assume_role.return_value = stub_token
mock_sts.connect_to_region.return_value = mock_conn
arn = 'arn:role/developer'
session_name = 'dev-session'
token = Actions.user_token('un-south-1', arn, session_name)
mock_conn.assume_role.assert_called_with(arn,
session_name,
mfa_serial_number=None,
mfa_token=None)
self.assertEqual(token, stub_token)
def test_credentials_are_generated_from_user(self, mock_sts):
stub_token = Struct({'credentials': None})
mock_conn = MagicMock()
mock_conn.assume_role.return_value = stub_token
mock_sts.connect_to_region.return_value = mock_conn
arn = 'arn:role/developer'
session_name = 'dev-session'
token = Actions.user_token('un-south-1',
arn, session_name)
mock_conn.assume_role.assert_called_with(arn, session_name,
mfa_serial_number=None,
mfa_token=None)
self.assertEqual(token, stub_token)
def user_action(args):
token_action(args)(Actions.user_token(**args))
def user_action(args):
token_action(args)(Actions.user_token(**args))
