Support await in a while loop.

1 files changed, 43 insertions, 0 deletions

diff --git a/src/test/scala/scala/async/run/ifelse0/WhileSpec.scala b/src/test/scala/scala/async/run/ifelse0/WhileSpec.scala
new file mode 100644
index 0000000..d08e2c5
--- /dev/null
+++ b/src/test/scala/scala/async/run/ifelse0/WhileSpec.scala
@@ -0,0 +1,43 @@
+package scala.async
+package run
+package ifelse0
+
+import org.junit.runner.RunWith
+import org.junit.runners.JUnit4
+import org.junit.Test
+
+@RunWith(classOf[JUnit4])
+class WhileSpec {
+
+  @Test
+  def whiling1() {
+    import AsyncId._
+
+    val result = async {
+      var xxx: Int = 0
+      var y = 0
+      while (xxx < 3) {
+        y = await(xxx)
+        xxx = xxx + 1
+      }
+      y
+    }
+    result mustBe (2)
+  }
+
+  @Test
+  def whiling2() {
+    import AsyncId._
+
+    val result = async {
+      var xxx: Int = 0
+      var y = 0
+      while (false) {
+        y = await(xxx)
+        xxx = xxx + 1
+      }
+      y
+    }
+    result mustBe (0)
+  }
+}
\ No newline at end of file