132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/json"
|
|
"encoding/xml"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func (t *Test) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
|
if len(t.Tests) == 0 {
|
|
start.Name = xml.Name{Local: "testcase"}
|
|
classname := ""
|
|
if t.parent != nil {
|
|
classname = t.parent.Name
|
|
}
|
|
var skipped *Result
|
|
if t.Skipped > 0 {
|
|
skipped = &Result{
|
|
Message: "skipped",
|
|
}
|
|
}
|
|
var failed *Result
|
|
if t.Failures > 0 {
|
|
failed = &Result{
|
|
Message: "failed",
|
|
}
|
|
}
|
|
log.Printf("TIME %f", t.Time)
|
|
parts := strings.Split(t.Name, "/")
|
|
tc := Testcase{
|
|
Name: parts[len(parts)-1],
|
|
// parent class name
|
|
Classname: classname,
|
|
Time: t.Time,
|
|
Skipped: skipped,
|
|
Error: nil,
|
|
Failure: failed,
|
|
SystemOut: t.SystemOut,
|
|
}
|
|
return e.EncodeElement(tc, start)
|
|
}
|
|
type Alias Test
|
|
start.Name = xml.Name{Local: "testsuite"}
|
|
return e.EncodeElement(Alias(*t), start)
|
|
}
|
|
|
|
func main() {
|
|
testsuites := Testsuites{}
|
|
|
|
if len(os.Args) != 2 && len(os.Args) != 3 {
|
|
fmt.Fprintf(os.Stderr, "Usage: go2junit <outputdir> [<prefix>]\n")
|
|
os.Exit(1)
|
|
}
|
|
path := os.Args[1]
|
|
path = filepath.Clean(path)
|
|
err := os.MkdirAll(path, 0755)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
prefix := ""
|
|
if len(os.Args) == 3 {
|
|
prefix = os.Args[2]
|
|
}
|
|
|
|
var file = os.Stdin
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
lineno := 0
|
|
|
|
testsuites = Testsuites{}
|
|
for scanner.Scan() {
|
|
lineno++
|
|
var item TestEvent
|
|
line := scanner.Bytes()
|
|
if err := json.Unmarshal(line, &item); err != nil {
|
|
fmt.Fprintf(os.Stderr, "%d: %s: %v", lineno, line, err)
|
|
continue
|
|
}
|
|
|
|
//.fmt.Printf("Parsed %d:\n%v\n\n", lineno, item)
|
|
|
|
pkg := prefix + item.Package
|
|
switch item.Action {
|
|
case "start":
|
|
testsuites.Start(item.Time, pkg)
|
|
case "run":
|
|
fmt.Println()
|
|
testsuites.Test(item.Time, pkg, item.Test)
|
|
case "output":
|
|
testsuites.Output(item.Time, pkg, item.Test, escapeIllegalChars(item.Output))
|
|
fmt.Printf("%s", item.Output)
|
|
case "pause":
|
|
testsuites.Output(item.Time, pkg, item.Test, "PAUSED")
|
|
case "cont":
|
|
testsuites.Output(item.Time, pkg, item.Test, "CONTINUED")
|
|
case "pass":
|
|
testsuites.Pass(item.Time, pkg, item.Test)
|
|
case "bench":
|
|
testsuites.Bench(item.Time, pkg, item.Test, escapeIllegalChars(item.Output))
|
|
case "fail":
|
|
testsuites.Fail(item.Time, pkg, item.Test)
|
|
case "skip":
|
|
testsuites.Skip(item.Time, pkg, item.Test)
|
|
}
|
|
}
|
|
testsuites.Complete()
|
|
|
|
for _, suite := range testsuites.Suites {
|
|
xml, err := xml.MarshalIndent(suite, "", " ")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fname := path + "/" + suite.Name + ".xml"
|
|
dir := filepath.Dir(fname)
|
|
err = os.MkdirAll(dir, 0755)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Fprintf(os.Stderr, "Writing %s\n", fname)
|
|
err = os.WriteFile(fname, xml, 0644)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
}
|