/*
* 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);
}
}