Fix await in if condition / match scrutinee.

The type-checking performed in ANF transform is
precarious, and needed to use the original condition/
scrutinee in a throwaway tree to get things to work.

1 files changed, 8 insertions, 0 deletions

diff --git a/src/test/scala/scala/async/run/ifelse0/IfElse0.scala b/src/test/scala/scala/async/run/ifelse0/IfElse0.scala
index 0a72f1e..e2b1ca6 100644
--- a/src/test/scala/scala/async/run/ifelse0/IfElse0.scala
+++ b/src/test/scala/scala/async/run/ifelse0/IfElse0.scala
@@ -47,4 +47,12 @@ class IfElseSpec {
     val res = Await.result(fut, 2 seconds)
     res mustBe (14)
   }
+
+  @Test def `await in condition`() {
+    import AsyncId.{async, await}
+    val result = async {
+      if ({await(true); await(true)}) await(1) else ???
+    }
+    result mustBe (1)
+  }
 }