core/src/test/scala/scala/js/TestDynamic.scala

package scala.js

import scala.virtualization.lms.common._
import java.io.PrintWriter
import java.io.FileOutputStream
import language.{JS, Dynamics}
import exp.{JSExp, DynamicsExp}
import gen.js.{GenJS, GenDynamics}

trait DynamicProg { this: Dynamics =>
  def test(x: Rep[Any]): Rep[Any] = {
    dynamic(x).foo.bar(x).baz()
  }
}

trait AllocProg { this: JS =>
  def test(x: Rep[Any]): Rep[Any] = {
    val f = fun { (y : Rep[Any]) => new Record { val a = (newDynamic("Foo")(): Rep[Any]) } }
    f.apply(x)
  }
}

trait NewDynamicInFunProg { this: JS =>
  def test(x: Rep[Any]): Rep[Any] = {
    val f = fun { y : Rep[Any] => newDynamic("Foo")() }
    f.apply(x)
  }
}

trait DynamicUpdateProg { this: Dynamics =>
  def test(x: Rep[Any]): Rep[Any] = {
    val x_ = dynamic(x)
    x_.foo = x
    x_.foo
  }
}

trait DynamicInlineProg { this: Dynamics =>
  def test(x: Rep[Any]): Rep[Any] = {
    val self = inlineDynamic("this")
    val obj = inlineDynamic("foo.bar")
    obj.call(self)
  }
}

class TestDynamic extends FileDiffSuite {
  
  val prefix = "test-out/"
  
  def testDynamic = {
    withOutFile(prefix+"dynamic") {
      new DynamicProg with DynamicsExp { self =>
        val codegen = new GenDynamics { val IR: self.type = self }
        codegen.emitSource(test _, "main", new PrintWriter(System.out))
      }
    }
    assertFileEqualsCheck(prefix+"dynamic")
  }

  def testAlloc = {
    withOutFile(prefix+"dynamic_alloc") {
      new AllocProg with JSExp { self =>
        val codegen = new GenJS { val IR: self.type = self }
        codegen.emitSource(test _, "main", new PrintWriter(System.out))
      }
    }
    assertFileEqualsCheck(prefix+"dynamic_alloc")
  }

  def testNewDynamicInFun = {
    withOutFile(prefix+"dynamic_newinfun") {
      new NewDynamicInFunProg with JSExp { self =>
        val codegen = new GenJS { val IR: self.type = self }
        codegen.emitSource(test _, "main", new PrintWriter(System.out))
      }
    }
    assertFileEqualsCheck(prefix+"dynamic_newinfun")
  }

  def testDynamicUpdate = {
    withOutFile(prefix+"dynamicupdate") {
      new DynamicUpdateProg with DynamicsExp { self =>
        val codegen = new GenDynamics { val IR: self.type = self }
        codegen.emitSource(test _, "main", new PrintWriter(System.out))
      }
    }
    assertFileEqualsCheck(prefix+"dynamicupdate")
  }

  def testDynamicInline = {
    withOutFile(prefix+"dynamicinline") {
      new DynamicInlineProg with DynamicsExp { self =>
        val codegen = new GenDynamics { val IR: self.type = self }
        codegen.emitSource(test _, "main", new PrintWriter(System.out))
      }
    }
    assertFileEqualsCheck(prefix+"dynamicinline")
  }

}