package akka.dispatch

import org.scalatest.junit.JUnitSuite
import org.junit.Test
import akka.actor.{ Actor, ActorRef }
import Actor._
import org.multiverse.api.latches.StandardLatch
import java.util.concurrent.{ TimeUnit, CountDownLatch }

object FutureSpec {
  class TestActor extends Actor {
    def receive = {
      case "Hello" ⇒
        self.reply("World")
      case "NoReply" ⇒ {}
      case "Failure" ⇒
        throw new RuntimeException("Expected exception; to test fault-tolerance")
    }
  }

  class TestDelayActor(await: StandardLatch) extends Actor {
    def receive = {
      case "Hello" ⇒
        await.await
        self.reply("World")
      case "NoReply" ⇒ { await.await }
      case "Failure" ⇒
        await.await
        throw new RuntimeException("Expected exception; to test fault-tolerance")
    }
  }
}

class JavaFutureSpec extends JavaFutureTests with JUnitSuite

class FutureSpec extends JUnitSuite {
  import FutureSpec._

  @Test
  def shouldActorReplyResultThroughExplicitFuture {
    val actor = actorOf[TestActor]
    actor.start()
    val future = actor !!! "Hello"
    future.await
    assert(future.result.isDefined)
    assert("World" === future.result.get)
    actor.stop()
  }

  @Test
  def shouldActorReplyExceptionThroughExplicitFuture {
    val actor = actorOf[TestActor]
    actor.start()
    val future = actor !!! "Failure"
    future.await
    assert(future.exception.isDefined)
    assert("Expected exception; to test fault-tolerance" === future.exception.get.getMessage)
    actor.stop()
  }

  @Test
  def shouldFutureCompose {
    val actor1 = actorOf[TestActor].start()
    val actor2 = actorOf(new Actor { def receive = { case s: String ⇒ self reply s.toUpperCase } }).start()
    val future1 = actor1 !!! "Hello" flatMap ((s: String) ⇒ actor2 !!! s)
    val future2 = actor1 !!! "Hello" flatMap (actor2 !!! (_: String))
    val future3 = actor1 !!! "Hello" flatMap (actor2 !!! (_: Int))
    assert((future1.get: Any) === "WORLD")
    assert((future2.get: Any) === "WORLD")
    intercept[ClassCastException] { future3.get }
    actor1.stop()
    actor2.stop()
  }

  @Test
  def shouldFutureComposePatternMatch {
    val actor1 = actorOf[TestActor].start()
    val actor2 = actorOf(new Actor { def receive = { case s: String ⇒ self reply s.toUpperCase } }).start()
    val future1 = actor1 !!! "Hello" collect { case (s: String) ⇒ s } flatMap (actor2 !!! _)
    val future2 = actor1 !!! "Hello" collect { case (n: Int) ⇒ n } flatMap (actor2 !!! _)
    assert((future1.get: Any) === "WORLD")
    intercept[MatchError] { future2.get }
    actor1.stop()
    actor2.stop()
  }

  @Test
  def shouldFutureForComprehension {
    val actor = actorOf(new Actor {
      def receive = {
        case s: String ⇒ self reply s.length
        case i: Int    ⇒ self reply (i * 2).toString
      }
    }).start()

    val future0 = actor !!! "Hello"

    val future1 = for {
      a: Int ← future0 // returns 5
      b: String ← actor !!! a // returns "10"
      c: String ← actor !!! 7 // returns "14"
    } yield b + "-" + c

    val future2 = for {
      a: Int ← future0
      b: Int ← actor !!! a
      c: String ← actor !!! 7
    } yield b + "-" + c

    assert(future1.get === "10-14")
    intercept[ClassCastException] { future2.get }
    actor.stop()
  }

