Forbid await in try/catch.

1 files changed, 30 insertions, 0 deletions

diff --git a/src/test/scala/scala/async/neg/NakedAwait.scala b/src/test/scala/scala/async/neg/NakedAwait.scala
index 8b85977..a0c4e4d 100644
--- a/src/test/scala/scala/async/neg/NakedAwait.scala
+++ b/src/test/scala/scala/async/neg/NakedAwait.scala
@@ -87,4 +87,34 @@ class NakedAwait {
       """.stripMargin
     }
   }
+
+  @Test
+  def tryBody() {
+    expectError("await must not be used under a try/catch.") {
+      """
+        | import _root_.scala.async.AsyncId._
+        | async { try { await(false) } catch { case _ => } }
+      """.stripMargin
+    }
+  }
+
+  @Test
+  def catchBody() {
+    expectError("await must not be used under a try/catch.") {
+      """
+        | import _root_.scala.async.AsyncId._
+        | async { try { () } catch { case _ => await(false) } }
+      """.stripMargin
+    }
+  }
+
+  @Test
+  def finallyBody() {
+    expectError("await must not be used under a try/catch.") {
+      """
+        | import _root_.scala.async.AsyncId._
+        | async { try { () } finally { await(false) } }
+      """.stripMargin
+    }
+  }
 }