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_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)