Added the ability to use AWSCredentialsProvider's that require non-empty contructor args e.g. ProfileCredentialsProvider where you provide the profile name to use from your ~/.aws/credentials file

pom.xml

@@ -52,6 +52,11 @@
       <artifactId>aws-java-sdk-cloudwatch</artifactId>
       <version>${aws-java-sdk.version}</version>
     </dependency>
+    <dependency>
+      <groupId>com.amazonaws</groupId>
+      <artifactId>aws-java-sdk-sts</artifactId>
+      <version>${aws-java-sdk.version}</version>
+    </dependency>
     <dependency>
       <groupId>com.google.guava</groupId>
       <artifactId>guava</artifactId>

src/main/java/com/amazonaws/services/kinesis/clientlibrary/config/AWSCredentialsProviderPropertyValueDecoder.java

@@ -17,6 +17,7 @@ package com.amazonaws.services.kinesis.clientlibrary.config;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
+import java.lang.reflect.Constructor;
 
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
@@ -30,6 +31,7 @@ class AWSCredentialsProviderPropertyValueDecoder implements IPropertyValueDecode
     private static final Log LOG = LogFactory.getLog(AWSCredentialsProviderPropertyValueDecoder.class);
     private static final String AUTH_PREFIX = "com.amazonaws.auth.";
     private static final String LIST_DELIMITER = ",";
+    private static final String ARG_DELIMITER = "|";
 
     /**
      * Constructor.
@@ -39,7 +41,7 @@ class AWSCredentialsProviderPropertyValueDecoder implements IPropertyValueDecode
 
     /**
      * Get AWSCredentialsProvider property.
-     * 
+     *
      * @param value property value as String
      * @return corresponding variable in correct type
      */
