From c1eb3429631a69a1517290084a5f9e68218ba870 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=ABl=20Stemmer?= <stemmertech@gmail.com>
Date: Sat, 21 Apr 2018 14:37:15 +0100
Subject: [PATCH] Add -match test flag to make testing specific tests easier

---
 go-junit-report_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/go-junit-report_test.go b/go-junit-report_test.go
index 1973077..2699c03 100644
--- a/go-junit-report_test.go
+++ b/go-junit-report_test.go
@@ -2,9 +2,11 @@ package main
 
 import (
 	"bytes"
+	"flag"
 	"fmt"
 	"io/ioutil"
 	"os"
+	"regexp"
 	"runtime"
 	"strings"
 	"testing"
@@ -13,6 +15,8 @@ import (
 	"github.com/jstemmer/go-junit-report/parser"
 )
 
+var matchTest = flag.String("match", "", "only test testdata matching this pattern")
+
 type TestCase struct {
 	name        string
 	reportName  string
@@ -830,7 +834,11 @@ var testCases = []TestCase{
 }
 
 func TestParser(t *testing.T) {
+	matchRegex := compileMatch(t)
 	for _, testCase := range testCases {
+		if !matchRegex.MatchString(testCase.name) {
+			continue
+		}
 		t.Logf("Test %s", testCase.name)
 
 		file, err := os.Open("testdata/" + testCase.name)
@@ -904,7 +912,12 @@ func TestVersionFlag(t *testing.T) {
 }
 
 func testJUnitFormatter(t *testing.T, goVersion string) {
+	match := compileMatch(t)
 	for _, testCase := range testCases {
+		if !match.MatchString(testCase.name) {
+			continue
+		}
+
 		report, err := loadTestReport(testCase.reportName, goVersion)
 		if err != nil {
 			t.Fatal(err)
@@ -938,3 +951,11 @@ func loadTestReport(name, goVersion string) (string, error) {
 
 	return report, nil
 }
+
+func compileMatch(t *testing.T) *regexp.Regexp {
+	rx, err := regexp.Compile(*matchTest)
+	if err != nil {
+		t.Fatalf("Error compiling -match flag %q: %v", *matchTest, err)
+	}
+	return rx
+}