Tests Redis auth against embedded Redis instance

Author: davidwadden

build.gradle

@@ -87,6 +87,7 @@ project(':dyno-jedis') {
         compile project(':dyno-core')
         compile project(':dyno-contrib')
         compile 'redis.clients:jedis:2.9.0'
+        testCompile 'com.netflix.spinnaker.embedded-redis:embedded-redis:0.8.0'
     }
 }
 

dyno-jedis/src/test/java/com/netflix/dyno/jedis/RedisAuthenticationIntegrationTest.java

@@ -0,0 +1,104 @@
+package com.netflix.dyno.jedis;
+
+import com.netflix.dyno.connectionpool.Host;
+import com.netflix.dyno.connectionpool.HostSupplier;
+import com.netflix.dyno.connectionpool.TokenMapSupplier;
+import com.netflix.dyno.connectionpool.impl.ConnectionPoolConfigurationImpl;
+import com.netflix.dyno.connectionpool.impl.lb.HostToken;
+import java.util.Collections;
+import java.util.List;
+import java.util.Set;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Test;
+import redis.embedded.RedisServer;
+import redis.embedded.RedisServerBuilder;
+
+public class RedisAuthenticationIntegrationTest {
+
+	private static final int REDIS_PORT = 8998;
+	private static final String REDIS_RACK = "rack-1c";
+	private static final String REDIS_DATACENTER = "rack-1";
+
+	private RedisServer redisServer;
+
+	@After
+	public void tearDown() throws Exception {
+		redisServer.stop();
+	}
+
+	@Test
+	public void testNoAuthClient() throws Exception {
+		redisServer = new RedisServer(REDIS_PORT);
+		redisServer.start();
+
+		Host noAuthHost = new Host("localhost", REDIS_PORT, REDIS_RACK, Host.Status.Up);
+		TokenMapSupplierImpl tokenMapSupplier = new TokenMapSupplierImpl(noAuthHost);
+		DynoJedisClient dynoClient = constructJedisClient(tokenMapSupplier,
+				() -> Collections.singletonList(noAuthHost));
+
+		String statusCodeReply = dynoClient.set("some-key", "some-value");
+		Assert.assertEquals("OK", statusCodeReply);
+
+		String value = dynoClient.get("some-key");
+		Assert.assertEquals("some-value", value);
+	}
+
+	@Test
+	public void testAuthClient() throws Exception {
+		redisServer = new RedisServerBuilder()
+				.port(REDIS_PORT)
+				.setting("requirepass some-password")
+				.build();
+		redisServer.start();
+
+		Host authHost = new Host("localhost", null, REDIS_PORT, REDIS_PORT, REDIS_RACK,
+				REDIS_DATACENTER, Host.Status.Up, null, "some-password");
+		TokenMapSupplierImpl tokenMapSupplier = new TokenMapSupplierImpl(authHost);
+		DynoJedisClient dynoClient = constructJedisClient(tokenMapSupplier,
+				() -> Collections.singletonList(authHost));
+
+		String statusCodeReply = dynoClient.set("some-key", "some-value");
+		Assert.assertEquals("OK", statusCodeReply);
+
+		String value = dynoClient.get("some-key");
+		Assert.assertEquals("some-value", value);
+	}
+
+	private DynoJedisClient constructJedisClient(TokenMapSupplier tokenMapSupplier,
+			HostSupplier hostSupplier) {
+
+		final ConnectionPoolConfigurationImpl connectionPoolConfiguration =
+				new ConnectionPoolConfigurationImpl(REDIS_RACK);
+		connectionPoolConfiguration.withTokenSupplier(tokenMapSupplier);
+		connectionPoolConfiguration.setLocalRack(REDIS_RACK);
+		connectionPoolConfiguration.setLocalDataCenter(REDIS_DATACENTER);
+
+		return new DynoJedisClient.Builder()
+				.withApplicationName("appname")
+				.withDynomiteClusterName(REDIS_RACK)
+				.withHostSupplier(hostSupplier)
+				.withCPConfig(connectionPoolConfiguration)
+				.build();
+	}
+
+	private static class TokenMapSupplierImpl implements TokenMapSupplier {
+
+		private final HostToken localHostToken;
+
+		private TokenMapSupplierImpl(Host host) {
+			this.localHostToken = new HostToken(100000L, host);
+		}
+
+		@Override
+		public List<HostToken> getTokens(Set<Host> activeHosts) {
+			return Collections.singletonList(localHostToken);
+		}
+
+		@Override
+		public HostToken getTokenForHost(Host host, Set<Host> activeHosts) {
+			return localHostToken;
+		}
+
+	}
+}