Adds tests.

Author: Nick Johnson

src/test/java/com/rabbitmq/client/test/ConnectionFactoryTest.java

@@ -21,6 +21,7 @@
 import com.rabbitmq.client.MetricsCollector;
 import com.rabbitmq.client.impl.AMQConnection;
 import com.rabbitmq.client.impl.ConnectionParams;
+import com.rabbitmq.client.impl.CredentialsProvider;
 import com.rabbitmq.client.impl.FrameHandler;
 import com.rabbitmq.client.impl.FrameHandlerFactory;
 import org.junit.Test;
@@ -29,8 +30,9 @@
 import java.util.Queue;
 import java.util.concurrent.ArrayBlockingQueue;
 import java.util.concurrent.TimeoutException;
+import java.util.concurrent.atomic.AtomicBoolean;
 
-import static org.junit.Assert.assertSame;
+import static org.junit.Assert.*;
 import static org.mockito.Mockito.*;
 
 public class ConnectionFactoryTest {
@@ -62,5 +64,30 @@ protected synchronized FrameHandlerFactory createFrameHandlerFactory() throws IO
         );
         assertSame(connectionThatSucceeds, returnedConnection);
     }
+    
+    // see https://github.com/rabbitmq/rabbitmq-java-client/pull/350
+    @Test public void customizeCredentialsProvider() throws Exception {
+        final CredentialsProvider provider = mock(CredentialsProvider.class);
+        final AMQConnection connection = mock(AMQConnection.class);
+        final AtomicBoolean createCalled = new AtomicBoolean(false);
+        
+        ConnectionFactory connectionFactory = new ConnectionFactory() {
+            @Override
+            protected AMQConnection createConnection(ConnectionParams params, FrameHandler frameHandler,
+                    MetricsCollector metricsCollector) {
+                assertSame(provider, params.getCredentialsProvider());
+                createCalled.set(true);
+                return connection;
+            }
+        };
+        connectionFactory.setCredentialsProvider(provider);
+        connectionFactory.setAutomaticRecoveryEnabled(false);
+        
+        doNothing().when(connection).start();
+        
+        Connection returnedConnection = connectionFactory.newConnection();
+        assertSame(returnedConnection, connection);
+        assertTrue(createCalled.get());
+    }
 
 }

src/test/java/com/rabbitmq/client/test/functional/ConnectionRecovery.java

@@ -16,7 +16,7 @@
 package com.rabbitmq.client.test.functional;
 
 import com.rabbitmq.client.*;
-
+import com.rabbitmq.client.impl.AbstractCredentialsProvider;
 import com.rabbitmq.client.impl.NetworkConnection;
 import com.rabbitmq.client.impl.recovery.*;
 import com.rabbitmq.client.test.BrokerTestCase;
@@ -34,6 +34,7 @@
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.concurrent.atomic.AtomicReference;
 
 import static org.junit.Assert.*;
@@ -121,6 +122,42 @@ public class ConnectionRecovery extends BrokerTestCase {
             c.abort();
         }
     }
+    
+    // See https://github.com/rabbitmq/rabbitmq-java-client/pull/350 . We want to request fresh creds when recovering.
+    @Test public void connectionRecoveryRequestsCredentialsAgain() throws Exception {
+        ConnectionFactory cf = buildConnectionFactoryWithRecoveryEnabled(false);
+        final String username = cf.getUsername();
+        final String password = cf.getPassword();
+        final AtomicLong usernameRequested = new AtomicLong(0);
+        final AtomicLong passwordRequested = new AtomicLong(0);
+        cf.setCredentialsProvider(new AbstractCredentialsProvider() {
+            
+            @Override
+            public String getUsername() {
+                usernameRequested.incrementAndGet();
+                return username;
+            }
+            
+            @Override
+            public String getPassword() {
+                passwordRequested.incrementAndGet();
+                return password;
+            }
+        });
+        RecoverableConnection c = (RecoverableConnection) cf.newConnection();
+        try {
+            assertTrue(c.isOpen());
+            assertEquals(1, usernameRequested.get());
+            assertEquals(1, passwordRequested.get());
+            
+            closeAndWaitForRecovery(c);
+            assertTrue(c.isOpen());
+            assertEquals(2, usernameRequested.get());
+            assertEquals(2, passwordRequested.get());
+        } finally {
+            c.abort();
+        }
+    }
 
     // see https://github.com/rabbitmq/rabbitmq-java-client/issues/135
     @Test public void thatShutdownHooksOnConnectionFireBeforeRecoveryStarts() throws IOException, InterruptedException {