Handle arbitrary number of sub suites in junit reports

go.mod

@@ -2,4 +2,7 @@ module github.com/joshdk/go-junit
 
 go 1.12
 
-require github.com/stretchr/testify v1.4.0
+require (
+	github.com/davecgh/go-spew v1.1.0
+	github.com/stretchr/testify v1.4.0
+)

ingest.go

@@ -30,6 +30,9 @@ func ingestSuite(root xmlNode) Suite {
 
 	for _, node := range root.Nodes {
 		switch node.XMLName.Local {
+		case "testsuite":
+			testsuite := ingestSuite(node)
+			suite.Suites = append(suite.Suites, testsuite)
 		case "testcase":
 			testcase := ingestTestcase(node)
 			suite.Tests = append(suite.Tests, testcase)

types.go

@@ -4,7 +4,9 @@
 
 package junit
 
-import "time"
+import (
+	"time"
+)
 
 // Status represents the result of a single a JUnit testcase. Indicates if a
 // testcase was run, and if it was successful.
@@ -71,6 +73,9 @@ type Suite struct {
 	// Tests is an ordered collection of tests with associated results.
 	Tests []Test `json:"tests,omitempty" yaml:"tests,omitempty"`
 
+	// Suites is an ordered collection of suites with associated results.
+	Suites []Suite `json:"suites,omitempty" yaml:"suites,omitempty"`
+
 	// SystemOut is textual test output for the suite. Usually output that is
 	// written to stdout.
 	SystemOut string `json:"stdout,omitempty" yaml:"stdout,omitempty"`
@@ -101,6 +106,15 @@ func (s *Suite) Aggregate() {
 		}
 	}
 
+	for _, suite := range s.Suites {
+		suite.Aggregate()
+		totals.Duration += suite.Totals.Duration
+		totals.Passed += suite.Totals.Passed
+		totals.Skipped += suite.Totals.Skipped
+		totals.Failed += suite.Totals.Failed
+		totals.Error += suite.Totals.Error
+	}
+
 	s.Totals = totals
 }