scalaprops.TestExecutorImpl.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalaprops_2.11 Show documentation

property based testing library for Scala

There is a newer version: 0.9.1

package scalaprops

import scala.concurrent.{Await, ExecutionContext, Future}
import scala.concurrent.duration.Duration
import java.lang.Thread.UncaughtExceptionHandler
import sbt.testing.Logger
import java.util.concurrent.ForkJoinPool

object TestExecutorImpl {
  private[this] def newInstance(log: Logger): TestExecutor = new TestExecutor {
    private[this] val executionContext = {
      lazy val executorService: ForkJoinPool = new ForkJoinPool(
        sys.runtime.availableProcessors(),
        ForkJoinPool.defaultForkJoinWorkerThreadFactory,
        new UncaughtExceptionHandler {
          def uncaughtException(t: Thread, e: Throwable): Unit = {
            log.error("uncaughtException Thread = " + t)
            log.trace(e)
            e.printStackTrace()
            executorService.shutdown()
          }
        },
        false
      )
      ExecutionContext.fromExecutorService(executorService)
    }

    override def execute[A](timeout: Duration)(f: => A): A =
      Await.result(Future(f)(executionContext), timeout)

    override def shutdown(): Unit =
      executionContext.shutdown()
  }
  def withExecutor[A](logger: Logger)(f: TestExecutor => A): A = {
    val executor = newInstance(logger)
    try f(executor) finally executor.shutdown()
  }
}