  @Test
  def shouldFutureForComprehensionPatternMatch {
    case class Req[T](req: T)
    case class Res[T](res: T)
    val actor = actorOf(new Actor {
      def receive = {
        case Req(s: String) ⇒ self reply Res(s.length)
        case Req(i: Int)    ⇒ self reply Res((i * 2).toString)
      }
    }).start()

    val future1 = for {
      Res(a: Int) ← actor !!! Req("Hello")
      Res(b: String) ← actor !!! Req(a)
      Res(c: String) ← actor !!! Req(7)
    } yield b + "-" + c

    val future2 = for {
      Res(a: Int) ← actor !!! Req("Hello")
      Res(b: Int) ← actor !!! Req(a)
      Res(c: Int) ← actor !!! Req(7)
    } yield b + "-" + c

    assert(future1.get === "10-14")
    intercept[MatchError] { future2.get }
    actor.stop()
  }

  @Test
  def shouldMapMatchedExceptionsToResult {
    val future1 = Future(5)
    val future2 = future1 map (_ / 0)
    val future3 = future2 map (_.toString)

    val future4 = future1 failure {
      case e: ArithmeticException ⇒ 0
    } map (_.toString)

    val future5 = future2 failure {
      case e: ArithmeticException ⇒ 0
    } map (_.toString)

    val future6 = future2 failure {
      case e: MatchError ⇒ 0
    } map (_.toString)

    val future7 = future3 failure { case e: ArithmeticException ⇒ "You got ERROR" }

    val actor = actorOf[TestActor].start()

    val future8 = actor !!! "Failure"
    val future9 = actor !!! "Failure" failure {
      case e: RuntimeException ⇒ "FAIL!"
    }
    val future10 = actor !!! "Hello" failure {
      case e: RuntimeException ⇒ "FAIL!"
    }
    val future11 = actor !!! "Failure" failure { case _ ⇒ "Oops!" }

    assert(future1.get === 5)
    intercept[ArithmeticException] { future2.get }
    intercept[ArithmeticException] { future3.get }
    assert(future4.get === "5")
    assert(future5.get === "0")
    intercept[ArithmeticException] { future6.get }
    assert(future7.get === "You got ERROR")
    intercept[RuntimeException] { future8.get }
    assert(future9.get === "FAIL!")
    assert(future10.get === "World")
    assert(future11.get === "Oops!")

    actor.stop()
  }

  @Test
  def shouldFoldResults {
    val actors = (1 to 10).toList map { _ ⇒
      actorOf(new Actor {
        def receive = { case (add: Int, wait: Int) ⇒ Thread.sleep(wait); self reply_? add }
      }).start()
    }
    val timeout = 10000
    def futures = actors.zipWithIndex map { case (actor: ActorRef, idx: Int) ⇒ actor.!!![Int]((idx, idx * 200), timeout) }
    assert(Futures.fold(0, timeout)(futures)(_ + _).await.result.get === 45)
  }

  @Test
  def shouldFoldResultsByComposing {
    val actors = (1 to 10).toList map { _ ⇒
      actorOf(new Actor {
        def receive = { case (add: Int, wait: Int) ⇒ Thread.sleep(wait); self reply_? add }
      }).start()
    }
    def futures = actors.zipWithIndex map { case (actor: ActorRef, idx: Int) ⇒ actor.!!![Int]((idx, idx * 200), 10000) }
    assert(futures.foldLeft(Future(0))((fr, fa) ⇒ for (r ← fr; a ← fa) yield (r + a)).get === 45)
  }

  @Test
  def shouldFoldResultsWithException {
    val actors = (1 to 10).toList map { _ ⇒
      actorOf(new Actor {
        def receive = {
          case (add: Int, wait: Int) ⇒
            Thread.sleep(wait)
            if (add == 6) throw new IllegalArgumentException("shouldFoldResultsWithException: expected")
            self reply_? add
        }
      }).start()
    }
    val timeout = 10000
    def futures = actors.zipWithIndex map { case (actor: ActorRef, idx: Int) ⇒ actor.!!![Int]((idx, idx * 100), timeout) }
    assert(Futures.fold(0, timeout)(futures)(_ + _).await.exception.get.getMessage === "shouldFoldResultsWithException: expected")
  }

