diff options
author | phaller <hallerp@gmail.com> | 2012-11-13 01:07:28 +0100 |
---|---|---|
committer | phaller <hallerp@gmail.com> | 2012-11-16 09:42:30 +0100 |
commit | c4ceea0ca8538297622634121b99e2357ca72acb (patch) | |
tree | 31c2265496175dde52244f2506c94997fd9194dd /src/test | |
parent | f451904320d02c7dbe6b298f6ff790ca5cf5f080 (diff) | |
download | scala-async-c4ceea0ca8538297622634121b99e2357ca72acb.tar.gz scala-async-c4ceea0ca8538297622634121b99e2357ca72acb.tar.bz2 scala-async-c4ceea0ca8538297622634121b99e2357ca72acb.zip |
Add selective ANF transform
- Does not descend into class and module defs
- Adds several tests, including tests for if-else
Diffstat (limited to 'src/test')
-rw-r--r-- | src/test/scala/scala/async/run/anf/AnfTransformSpec.scala | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/src/test/scala/scala/async/run/anf/AnfTransformSpec.scala b/src/test/scala/scala/async/run/anf/AnfTransformSpec.scala new file mode 100644 index 0000000..f38efa9 --- /dev/null +++ b/src/test/scala/scala/async/run/anf/AnfTransformSpec.scala @@ -0,0 +1,94 @@ +/** + * Copyright (C) 2012 Typesafe Inc. <http://www.typesafe.com> + */ + +package scala.async +package run +package anf + +import language.{reflectiveCalls, postfixOps} +import scala.concurrent.{Future, ExecutionContext, future, Await} +import scala.concurrent.duration._ +import scala.async.Async.{async, await} +import org.junit.Test +import org.junit.runner.RunWith +import org.junit.runners.JUnit4 + + +class AnfTestClass { + + import ExecutionContext.Implicits.global + + def base(x: Int): Future[Int] = future { + x + 2 + } + + def m(y: Int): Future[Int] = async { + val f = base(y) + await(f) + } + + def m2(y: Int): Future[Int] = async { + val f = base(y) + val f2 = base(y + 1) + await(f) + await(f2) + } + + def m3(y: Int): Future[Int] = async { + val f = base(y) + var z = 0 + if (y > 0) { + z = await(f) + 2 + } else { + z = await(f) - 2 + } + z + } + + def m4(y: Int): Future[Int] = async { + val f = base(y) + val z = if (y > 0) { + await(f) + 2 + } else { + await(f) - 2 + } + z + 1 + } +} + + +@RunWith(classOf[JUnit4]) +class AnfTransformSpec { + + @Test + def `simple ANF transform`() { + val o = new AnfTestClass + val fut = o.m(10) + val res = Await.result(fut, 2 seconds) + res mustBe (12) + } + + @Test + def `simple ANF transform 2`() { + val o = new AnfTestClass + val fut = o.m2(10) + val res = Await.result(fut, 2 seconds) + res mustBe (25) + } + + @Test + def `simple ANF transform 3`() { + val o = new AnfTestClass + val fut = o.m3(10) + val res = Await.result(fut, 2 seconds) + res mustBe (14) + } + + @Test + def `ANF transform of assigning the result of an if-else`() { + val o = new AnfTestClass + val fut = o.m4(10) + val res = Await.result(fut, 2 seconds) + res mustBe (15) + } +} |