Enables testing the resetting of lifted local variables

- Adds a hook that lets a derived macro insert additional code
  when zero-ing out a lifted field.
- Adds a variant of the `AsyncId` macro that logs zeroed-out fields.
- Adds a test using this mechanism

1 files changed, 40 insertions, 0 deletions

diff --git a/src/test/scala/scala/async/run/live/LiveVariablesSpec.scala b/src/test/scala/scala/async/run/live/LiveVariablesSpec.scala
new file mode 100644
index 0000000..2cecffa
--- /dev/null
+++ b/src/test/scala/scala/async/run/live/LiveVariablesSpec.scala
@@ -0,0 +1,40 @@
+/*
+ * Copyright (C) 2012-2013 Typesafe Inc. <http://www.typesafe.com>
+ */
+
+package scala.async
+package run
+package live
+
+import org.junit.Test
+
+import internal.AsyncTestLV
+import AsyncTestLV._
+
+class LiveVariablesSpec {
+
+  @Test
+  def liveVars1() {
+    val f = async { 1 }
+
+    def m1(x: Int): Int =
+      async { x + 1 }
+
+    def m2(x: Int): String =
+      async { x.toString }
+
+    def m3() = async {
+      val a = await(f)          // await$1$1
+      // a == 1
+      val b = await(m1(a))      // await$2$1
+      // b == 2
+      assert(AsyncTestLV.log.exists(_ == ("await$1$1" -> 0)))
+      val res = await(m2(b))    // await$3$1
+      assert(AsyncTestLV.log.exists(_ == ("await$2$1" -> 0)))
+      res
+    }
+
+    assert(m3() == "2")
+  }
+
+}