more loom tests

rssh · Sep 28, 2023 · 022289e · 022289e
1 parent ef76d2e
commit 022289e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/compiler-plugin/src/test/scala/cc/Test18Loom.scala b/compiler-plugin/src/test/scala/cc/Test18Loom.scala
@@ -9,17 +9,29 @@ class Test18Loom {
     DotcInvocations.compileAndRunJunitTestAfterDependency(dirname, testClassName, invocationArgs, Test18Loom.commonDependency)
   }
 
+
   @Test
   def testHOFunctionArg() = {
     compileAndRunTestAfterCommon("testdata/set18loom/m1", "cpsloomtest.TestHOFunctionArgLoomTransform",
       DotcInvocationArgs(checkAll = true,
         extraDotcArgs = List(
-           "-Vprint:rssh.cps"
+          // "-Vprint:rssh.cps"
         )
       )
     )
   }
+
 
+  @Test
+  def testTwiceWithFuture() = {
+    compileAndRunTestAfterCommon("testdata/set18loom/m2", "cpsloomtest.TestTwice",
+      DotcInvocationArgs(checkAll = true,
+        extraDotcArgs = List(
+          // "-Vprint:rssh.cps"
+        )
+      )
+    )
+  }
 
 }
 

diff --git a/jvm-loom-addon/src/test/scala/cpsloomtest/TestTwice.scala b/jvm-loom-addon/src/test/scala/cpsloomtest/TestTwice.scala
@@ -17,12 +17,13 @@ class TestTwice {
   def testAwaitInTwiceArg(): Unit = {
 
     val f = async[Future]{
-      twice{ (x: Int) =>
-        await(Future.successful(x+1))
+      twice { (x: Int) =>
+        await(Future.successful(x + 1))
       }(1)
     }
     val r = Await.result(f, 1.second)
     assert(r == 3)
+
   }
 
 }