WaitForLogLine.scala

package utils

import java.util.function.Predicate

import org.testcontainers.containers.GenericContainer.AbstractWaitStrategy
import org.testcontainers.containers.output.OutputFrame.OutputType.STDOUT
import org.testcontainers.containers.output.{OutputFrame, WaitingConsumer}


class WaitForLogLine(val msg: String) extends AbstractWaitStrategy {
  val consumer = new WaitingConsumer()

  override def waitUntilReady(): Unit = {
    container.followOutput(consumer, STDOUT)
    consumer.waitUntil(new Predicate[OutputFrame]() {
      override def test(frame: OutputFrame): Boolean = {
        frame.getUtf8String().contains(msg)
      }
    })
  }
}

object WaitForLogLine {
  def apply(msg: String): WaitForLogLine = new WaitForLogLine(msg)
}