junit.go

package junit

import (
	"bytes"
	"encoding/xml"
	"fmt"
	"math"
	"os"
	"runtime"
	"sync/atomic"
	"time"

	"github.com/aphistic/sweet"
)

func roundTime(duration time.Duration) float64 {
	seconds := float64(duration) / float64(time.Second)
	ms := math.Floor(seconds * 10000)
	return ms / 10000
}

type JUnitPlugin struct {
	suites *testSuites
	output string
}

var _ sweet.Plugin = &JUnitPlugin{}

func NewPlugin() *JUnitPlugin {
	return &JUnitPlugin{
		suites: newTestSuites(),
	}
}

func (p *JUnitPlugin) Name() string {
	return "JUnit Output"
}

func (p *JUnitPlugin) Options() *sweet.PluginOptions {
	return &sweet.PluginOptions{
		Prefix: "junit",
		Options: map[string]*sweet.PluginOption{
			"output": &sweet.PluginOption{
				Help: "Results of the test run will be written in JUnit format to the path provided",
			},
		},
	}
}

func (p *JUnitPlugin) SetOption(name, value string) {
	if name == "output" {
		p.output = value
	}
}

func (p *JUnitPlugin) Starting() {

}
func (p *JUnitPlugin) SuiteStarting(suite string) {
	s := p.suites.GetSuite(suite)
	s.AddProperty("go.version", runtime.Version())
}
func (p *JUnitPlugin) TestStarting(testName *sweet.TestName) {

}
func (p *JUnitPlugin) TestPassed(testName *sweet.TestName, stats *sweet.TestPassedStats) {
	s := p.suites.GetSuite(testName.SuiteName)
	atomic.AddInt64(&s.Tests, 1)
	s.AddTestCase(&testCase{
		Name:      testName.String(),
		ClassName: testName.SuiteName,
		Time:      roundTime(stats.Time),
	})
}
func (p *JUnitPlugin) TestFailed(testName *sweet.TestName, stats *sweet.TestFailedStats) {
	s := p.suites.GetSuite(testName.SuiteName)
	atomic.AddInt64(&s.Tests, 1)
	atomic.AddInt64(&s.Failures, 1)

	tc := &testCase{
		Name:      testName.String(),
		ClassName: testName.SuiteName,
		Time:      roundTime(stats.Time),
	}

	file := "<unknown>"
	line := 0
	if len(stats.Frames) > 0 {
		file = stats.Frames[0].File
		line = stats.Frames[0].Line
	}
	tc.SetFailure(file, line, stats.Message)
	s.AddTestCase(tc)
}
func (p *JUnitPlugin) TestSkipped(testName *sweet.TestName, stats *sweet.TestSkippedStats) {
}
func (p *JUnitPlugin) SuiteFinished(suite string, stats *sweet.SuiteFinishedStats) {
	s := p.suites.GetSuite(suite)
	s.Time = roundTime(stats.Time)
}
func (p *JUnitPlugin) Finished() {
	if p.output == "" {
		return
	}

	of, err := os.OpenFile(p.output, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
	if err != nil {
		fmt.Printf("Unable to open file for JUnit output: %s\n", err)
		os.Exit(1)
	}
	defer of.Close()

	x, err := p.generateXML()
	if err != nil {
		fmt.Printf("Error generating xml: %s", err)
		return
	}

	of.WriteString(xml.Header)
	of.WriteString(x)
	of.WriteString("\n")
}

func (p *JUnitPlugin) generateXML() (string, error) {
	buffer := &bytes.Buffer{}
	enc := xml.NewEncoder(buffer)
	enc.Indent("", "    ")

	err := enc.Encode(p.suites)
	if err != nil {
		return "", err
	}

	return buffer.String(), nil
}