org.scalajs.testinterface.internal.Master.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalajs-test-interface_2.10 Show documentation

Scala.js test interface

There is a newer version: 1.0.0-M2

Show newest version

package org.scalajs.testinterface.internal

import scala.scalajs.js
import js.Dynamic.{literal => lit}
import js.annotation.JSExport

import sbt.testing._

import scala.util.Try

import org.scalajs.testinterface.ScalaJSClassLoader

@JSExport
final class Master(frameworkName: String) extends BridgeBase(frameworkName) {

  private[this] var runner: Runner = _

  protected def handleMsgImpl(cmd: String, strArg: => String): Unit = {
    def jsonArg = js.JSON.parse(strArg)
    cmd match {
      case "newRunner" =>
        reply(newRunner(jsonArg))
      case "runnerDone" =>
        reply(runnerDone())
      case "tasks" =>
        reply(tasks(jsonArg))
      case "msg" =>
        reply(inboundMessage(strArg))
      case cmd =>
        throw new IllegalArgumentException(s"Unknown command: $cmd")
    }
  }

  // Message handler methods

  private def newRunner(data: js.Dynamic): Try[Unit] = {
    val args = data.args.asInstanceOf[js.Array[String]].toArray
    val remoteArgs = data.remoteArgs.asInstanceOf[js.Array[String]].toArray
    val loader = new ScalaJSClassLoader(js.Dynamic.global)

    Try(runner = framework.runner(args, remoteArgs, loader))
  }

  private def runnerDone(): Try[String] = {
    ensureRunnerExists()

    val result = Try(runner.done())
    runner = null
    result
  }

  private def tasks(data: js.Dynamic): Try[String] = {
    ensureRunnerExists()

    val taskDefs = data.asInstanceOf[js.Array[js.Dynamic]]
      .map(TaskDefSerializer.deserialize).toArray

    Try {
      val tasks = runner.tasks(taskDefs)
      js.JSON.stringify(tasks2TaskInfos(tasks, runner))
    }
  }

  private def inboundMessage(msg: String): Try[String] = {
    ensureRunnerExists()
    Try(runner.receiveMessage(msg).fold(":n")(":s:" + _))
  }

  // Utility methods

  private def ensureRunnerExists(): Unit = {
    if (runner == null)
      throw new IllegalStateException("No runner created")
  }

}