Dogcows Code - chaz/sbt-tap/blob - src/main/scala/SbtTapReporting.scala

   1 import java.io.{PrintWriter, StringWriter, File, FileWriter}
   2 import sbt._
   3 import org.scalatools.testing.{Event => TEvent, Result => TResult}
   4
   5 import java.util.concurrent.atomic.AtomicInteger
   6
   7 object SbtTapReporting extends Plugin {
   8   def apply() = new SbtTapListener
   9 }
  10
  11 /**
  12  * Listens to sbt test listener events and writes them to a tap compatible file. Results for all groups
  13  * go to a single file although it might be desirable to generate one tap file per group.
  14  * <p>
  15  * sbt runs tests in parallel and the protocol does not seem to provide a way to match a group to a test event. It
  16  * does look line one thread calls startGroup/testEvent/endGroup sequentially and using thread local to keep
  17  * the current active group might be one way to go.
  18  */
  19 class SbtTapListener extends TestsListener {
  20   var testId = new AtomicInteger(0)
  21   var fileWriter: FileWriter = _
  22
  23   override def doInit {
  24     new File("test-results").mkdirs()
  25
  26     fileWriter = new FileWriter("test-results/test.tap")
  27   }
  28
  29   def startGroup(name: String) {}
  30
  31   def testEvent(event: TestEvent) {
  32     event.detail.foreach { e: TEvent =>
  33       e.result match {
  34         case TResult.Success => writeTapFields("ok", testId.incrementAndGet(), "-", e.testName())
  35         case TResult.Error | TResult.Failure =>
  36           writeTapFields("not ok", testId.incrementAndGet(), "-", e.testName())
  37           // According to the TAP spec, as long as there is any kind of whitespace, this output should belong to the
  38           // the test that failed and it should get displayed in the UI.
  39           // TODO:It would be nice if we could report the exact line in the test where this happened.
  40           writeTapFields(" ", stackTraceForError(e.error()))
  41         case TResult.Skipped =>
  42           // it doesn't look like this framework distinguishes between pending and ignored.
  43           writeTapFields("ok", testId.incrementAndGet(), e.testName(), "#", "skip", e.testName())
  44       }
  45     }
  46   }
  47
  48   override def doComplete(finalResult: TestResult.Value) {
  49     writeTapFields("1.." + testId.get)
  50     fileWriter.close()
  51   }
  52
  53   private def writeTapFields(s: Any*) { fileWriter.write(s.mkString("",  " ", "\n")) }
  54
  55   private def stackTraceForError(t: Throwable): String = {
  56     val sw = new StringWriter()
  57     val printWriter = new PrintWriter(sw)
  58     t.printStackTrace(printWriter)
  59     sw.toString
  60   }
  61   def endGroup(name: String, t: Throwable) { }
  62
  63   def endGroup(name: String, result: TestResult.Value) { }
  64 }