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

Go to download
package com.dimafeng.testcontainers

import org.testcontainers.containers.{ToxiproxyContainer => JavaToxiproxyContainer}
import org.testcontainers.utility.DockerImageName

case class ToxiproxyContainer(
  dockerImageName: DockerImageName = DockerImageName.parse(ToxiproxyContainer.defaultDockerImageName)
) extends SingleContainer[JavaToxiproxyContainer] {

  import ToxiproxyContainer._

  override val container: JavaToxiproxyContainer = new JavaToxiproxyContainer(dockerImageName)

  def proxy(hostname: String, port: Int): ContainerProxy = container.getProxy(hostname, port)

  def proxy(container: SingleContainer[_], port: Int): ContainerProxy = proxy(container.networkAliases.head, port)
}

object ToxiproxyContainer {

  val defaultDockerImageName = "shopify/toxiproxy:2.1.4"

  type ContainerProxy = JavaToxiproxyContainer.ContainerProxy

  case class Def(
    dockerImageName: DockerImageName = DockerImageName.parse(ToxiproxyContainer.defaultDockerImageName)
  ) extends ContainerDef {

    override type Container = ToxiproxyContainer

    override def createContainer(): ToxiproxyContainer = {
      new ToxiproxyContainer(
        dockerImageName
      )
    }
  }

}