  @Test
  def shouldFoldReturnZeroOnEmptyInput {
    assert(Futures.fold(0)(List[Future[Int]]())(_ + _).get === 0)
  }

  @Test
  def shouldReduceResults {
    val actors = (1 to 10).toList map { _ ⇒
      actorOf(new Actor {
        def receive = { case (add: Int, wait: Int) ⇒ Thread.sleep(wait); self reply_? add }
      }).start()
    }
    val timeout = 10000
    def futures = actors.zipWithIndex map { case (actor: ActorRef, idx: Int) ⇒ actor.!!![Int]((idx, idx * 200), timeout) }
    assert(Futures.reduce(futures, timeout)(_ + _).get === 45)
  }

  @Test
  def shouldReduceResultsWithException {
    val actors = (1 to 10).toList map { _ ⇒
      actorOf(new Actor {
        def receive = {
          case (add: Int, wait: Int) ⇒
            Thread.sleep(wait)
            if (add == 6) throw new IllegalArgumentException("shouldFoldResultsWithException: expected")
            self reply_? add
        }
      }).start()
    }
    val timeout = 10000
    def futures = actors.zipWithIndex map { case (actor: ActorRef, idx: Int) ⇒ actor.!!![Int]((idx, idx * 100), timeout) }
    assert(Futures.reduce(futures, timeout)(_ + _).await.exception.get.getMessage === "shouldFoldResultsWithException: expected")
  }

  @Test(expected = classOf[UnsupportedOperationException])
  def shouldReduceThrowIAEOnEmptyInput {
    Futures.reduce(List[Future[Int]]())(_ + _).await.resultOrException
  }

  @Test
  def receiveShouldExecuteOnComplete {
    val latch = new StandardLatch
    val actor = actorOf[TestActor].start()
    actor !!! "Hello" receive { case "World" ⇒ latch.open }
    assert(latch.tryAwait(5, TimeUnit.SECONDS))
    actor.stop()
  }

