ScalaTest Helpers for testing Futures. Scala, Currying, Pattern Matching and Functional Programming is awesome.

Helpers.scala

def whenReady[A](result:Future[A], timeout:Duration = 1 second)(expectation: A => Unit) = {
  expectation(Await.result(result, timeout))
}

def tryWhenReady[A](result:Future[Try[A]], timeout:Duration = 1 second)
                   (failure:Throwable => Unit)
                   (expectation: A => Unit) = {
  Await.result(result, timeout) match {
    case Failure(e) => failure(e)
    case Success(result:A) => expectation(result)
  }
}

"deleting a document" should {
  "be successful" in {
    tryWhenReady(User.create(scott))(fail) { userDoc =>
      tryWhenReady(User.delete(userDoc))(fail) { _ must be ('ok) }
    }
    
    whenReady(User.create(scott)) {
      case Failure(e) => fail(e)
      case Success(user) =>
        whenReady(User.delete(user)) {
          case Failure(e) => fail(e)
          case Success(result) => result must be ('ok)
        }
    }
  }
}