@@ -70,11 +72,25 @@ class AWSCredentialsProviderPropertyValueDecoder implements IPropertyValueDecode
     private static List<AWSCredentialsProvider> getValidCredentialsProviders(List<String> providerNames) {
         List<AWSCredentialsProvider> credentialsProviders = new ArrayList<AWSCredentialsProvider>();
         for (String providerName : providerNames) {
-            try {
+            if (providerName.contains(ARG_DELIMITER)) {
-                Class<?> className = Class.forName(providerName);
+              String[] nameAndArgs = providerName.split("\\" + ARG_DELIMITER);
-                credentialsProviders.add((AWSCredentialsProvider) className.newInstance());
+              Class<?>[] argTypes = new Class<?>[nameAndArgs.length - 1];
-            } catch (Exception e) {
+              Arrays.fill(argTypes, String.class);
-                LOG.debug("Can't find any credentials provider matching " + providerName + ".");
+              try {
+                  Class<?> className = Class.forName(nameAndArgs[0]);
+                  Constructor<?> c = className.getConstructor(argTypes);
+                  credentialsProviders.add((AWSCredentialsProvider) c.newInstance(
+                        Arrays.copyOfRange(nameAndArgs, 1, nameAndArgs.length)));
+              } catch (Exception e) {
+                  LOG.debug("Can't find any credentials provider matching " + providerName + ".");
+              }
+            } else {
+              try {
+                  Class<?> className = Class.forName(providerName);
+                  credentialsProviders.add((AWSCredentialsProvider) className.newInstance());
+              } catch (Exception e) {
+                  LOG.debug("Can't find any credentials provider matching " + providerName + ".");
+              }
             }
         }
         return credentialsProviders;
@@ -97,13 +113,13 @@ class AWSCredentialsProviderPropertyValueDecoder implements IPropertyValueDecode
     private static List<String> getPossibleFullClassNames(String s) {
         /*
          * We take care of three cases :
-         * 
+         *
          * 1. Customer provides a short name of common providers in com.amazonaws.auth package i.e. any classes
          * implementing the AWSCredentialsProvider interface:
          * http://docs.aws.amazon.com/AWSJavaSDK/latest/javadoc/com/amazonaws/auth/AWSCredentialsProvider.html
-         * 
+         *
          * 2. Customer provides a full name of common providers e.g. com.amazonaws.auth.ClasspathFileCredentialsProvider
-         * 
+         *
          * 3. Customer provides a custom credentials provider with full name of provider
          */
 

src/test/java/com/amazonaws/services/kinesis/clientlibrary/config/AWSCredentialsProviderPropertyValueDecoderTest.java

@@ -0,0 +1,104 @@
+/*
+ * Copyright 2012-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Amazon Software License (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/asl/
+ *
+ * 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.services.kinesis.clientlibrary.config;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+import java.io.ByteArrayInputStream;
+import java.io.InputStream;
+import java.util.Set;
+import java.io.File;
+import java.io.IOException;
+
+import org.apache.commons.lang.StringUtils;
+import org.junit.Test;
+
+import com.amazonaws.auth.AWSCredentials;
+import com.amazonaws.auth.AWSCredentialsProvider;
+import com.amazonaws.auth.AWSCredentialsProviderChain;
+import com.amazonaws.auth.profile.ProfileCredentialsProvider;
+import com.amazonaws.services.kinesis.clientlibrary.config.AWSCredentialsProviderPropertyValueDecoder;
+
+public class AWSCredentialsProviderPropertyValueDecoderTest {
+
+    private String credentialName1 =
+            "com.amazonaws.services.kinesis.clientlibrary.config.KinesisClientLibConfiguratorTest$AlwaysSucceedCredentialsProvider";
+    private String credentialName2 =
+            "com.amazonaws.services.kinesis.clientlibrary.config.KinesisClientLibConfiguratorTest$AlwaysFailCredentialsProvider";
+    private AWSCredentialsProviderPropertyValueDecoder decoder = new AWSCredentialsProviderPropertyValueDecoder();
+
+    @Test
+    public void testSingleProvider() {
+        AWSCredentialsProvider provider = decoder.decodeValue(credentialName1);
+        assertEquals(provider.getClass(), AWSCredentialsProviderChain.class);
+    }
+
+    @Test
+    public void testTwoProviders() {
+        AWSCredentialsProvider provider = decoder.decodeValue(credentialName1 + "," + credentialName2);
+        assertEquals(provider.getClass(), AWSCredentialsProviderChain.class);
+    }
+
+    @Test
+    public void testProfileProviderWithOneArg() {
+        AWSCredentialsProvider provider = decoder.decodeValue(ProfileCredentialsProvider.class.getName() + "|profileName");
+        assertEquals(provider.getClass(), AWSCredentialsProviderChain.class);
+    }
+
+    @Test
+    public void testProfileProviderWithTwoArgs() throws IOException {
+        File temp = File.createTempFile("test-profiles-file", ".tmp");
+        temp.deleteOnExit();
+        AWSCredentialsProvider provider = decoder.decodeValue(ProfileCredentialsProvider.class.getName() +
+            "|" + temp.getAbsolutePath() + "|profileName");
+        assertEquals(provider.getClass(), AWSCredentialsProviderChain.class);
+    }
+
+   /**
+     * This credentials provider will always succeed
+     */
+    public static class AlwaysSucceedCredentialsProvider implements AWSCredentialsProvider {
+
+        @Override
+        public AWSCredentials getCredentials() {
+            return null;
+        }
+
+        @Override
+        public void refresh() {
+        }
+
+    }
+
+    /**
+     * This credentials provider will always fail
+     */
+    public static class AlwaysFailCredentialsProvider implements AWSCredentialsProvider {
+
+        @Override
+        public AWSCredentials getCredentials() {
+            throw new IllegalArgumentException();
+        }
+
+        @Override
+        public void refresh() {
+        }
+
+    }
+}

src/test/java/com/amazonaws/services/kinesis/clientlibrary/lib/worker/KinesisClientLibConfigurationTest.java

@@ -56,6 +56,7 @@ public class KinesisClientLibConfigurationTest {
         // Test constructor with all valid arguments.
         config =
                 new KinesisClientLibConfiguration(TEST_STRING,
+                        TEST_STRING,
                         TEST_STRING,
                         TEST_STRING,
                         null,
@@ -93,6 +94,7 @@ public class KinesisClientLibConfigurationTest {
             try {
                 config =
                         new KinesisClientLibConfiguration(TEST_STRING,
+                                TEST_STRING,
                                 TEST_STRING,
                                 TEST_STRING,
                                 null,
@@ -126,6 +128,7 @@ public class KinesisClientLibConfigurationTest {
             try {
                 config =
                         new KinesisClientLibConfiguration(TEST_STRING,
+                                TEST_STRING,
                                 TEST_STRING,
                                 TEST_STRING,
                                 null,
@@ -203,7 +206,7 @@ public class KinesisClientLibConfigurationTest {
         AmazonDynamoDBClient dclient = Mockito.mock(AmazonDynamoDBClient.class);
         AmazonCloudWatchClient cclient = Mockito.mock(AmazonCloudWatchClient.class);
         Region region = RegionUtils.getRegion("us-west-2");
-        
+
         AWSCredentialsProvider credentialsProvider = Mockito.mock(AWSCredentialsProvider.class);
         KinesisClientLibConfiguration kclConfig =
                 new KinesisClientLibConfiguration("Test", "Test", credentialsProvider, "0")
@@ -256,7 +259,7 @@ public class KinesisClientLibConfigurationTest {
 
         Mockito.verify(kclConfig, Mockito.times(9)).getRegionName();
         Mockito.verify(kclConfig, Mockito.times(4)).getKinesisEndpoint();
-        
+
         kclConfig = Mockito.spy(
                 new KinesisClientLibConfiguration("Test", "Test", credentialsProvider, "0")
                         .withKinesisEndpoint("https://kinesis.eu-west-1.amazonaws.com"));
@@ -288,6 +291,7 @@ public class KinesisClientLibConfigurationTest {
         Mockito.mock(AWSCredentialsProvider.class);
         try {
             new KinesisClientLibConfiguration(TEST_STRING,
+                        TEST_STRING,
                         TEST_STRING,
                         TEST_STRING,
                         null,