Issue debasishg#282 added unit test demonstrating that Pools support …

…passwords - with some refactoring.
noahlz · Nov 3, 2021 · e8040c8 · e8040c8
1 parent 59645f1
commit e8040c8
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 3 deletions.
diff --git a/src/test/scala/com/redis/PoolSpec.scala b/src/test/scala/com/redis/PoolSpec.scala
@@ -8,12 +8,13 @@ import org.scalatest.matchers.should.Matchers
 import scala.concurrent._
 import scala.concurrent.duration._
 
-class PoolSpec extends AnyFunSpec
+abstract class BasePoolSpec extends AnyFunSpec
                with Matchers
                with BeforeAndAfterEach
                with RedisDocker {
 
-  implicit lazy val clients: RedisClientPool = new RedisClientPool(redisContainerHost, redisContainerPort)
+  implicit lazy val clients: RedisClientPool =
+    new RedisClientPool(redisContainerHost, redisContainerPort, secret = redisPassword)
 
   override protected def beforeEach(): Unit = {
     super.beforeEach()
@@ -32,6 +33,10 @@ class PoolSpec extends AnyFunSpec
     clients.close()
     super.afterAll()
   }
+}
+
+trait PoolLoadTesting {
+  self: BasePoolSpec =>
 
   def lp(msgs: List[String]) = {
     clients.withClient {
@@ -63,6 +68,9 @@ class PoolSpec extends AnyFunSpec
       }
     }
   }
+}
+
+class PoolSpec extends BasePoolSpec with PoolLoadTesting {
 
   describe("pool test") {
     it("should distribute work amongst the clients") {
@@ -127,3 +135,18 @@ class PoolSpec extends AnyFunSpec
     }
   }
 }
+
+class SecurePoolSpec extends BasePoolSpec {
+
+  override val redisPassword = Option("mayonaise")
+
+  describe("redis pools with authentication") {
+    it("should be able to connect as normal") {
+      val response = clients.withClient { client =>
+        client.ping
+       }
+      response should equal(Some("PONG"))
+    }
+  }
+}
+
diff --git a/src/test/scala/com/redis/common/RedisDocker.scala b/src/test/scala/com/redis/common/RedisDocker.scala
@@ -54,8 +54,12 @@ trait RedisContainer extends DockerKit with DockerTestKit with DockerKitDockerJa
 
   protected val redisContainerHost: String = "localhost"
   protected val redisPort: Int = 6379
+  protected val redisPassword: Option[String] = None
 
-  protected def baseContainer(name: Option[String]) = DockerContainer("redis:latest", name=name)
+  // just password supported for now
+  private def args = redisPassword.map(password => Seq("--requirepass", password))
+
+  protected def baseContainer(name: Option[String]) = DockerContainer("redis:latest", name=name, command=args)
 
   protected def createContainer(name: Option[String] = Some(RandomStringUtils.randomAlphabetic(10)),
                                 ports: Map[Int, Int] = Map.empty): DockerContainer = {