def test_provider_not_registered_on_error(self):
session = mock.Mock()
session.get_component.side_effect = Exception(
"Couldn't get credential_provider.")
assumerole.inject_assume_role_provider(
session, event_name='building-command-table.foo')
self.assertFalse(
session.get_component.return_value.insert_before.called)
def test_assume_role_provider_not_injected_for_main_command_table(self):
session = mock.Mock()
# When the main/top-level command table is created, it's emitted with
# an event name of building-command-table.main. We want to verify
# that the assumerole provider is not hooked up when that happens.
assumerole.inject_assume_role_provider(
session, event_name='building-command-table.main')
self.assertFalse(session.get_component.called)
def test_provider_not_registered_on_error(self):
session = mock.Mock()
session.get_component.side_effect = Exception(
"Couldn't get credential_provider.")
assumerole.inject_assume_role_provider(
session, event_name='building-command-table.foo')
self.assertFalse(
session.get_component.return_value.insert_before.called)
def test_assume_role_provider_injected(self):
session = mock.Mock()
assumerole.inject_assume_role_provider(
session, event_name='building-command-table.foo')
session.get_component.assert_called_with('credential_provider')
credential_provider = session.get_component.return_value
call_args = credential_provider.insert_before.call_args[0]
self.assertEqual(call_args[0], 'shared-credentials-file')
self.assertIsInstance(call_args[1], assumerole.AssumeRoleProvider)
def test_assume_role_provider_injected(self):
session = mock.Mock()
assumerole.inject_assume_role_provider(
session, event_name='building-command-table.foo')
session.get_component.assert_called_with('credential_provider')
credential_provider = session.get_component.return_value
call_args = credential_provider.insert_before.call_args[0]
self.assertEqual(call_args[0], 'shared-credentials-file')
self.assertIsInstance(call_args[1], assumerole.AssumeRoleProvider)
