FutureTimeoutTest.scala

import org.scalatest._

import scala.concurrent.{Future, TimeoutException}
import scala.concurrent.duration._

class TestFutureUtil extends AsyncFlatSpec with Matchers with OptionValues with Inside with Inspectors {

  implicit override def executionContext = scala.concurrent.ExecutionContext.Implicits.global

  "futureWithTimeout" should "complete the users future when it returns before the timeout" in {

    def myFuture = Future[Int] {
      //println(s"starting user future ${System.currentTimeMillis()}")
      Thread.sleep((2 seconds).toMillis)
      //println(s"user future done ${System.currentTimeMillis()}")
      100
    }

    FutureUtil.futureWithTimeout(myFuture, 3 seconds).map {
      result => if(result == 100) succeed else fail
    }
  }

  it should "not complete the future when it returns after the timeout" in {

    lazy val myFuture = Future[Int] {
      //println(s"user future waiting 4 seconds ${System.currentTimeMillis()}")
      Thread.sleep((4 seconds).toMillis)
      //println(s"user future done at ${System.currentTimeMillis()}")
      100
    }

    recoverToSucceededIf[TimeoutException] {
      FutureUtil.futureWithTimeout(myFuture, 2 seconds)
    }
  }


}