package com.iinteractive.test.sbt

import org.scalatools.testing

import com.iinteractive.test.harness.{Reporter,SummarizedTests}
import com.iinteractive.test.tap.{TAPEvent,ResultEvent,EndEvent}
import com.iinteractive.test.Test

/** Runs a single test under the SBT test harness. */
class SBTReporter (
  loader:       ClassLoader,
  loggers:      Array[testing.Logger],
  eventHandler: testing.EventHandler
) extends Reporter with SummarizedTests {
  def run (testName: String): Int = {
    val cb = (e: TAPEvent) => e match {
      case ResultEvent(r) => {
        val event = new testing.Event {
          val testName:    String = r.description
          val description: String = r.description
          val result:      testing.Result =
            if (r.passed) {
              testing.Result.Success
            }
            else if (r.directive.isDefined) {
              testing.Result.Skipped
            }
            else {
              testing.Result.Failure
            }
          val error: Throwable = null
        }
        eventHandler.handle(event)
      }
      case EndEvent(result) => {
        val testsPassed = result.success
        val correctCode = result.exitCode == 0
        val event = new testing.Event {
          val testName:    String = "exit code is 0"
          val description: String = "exit code is 0"
          val result:      testing.Result =
            if (correctCode) {
              testing.Result.Success
            }
            else {
              testing.Result.Failure
            }
          val error: Throwable = null
        }
        eventHandler.handle(event)

        if (testsPassed && correctCode) {
          logInfo("PASS " + testName)
        }
        else {
          val results = result.results.length
          val failed = result.results.count { t =>
            !t.passed && !t.directive.isDefined
          }

          val errors = Seq(
            (if (testsPassed)
              None
            else
              Some("failed " + failed + "/" + results)),
            (if (correctCode)
              None
            else
              Some("non-zero exit code: " + result.exitCode))
          ).flatten.mkString("(", ", ", ")")

          logError("FAIL " + testName + " " + errors)
        }
      }
      case _ => ()
    }

    runOneTest(
      loader.loadClass(testName).newInstance.asInstanceOf[Test],
      cb
    ).exitCode
  }

  private def logDebug (msg: String) {
    loggers.foreach(_.debug(msg))
  }

  private def logInfo (msg: String) {
    loggers.foreach(_.info(msg))
  }

  private def logWarn (msg: String) {
    loggers.foreach(_.warn(msg))
  }

  private def logError (msg: String) {
    loggers.foreach(_.error(msg))
  }
}