gotools/cmd/go2junit/go2junit.go

161 lines
3.9 KiB
Go
Raw Normal View History

2024-11-20 17:47:02 +00:00
package main
2024-11-20 20:13:17 +00:00
import (
"bufio"
"encoding/json"
"encoding/xml"
"fmt"
2024-11-22 21:03:23 +00:00
"log"
2024-11-20 20:13:17 +00:00
"os"
"path/filepath"
2024-11-22 20:50:28 +00:00
"strings"
2024-11-20 20:13:17 +00:00
)
2024-11-20 17:47:02 +00:00
func (t *Test) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
2024-11-22 21:45:58 +00:00
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",
}
}
2024-11-22 21:03:23 +00:00
log.Printf("TIME %f", t.Time)
2024-11-22 20:50:28 +00:00
parts := strings.Split(t.Name, "/")
tc := Testcase{
2024-11-22 20:50:28 +00:00
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)
}
2024-11-20 20:13:17 +00:00
func main() {
2024-11-20 20:13:17 +00:00
testsuites := Testsuites{}
if len(os.Args) != 2 && len(os.Args) != 3 {
fmt.Fprintf(os.Stderr, "Usage: go2junit <outputdir> [<prefix>]\n")
2024-11-20 20:13:17 +00:00
os.Exit(1)
}
path := os.Args[1]
path = filepath.Clean(path)
err := os.MkdirAll(path, 0755)
if err != nil {
panic(err)
2024-11-20 20:29:35 +00:00
}
prefix := ""
if len(os.Args) == 3 {
prefix = os.Args[2]
}
2024-11-20 20:13:17 +00:00
var file = os.Stdin
2024-11-20 20:13:17 +00:00
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
2024-11-20 20:13:17 +00:00
switch item.Action {
case "start":
testsuites.Start(item.Time, pkg)
2024-11-20 20:13:17 +00:00
case "run":
fmt.Println()
testsuites.Test(item.Time, pkg, item.Test)
2024-11-20 20:13:17 +00:00
case "output":
testsuites.Output(item.Time, pkg, item.Test, escapeIllegalChars(item.Output))
fmt.Printf("%s", item.Output)
2024-11-20 20:13:17 +00:00
case "pause":
testsuites.Output(item.Time, pkg, item.Test, "PAUSED")
2024-11-20 20:13:17 +00:00
case "cont":
testsuites.Output(item.Time, pkg, item.Test, "CONTINUED")
2024-11-20 20:13:17 +00:00
case "pass":
testsuites.Pass(item.Time, pkg, item.Test)
2024-11-20 20:13:17 +00:00
case "bench":
testsuites.Bench(item.Time, pkg, item.Test, escapeIllegalChars(item.Output))
2024-11-20 20:13:17 +00:00
case "fail":
testsuites.Fail(item.Time, pkg, item.Test)
2024-11-20 20:13:17 +00:00
case "skip":
testsuites.Skip(item.Time, pkg, item.Test)
2024-11-20 20:13:17 +00:00
}
}
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)
}
2024-11-20 20:13:17 +00:00
}
fmt.Printf("\n\nSUMMARY\n\n")
fmt.Printf("%-60s %-10s %-10s %-10s %-10s %-10s %-10s\n\n", "SUITE", "COUNT", "PASSED", "FAILURES", "ERRORS", "DISABLED", "SKIPPED")
for _, suite := range testsuites.Suites {
fmt.Printf("%-60s %-10d %-10d %-10d %-10d %-10d %-10d\n",
suite.Name,
suite.TestCount,
suite.TestCount-suite.Failures-suite.Errors-suite.Skipped-suite.Disabled,
suite.Failures, suite.Errors, suite.Disabled, suite.Skipped)
}
fmt.Printf("\n%-60s %-10d %-10d %-10d %-10d %-10d %-10d\n",
"TOTAL",
testsuites.Tests,
testsuites.Tests-testsuites.Failures-testsuites.Errors-testsuites.Skipped-testsuites.Disabled,
testsuites.Failures, testsuites.Errors, testsuites.Disabled, testsuites.Skipped)
if testsuites.Failures+testsuites.Errors+testsuites.Skipped+testsuites.Disabled > 0 {
fmt.Printf("\nFAILED TESTS\n\n")
printFailedTests("", "", testsuites.Suites)
}
}
func printFailedTests(indent string, parentTest string, tests []*Test) {
for _, test := range tests {
if test.Failures > 0 {
testName := strings.TrimPrefix(test.Name, parentTest+"/")
fmt.Printf("%s%s\n", indent, testName)
printFailedTests(indent+" ", test.Name, test.Tests)
}
}
2024-11-20 17:47:02 +00:00
}