CredentialUtilsTest.java

/*
 * Copyright 2015-2017 Amazon.com, Inc. or its affiliates. All Rights
 * Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file is
 * distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either
 * express or implied. See the License for the specific language
 * governing
 * permissions and limitations under the License.
 */
package com.amazonaws.util;

import com.amazonaws.AmazonWebServiceRequest;
import com.amazonaws.RequestConfig;
import com.amazonaws.auth.AWSCredentials;
import com.amazonaws.auth.AWSCredentialsProvider;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.auth.DefaultAWSCredentialsProviderChain;
import com.amazonaws.internal.StaticCredentialsProvider;

import org.hamcrest.Matchers;
import org.junit.Assert;
import org.junit.Test;

import utils.model.EmptyAmazonWebServiceRequest;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class CredentialUtilsTest {

    private static final AWSCredentialsProvider CLIENT_CREDENTIALS = new DefaultAWSCredentialsProviderChain();

    @Test
    public void request_credentials_takes_precendence_over_client_credentials() {
        final String awsAccessKeyId = "foo";
        final String awsSecretAccessKey = "bar";
        final AWSCredentials reqCredentials = new BasicAWSCredentials(awsAccessKeyId,
                                                                      awsSecretAccessKey);
        EmptyAmazonWebServiceRequest req = new EmptyAmazonWebServiceRequest();
        req.setRequestCredentials(reqCredentials);
        AWSCredentialsProvider actual = CredentialUtils.getCredentialsProvider(req,
                                                                               null);

        Assert.assertThat(actual, Matchers.instanceOf(StaticCredentialsProvider
                                                              .class));
        assertEquals(awsAccessKeyId, actual.getCredentials().getAWSAccessKeyId());
        assertEquals(awsSecretAccessKey, actual.getCredentials().getAWSSecretKey());
    }

    @Test
    public void base_credentials_returned_when_no_request_credentials_is_present() {
        final String awsAccessKeyId = "foo";
        final String awsSecretAccessKey = "bar";
        final StaticCredentialsProvider base = new StaticCredentialsProvider
                (new BasicAWSCredentials(awsAccessKeyId, awsSecretAccessKey));

        AWSCredentialsProvider actual = CredentialUtils
                .getCredentialsProvider((AmazonWebServiceRequest) null, base);
        Assert.assertThat(actual, Matchers.instanceOf(StaticCredentialsProvider
                                                              .class));
        assertEquals(awsAccessKeyId, actual.getCredentials().getAWSAccessKeyId());
        assertEquals(awsSecretAccessKey, actual.getCredentials().getAWSSecretKey());
    }

    @Test
    public void requestCredentialsInRequestConfig_TakesPrecedenceOverClientCredentials() {
        AWSCredentialsProvider requestCredentials = mock(AWSCredentialsProvider.class);
        RequestConfig requestConfig = mock(RequestConfig.class);
        when(requestConfig.getCredentialsProvider()).thenReturn(requestCredentials);
        AWSCredentialsProvider actual = CredentialUtils
                .getCredentialsProvider(requestConfig, CLIENT_CREDENTIALS);
        assertEquals(requestCredentials, actual);
    }

    @Test
    public void requestCredentialsNotSetInRequestConfig_ReturnsClientCredentials() {
        RequestConfig requestConfig = mock(RequestConfig.class);
        AWSCredentialsProvider actual = CredentialUtils
                .getCredentialsProvider(requestConfig, CLIENT_CREDENTIALS);
        assertEquals(CLIENT_CREDENTIALS, actual);
    }
}