dev.chopsticks.testkit.ManualTimeAkkaTestKit.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of chopsticks-testkit_2.13 Show documentation

Essential Scala libraries for everyday use

There is a newer version: 3.12.0

Show newest version

package dev.chopsticks.testkit

import java.util.concurrent.atomic.AtomicLong

import akka.actor.testkit.typed.scaladsl.ManualTime
import com.typesafe.config.{Config, ConfigFactory}
import akka.actor.typed
import akka.actor.typed.scaladsl.adapter._
import zio.Runtime
import zio.internal.Platform
import zio.test.environment.TestClock

import scala.concurrent.duration.FiniteDuration

object ManualTimeAkkaTestKit {
  final class ManualClock(mockClock: Option[TestClock.Service] = None)(implicit typedSystem: typed.ActorSystem[_]) {
    private val controller = ManualTime()
    private val totalNanoPassed = new AtomicLong()
    private val rt = Runtime[Any]((), Platform.fromExecutionContext(typedSystem.executionContext))

    def timePasses(amount: FiniteDuration): Unit = {
      val _ = totalNanoPassed.getAndAdd(amount.toNanos)
      mockClock.foreach { c =>
        val d = zio.duration.Duration.fromScala(amount)
        rt.unsafeRun(c.adjust(d))
      }
      controller.timePasses(amount)
    }

    def nanoTime(): Long = totalNanoPassed.get()
  }
}

trait ManualTimeAkkaTestKit extends AkkaTestKit {
  override lazy val typesafeConfig: Config = {
    val cfg = ManualTime.config.withFallback(ConfigFactory.load())
    assert(
      cfg.getBoolean("akka.stream.materializer.debug.fuzzing-mode"),
      "akka.stream.materializer.debug.fuzzing-mode is not 'on' for testing, config loading is not working properly?"
    )
    cfg
  }
  implicit lazy val typedSystem: typed.ActorSystem[Nothing] = system.toTyped
}