  @Test
  def shouldTraverseFutures {
    val oddActor = actorOf(new Actor {
      var counter = 1
      def receive = {
        case 'GetNext ⇒
          self reply counter
          counter += 2
      }
    }).start()

    val oddFutures: List[Future[Int]] = List.fill(100)(oddActor !!! 'GetNext)
    assert(Future.sequence(oddFutures).get.sum === 10000)
    oddActor.stop()

    val list = (1 to 100).toList
    assert(Future.traverse(list)(x ⇒ Future(x * 2 - 1)).get.sum === 10000)
  }

  @Test
  def shouldHandleThrowables {
    class ThrowableTest(m: String) extends Throwable(m)

    val f1 = Future { throw new ThrowableTest("test") }
    f1.await
    intercept[ThrowableTest] { f1.resultOrException }

    val latch = new StandardLatch
    val f2 = Future { latch.tryAwait(5, TimeUnit.SECONDS); "success" }
    f2 foreach (_ ⇒ throw new ThrowableTest("dispatcher foreach"))
    f2 receive { case _ ⇒ throw new ThrowableTest("dispatcher receive") }
    val f3 = f2 map (s ⇒ s.toUpperCase)
    latch.open
    f2.await
    assert(f2.resultOrException === Some("success"))
    f2 foreach (_ ⇒ throw new ThrowableTest("current thread foreach"))
    f2 receive { case _ ⇒ throw new ThrowableTest("current thread receive") }
    f3.await
    assert(f3.resultOrException === Some("SUCCESS"))

    // make sure all futures are completed in dispatcher
    assert(Dispatchers.defaultGlobalDispatcher.pendingFutures === 0)
  }

  @Test
  def shouldBlockUntilResult {
    val latch = new StandardLatch

    val f = Future({ latch.await; 5 })
    val f2 = Future({ f.get + 5 })

    assert(f2.resultOrException === None)
    latch.open
    assert(f2.get === 10)

    val f3 = Future({ Thread.sleep(100); 5 }, 10)
    intercept[FutureTimeoutException] {
      f3.get
    }
  }

  @Test
  def futureComposingWithContinuations {
    import Future.flow

    val actor = actorOf[TestActor].start

    val x = Future("Hello")
    val y = x flatMap (actor !!! _)

    val r = flow(x() + " " + y[String]() + "!")

    assert(r.get === "Hello World!")

    actor.stop
  }

  @Test
  def futureComposingWithContinuationsFailureDivideZero {
    import Future.flow

    val x = Future("Hello")
    val y = x map (_.length)

    val r = flow(x() + " " + y.map(_ / 0).map(_.toString)(), 100)

    intercept[java.lang.ArithmeticException](r.get)
  }

  @Test
  def futureComposingWithContinuationsFailureCastInt {
    import Future.flow

    val actor = actorOf[TestActor].start

    val x = Future(3)
    val y = actor !!! "Hello"

    val r = flow(x() + y[Int](), 100)

    intercept[ClassCastException](r.get)
  }

  @Test
  def futureComposingWithContinuationsFailureCastNothing {
    import Future.flow

    val actor = actorOf[TestActor].start

    val x = Future("Hello")
    val y = actor !!! "Hello"

    val r = flow(x() + y())

    intercept[ClassCastException](r.get)
  }

  @Test
  def futureCompletingWithContinuations {
    import Future.flow

    val x, y, z = Promise[Int]()
    val ly, lz = new StandardLatch

    val result = flow {
      y completeWith x
      ly.open // not within continuation

      z << x
      lz.open // within continuation, will wait for 'z' to complete
      z() + y()
    }

    assert(ly.tryAwaitUninterruptible(100, TimeUnit.MILLISECONDS))
    assert(!lz.tryAwaitUninterruptible(100, TimeUnit.MILLISECONDS))

    flow { x << 5 }

    assert(y.get === 5)
    assert(z.get === 5)
    assert(lz.isOpen)
    assert(result.get === 10)

    val a, b, c = Promise[Int]()

    val result2 = flow {
      val n = (a << c).result.get + 10
      b << (c() - 2)
      a() + n * b()
    }

    c completeWith Future(5)

    assert(a.get === 5)
    assert(b.get === 3)
    assert(result2.get === 50)
    Thread.sleep(100)

    // make sure all futures are completed in dispatcher
    assert(Dispatchers.defaultGlobalDispatcher.pendingFutures === 0)
  }

  @Test
  def shouldNotAddOrRunCallbacksAfterFailureToBeCompletedBeforeExpiry {
    val latch = new StandardLatch
    val f = Promise[Int](0)
    Thread.sleep(25)
    f.onComplete(_ ⇒ latch.open) //Shouldn't throw any exception here

    assert(f.isExpired) //Should be expired

    f.complete(Right(1)) //Shouldn't complete the Future since it is expired

    assert(f.value.isEmpty) //Shouldn't be completed
    assert(!latch.isOpen) //Shouldn't run the listener
  }

  @Test
  def futureDataFlowShouldEmulateBlocking1 {
    import Future.flow

    val one, two = Promise[Int](1000 * 60)
    val simpleResult = flow {
      one() + two()
    }

    assert(List(one, two, simpleResult).forall(_.isCompleted == false))

    flow { one << 1 }

    assert(one.isCompleted)
    assert(List(two, simpleResult).forall(_.isCompleted == false))

    flow { two << 9 }

    assert(List(one, two).forall(_.isCompleted == true))
    assert(simpleResult.get === 10)

  }

  @Test
  def futureDataFlowShouldEmulateBlocking2 {
    import Future.flow
    val x1, x2, y1, y2 = Promise[Int](1000 * 60)
    val lx, ly, lz = new StandardLatch
    val result = flow {
      lx.open()
      x1 << y1
      ly.open()
      x2 << y2
      lz.open()
      x1() + x2()
    }
    assert(lx.isOpen)
    assert(!ly.isOpen)
    assert(!lz.isOpen)
    assert(List(x1, x2, y1, y2).forall(_.isCompleted == false))

    flow { y1 << 1 } // When this is set, it should cascade down the line

    assert(ly.tryAwaitUninterruptible(2000, TimeUnit.MILLISECONDS))
    assert(x1.get === 1)
    assert(!lz.isOpen)

    flow { y2 << 9 } // When this is set, it should cascade down the line

    assert(lz.tryAwaitUninterruptible(2000, TimeUnit.MILLISECONDS))
    assert(x2.get === 9)

    assert(List(x1, x2, y1, y2).forall(_.isCompleted == true))

    assert(result.get === 10)
  }

  @Test
  def dataFlowAPIshouldbeSlick {
    import Future.flow

    val i1, i2, s1, s2 = new StandardLatch

    val callService1 = Future { i1.open; s1.awaitUninterruptible; 1 }
    val callService2 = Future { i2.open; s2.awaitUninterruptible; 9 }

    val result = flow { callService1() + callService2() }

    assert(!s1.isOpen)
    assert(!s2.isOpen)
    assert(!result.isCompleted)
    assert(i1.tryAwaitUninterruptible(2000, TimeUnit.MILLISECONDS))
    assert(i2.tryAwaitUninterruptible(2000, TimeUnit.MILLISECONDS))
    s1.open
    s2.open
    assert(result.get === 10)
  }

  @Test
  def futureCompletingWithContinuationsFailure {
    import Future.flow

    val x, y, z = Promise[Int]()
    val ly, lz = new StandardLatch

    val result = flow {
      y << x
      ly.open
      val oops = 1 / 0
      z << x
      lz.open
      z() + y() + oops
    }

    assert(!ly.tryAwaitUninterruptible(100, TimeUnit.MILLISECONDS))
    assert(!lz.tryAwaitUninterruptible(100, TimeUnit.MILLISECONDS))

    flow { x << 5 }

    assert(y.get === 5)
    intercept[java.lang.ArithmeticException](result.get)
    assert(z.value === None)
    assert(!lz.isOpen)
  }

  @Test
  def futureContinuationsShouldNotBlock {
    import Future.flow

    val latch = new StandardLatch
    val future = Future {
      latch.await
      "Hello"
    }

    val result = flow {
      Some(future()).filter(_ == "Hello")
    }

    assert(!result.isCompleted)

    latch.open

    assert(result.get === Some("Hello"))
  }

  @Test
  def futureFlowShouldBeTypeSafe {
    import Future.flow

    def checkType[A: Manifest, B](in: Future[A], refmanifest: Manifest[B]): Boolean = manifest[A] == refmanifest

    val rString = flow {
      val x = Future(5)
      x().toString
    }

    val rInt = flow {
      val x = rString.apply
      val y = Future(5)
      x.length + y()
    }

    assert(checkType(rString, manifest[String]))
    assert(checkType(rInt, manifest[Int]))
    assert(!checkType(rInt, manifest[String]))
    assert(!checkType(rInt, manifest[Nothing]))
    assert(!checkType(rInt, manifest[Any]))

    rString.await
    rInt.await
  }

  @Test
  def futureFlowSimpleAssign {
    import Future.flow

    val x, y, z = Promise[Int]()

    flow {
      z << x() + y()
    }
    flow { x << 40 }
    flow { y << 2 }

    assert(z.get === 42)
  }

  @Test
  def ticket812FutureDispatchCleanup {
    val dispatcher = implicitly[MessageDispatcher]
    assert(dispatcher.pendingFutures === 0)
    val future = Future({ Thread.sleep(100); "Done" }, 10)
    intercept[FutureTimeoutException] { future.await }
    assert(dispatcher.pendingFutures === 1)
    Thread.sleep(100)
    assert(dispatcher.pendingFutures === 0)
  }
}