aboutsummaryrefslogtreecommitdiff
path: root/src/test/scala/scala/async/TreeInterrogation.scala
blob: f005b8a07974202896f0f9f69520638aee30111f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
/*
 * Copyright (C) 2012 Typesafe Inc. <http://www.typesafe.com>
 */

package scala.async

import org.junit.runner.RunWith
import org.junit.runners.JUnit4
import org.junit.Test
import AsyncId._

@RunWith(classOf[JUnit4])
class TreeInterrogation {
  @Test
  def `a minimal set of vals are lifted to vars`() {
    val cm = reflect.runtime.currentMirror
    val tb = mkToolbox("-cp target/scala-2.10/classes")
    val tree = tb.parse(
      """| import _root_.scala.async.AsyncId._
        | async {
        |   val x = await(1)
        |   val y = x * 2
        |   val z = await(x * 3)
        |   z
        | }""".stripMargin)
    val tree1 = tb.typeCheck(tree)

    //println(cm.universe.show(tree1))

    import tb.mirror.universe._
    val functions = tree1.collect {
      case f: Function => f
    }
    functions.size mustBe 1

    val varDefs = tree1.collect {
      case ValDef(mods, name, _, _) if mods.hasFlag(Flag.MUTABLE) => name
    }
    varDefs.map(_.decoded).toSet mustBe (Set("state$async", "onCompleteHandler$async", "await$1", "await$2"))
  }
}

object TreeInterrogation extends App {
  sys.props("scala.async.debug") = true.toString
  sys.props("scala.async.trace") = true.toString

  val cm   = reflect.runtime.currentMirror
  val tb   = mkToolbox("-cp target/scala-2.10/classes -Xprint:all")
  val tree = tb.parse(
    """ import _root_.scala.async.AsyncId._
      | async {
      |  val x = 1
      |  Option(x) match {
      |    case op @ Some(x) =>
      |      assert(op != null)
      |      println((op, x))
      |      x + await(x)
      |    case None => await(0)
      |  }
      | }
      | """.stripMargin)
  println(tree)
  val tree1 = tb.typeCheck(tree.duplicate)
  println(cm.universe.show(tree1))
  println(tb.eval(tree))
}