From b0a9864d1ec36fad2bd2f692983564112c0fc46a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=ABl=20Stemmer?= <stemmertech@gmail.com>
Date: Tue, 15 Mar 2022 19:41:49 +0000
Subject: [PATCH] parser/gotest: Add Option to override the timestamp function

By default, the current local time is used when generating a Report. The
TimestampFunc Option makes it possible to override this default
behaviour. This can for example be used in tests to make sure the
timestamp is a fixed point in time.
---
 pkg/parser/gotest/gotest.go | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/pkg/parser/gotest/gotest.go b/pkg/parser/gotest/gotest.go
index 44efbbe..122fbaa 100644
--- a/pkg/parser/gotest/gotest.go
+++ b/pkg/parser/gotest/gotest.go
@@ -38,14 +38,23 @@ var (
 // Option defines options that can be passed to gotest.New.
 type Option func(*Parser)
 
-// PackageName sets the default package name to use when it cannot be
-// determined from the test output.
+// PackageName is an Option that sets the default package name to use when it
+// cannot be determined from the test output.
 func PackageName(name string) Option {
 	return func(p *Parser) {
 		p.packageName = name
 	}
 }
 
+// TimestampFunc is an Option that sets the timestamp function that is used to
+// determine the current time when creating the Report. This can be used to
+// override the default behaviour of using time.Now().
+func TimestampFunc(f func() time.Time) Option {
+	return func(p *Parser) {
+		p.timestampFunc = f
+	}
+}
+
 // New returns a new Go test output parser.
 func New(options ...Option) *Parser {
 	p := &Parser{}
@@ -57,7 +66,8 @@ func New(options ...Option) *Parser {
 
 // Parser is a Go test output Parser.
 type Parser struct {
-	packageName string
+	packageName   string
+	timestampFunc func() time.Time
 
 	events []Event
 }
@@ -77,6 +87,7 @@ func (p *Parser) Parse(r io.Reader) (gtr.Report, error) {
 func (p *Parser) report(events []Event) gtr.Report {
 	rb := gtr.NewReportBuilder()
 	rb.PackageName = p.packageName
+	rb.TimestampFunc = p.timestampFunc
 	for _, ev := range events {
 		switch ev.Type {
 		case "run_test":