ZooKeeperControllerSpec.scala

package com.github.j5ik2o.dockerController

import com.github.j5ik2o.dockerController.WaitPredicates.WaitPredicate
import com.github.j5ik2o.dockerController.zooKeeper.ZooKeeperController
import org.apache.zookeeper.{ WatchedEvent, Watcher, ZooKeeper }
import org.scalatest.freespec.AnyFreeSpec
import org.apache.zookeeper.Watcher.Event.KeeperState

import java.util.concurrent.{ CountDownLatch, TimeUnit }
import scala.concurrent.duration._

class ZooKeeperControllerSpec extends AnyFreeSpec with DockerControllerSpecSupport {
  val testTimeFactor: Int = sys.env.getOrElse("TEST_TIME_FACTOR", "1").toInt
  logger.debug(s"testTimeFactor = $testTimeFactor")

  lazy val hostPort: Int                            = temporaryServerPort()
  lazy val zooKeeperController: ZooKeeperController = ZooKeeperController(dockerClient)(1, hostPort)

  override protected val dockerControllers: Vector[DockerController] = Vector(zooKeeperController)

  val waitPredicate: WaitPredicate =
    WaitPredicates.forLogMessageByRegex(ZooKeeperController.RegexForWaitPredicate, Some((1 * testTimeFactor).seconds))

  val waitPredicateSetting: WaitPredicateSetting = WaitPredicateSetting(Duration.Inf, waitPredicate)

  override protected val waitPredicatesSettings: Map[DockerController, WaitPredicateSetting] =
    Map(zooKeeperController -> waitPredicateSetting)

  "ZooKeeperControllerSpec" - {
    "run" in {
      var zk: ZooKeeper = null
      try {
        val connectionLatch = new CountDownLatch(1)
        zk = new ZooKeeper(
          s"$dockerHost:$hostPort",
          3000,
          new Watcher {
            override def process(event: WatchedEvent): Unit = {
              logger.debug(s"event = $event")
              if (event.getState == KeeperState.SyncConnected)
                connectionLatch.countDown()
            }
          }
        )
        connectionLatch.await(10, TimeUnit.SECONDS)
      } finally
        if (zk != null)
          zk.close()
    }
  }

}