refactor: split into packages and add tests

internal/report/junit/junit_test.go

@@ -0,0 +1,116 @@
+package junit
+
+import (
+	"encoding/xml"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+	"time"
+
+	"gitlab.com/technofab/nixtest/internal/types"
+)
+
+func formatDurationSeconds(d time.Duration) string {
+	return fmt.Sprintf("%.3f", d.Seconds())
+}
+
+func TestGenerateReport(t *testing.T) {
+	results := types.Results{
+		"Suite1": []types.TestResult{
+			{
+				Spec:     types.TestSpec{Name: "Test1_Success", Suite: "Suite1", Pos: "file1.nix:10"},
+				Status:   types.StatusSuccess,
+				Duration: 123 * time.Millisecond,
+			},
+			{
+				Spec:     types.TestSpec{Name: "Test2_Failure", Suite: "Suite1", Pos: "file1.nix:20"},
+				Status:   types.StatusFailure,
+				Duration: 234 * time.Millisecond,
+				Expected: "hello",
+				Actual:   "world",
+			},
+		},
+		"Suite2": []types.TestResult{
+			{
+				Spec:         types.TestSpec{Name: "Test3_Error", Suite: "Suite2"},
+				Status:       types.StatusError,
+				Duration:     345 * time.Millisecond,
+				ErrorMessage: "Something went very wrong",
+			},
+			{
+				Spec:         types.TestSpec{Name: "Test4_Failure_Message", Suite: "Suite2"},
+				Status:       types.StatusFailure,
+				Duration:     456 * time.Millisecond,
+				ErrorMessage: "hello world",
+			},
+			{
+				Spec:     types.TestSpec{Name: "Test5_Skipped", Suite: "Suite2"},
+				Status:   types.StatusSkipped,
+				Duration: 567 * time.Millisecond,
+			},
+		},
+	}
+
+	totalDuration := (123 + 234 + 345 + 456 + 567) * time.Millisecond
+	reportName := "MyNixtestReport"
+	xmlString, err := GenerateReport(reportName, results)
+	if err != nil {
+		t.Fatalf("GenerateReport() failed: %v", err)
+	}
+
+	if !strings.HasPrefix(xmlString, xml.Header) {
+		t.Error("GenerateReport() output missing XML header")
+	}
+	if !strings.Contains(xmlString, "<testsuites name=\"MyNixtestReport\"") {
+		t.Errorf("GenerateReport() missing root <testsuites>. Got: %s", xmlString)
+	}
+	if !strings.Contains(xmlString, "tests=\"5\"") {
+		t.Errorf("GenerateReport() incorrect total tests count. Got: %s", xmlString)
+	}
+	if !strings.Contains(xmlString, "failures=\"2\"") {
+		t.Errorf("GenerateReport() incorrect total failures count. Got: %s", xmlString)
+	}
+	if !strings.Contains(xmlString, "errors=\"1\"") {
+		t.Errorf("GenerateReport() incorrect total errors count. Got: %s", xmlString)
+	}
+	if !strings.Contains(xmlString, "time=\""+formatDurationSeconds(totalDuration)+"\"") {
+		t.Errorf("GenerateReport() incorrect total time. Expected %s. Got part: %s", formatDurationSeconds(totalDuration), xmlString)
+	}
+
+	var report JUnitReport
+	if err := xml.Unmarshal([]byte(strings.TrimPrefix(xmlString, xml.Header)), &report); err != nil {
+		t.Fatalf("Failed to unmarshal generated XML: %v\nXML:\n%s", err, xmlString)
+	}
+
+	if report.Name != reportName {
+		t.Errorf("Report.Name = %q, want %q", report.Name, reportName)
+	}
+	if len(report.Suites) != 2 {
+		t.Fatalf("Report.Suites length = %d, want 2", len(report.Suites))
+	}
+}
+
+func TestWriteFile(t *testing.T) {
+	tempDir := t.TempDir()
+	filePath := filepath.Join(tempDir, "junit_report.xml")
+	results := types.Results{
+		"SuiteSimple": []types.TestResult{
+			{Spec: types.TestSpec{Name: "SimpleTest", Suite: "SuiteSimple"}, Status: types.StatusSuccess, Duration: 1 * time.Second},
+		},
+	}
+
+	err := WriteFile(filePath, "TestReport", results)
+	if err != nil {
+		t.Fatalf("WriteFile() failed: %v", err)
+	}
+
+	content, err := os.ReadFile(filePath)
+	if err != nil {
+		t.Fatalf("Failed to read written JUnit file: %v", err)
+	}
+	if !strings.Contains(string(content), "<testsuites name=\"TestReport\"") {
+		t.Error("Written JUnit file content seems incorrect.")
+	}
+}