com.dimafeng.testcontainers.Db2Container.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of testcontainers-scala-db2_2.12 Show documentation

testcontainers-scala-db2

There is a newer version: 0.41.8

Show newest version

package com.dimafeng.testcontainers

import org.testcontainers.containers.{Db2Container => JavaDb2Container}
import org.testcontainers.utility.DockerImageName

case class Db2Container(
  dockerImageName: DockerImageName = DockerImageName.parse(Db2Container.defaultDockerImageName),
  dbName: String = Db2Container.defaultDatabaseName,
  dbUsername: String = Db2Container.defaultUsername,
  dbPassword: String = Db2Container.defaultPassword,
  urlParams: Map[String, String] = Map.empty,
  commonJdbcParams: JdbcDatabaseContainer.CommonParams = JdbcDatabaseContainer.CommonParams()
) extends SingleContainer[JavaDb2Container] with JdbcDatabaseContainer {

  override val container: JavaDb2Container = {
    val c = new JavaDb2Container(dockerImageName)

    c.withDatabaseName(dbName)
    c.withUsername(username)
    c.withPassword(password)
    c.acceptLicense()

    urlParams.foreach { case (key, value) =>
      c.withUrlParam(key, value)
    }

    commonJdbcParams.applyTo(c)

    c
  }
}

object Db2Container {

  val defaultDockerImageName = s"${JavaDb2Container.DEFAULT_DB2_IMAGE_NAME}:${JavaDb2Container.DEFAULT_TAG}"
  val defaultDatabaseName = "test"
  val defaultUsername = "db2inst1"
  val defaultPassword = "foobar1234"

  case class Def(
    dockerImageName: DockerImageName = DockerImageName.parse(Db2Container.defaultDockerImageName),
    dbName: String = Db2Container.defaultDatabaseName,
    dbUsername: String = Db2Container.defaultUsername,
    dbPassword: String = Db2Container.defaultPassword,
    urlParams: Map[String, String] = Map.empty,
    commonJdbcParams: JdbcDatabaseContainer.CommonParams = JdbcDatabaseContainer.CommonParams()
  ) extends ContainerDef {

    override type Container = Db2Container

    override def createContainer(): Db2Container = {
      new Db2Container(
        dockerImageName,
        dbName,
        dbUsername,
        dbPassword,
        urlParams,
        commonJdbcParams
      )
    }
  }
}