Add methods on ContainerState (#441)

Author: sideeffffect

src/main/scala/zio/testcontainers/ZIOTestcontainers.scala

@@ -1,7 +1,7 @@
 package zio.testcontainers
 
 import com.dimafeng.testcontainers.{DockerComposeContainer, SingleContainer}
-import org.testcontainers.containers.{ComposeContainer, GenericContainer => JavaGenericContainer}
+import org.testcontainers.containers.{ComposeContainer, ContainerState, GenericContainer => JavaGenericContainer}
 import org.testcontainers.lifecycle.Startable
 import zio.{Scope, Tag, UIO, ULayer, URIO, ZIO, ZLayer}
 
@@ -14,6 +14,13 @@ object ZIOTestcontainers {
     port <- ZIO.succeed(container.getServicePort(serviceName, servicePort))
   } yield (host, port)
 
+  def getHostAndPort(
+      container: ContainerState,
+  )(port: Int): UIO[(String, Int)] = for {
+    host <- ZIO.succeed(container.getHost)
+    port <- ZIO.succeed(container.getMappedPort(port))
+  } yield (host, port)
+
   def getHostAndPort(
       container: DockerComposeContainer,
   )(serviceName: String)(servicePort: Int): UIO[(String, Int)] = for {

src/main/scala/zio/testcontainers/package.scala

@@ -1,7 +1,7 @@
 package zio
 
 import com.dimafeng.testcontainers.{DockerComposeContainer, SingleContainer}
-import org.testcontainers.containers.{ComposeContainer, GenericContainer => JavaGenericContainer}
+import org.testcontainers.containers.{ComposeContainer, ContainerState, GenericContainer => JavaGenericContainer}
 import org.testcontainers.lifecycle.Startable
 
 package object testcontainers {
@@ -13,6 +13,13 @@ package object testcontainers {
 
   }
 
+  implicit final class ContainerStateOps(private val self: ContainerState) extends AnyVal {
+
+    def getHostAndPort(port: Int): UIO[(String, Int)] =
+      ZIOTestcontainers.getHostAndPort(self)(port)
+
+  }
+
   implicit final class DockerComposeContainerOps(private val container: DockerComposeContainer) extends AnyVal {
 
     def getHostAndPort(serviceName: String)(servicePort: Int): UIO[(String, Int)] =