diff options
author | Jesse Duffield <jessedduffield@gmail.com> | 2021-12-28 13:58:09 +1100 |
---|---|---|
committer | Jesse Duffield <jessedduffield@gmail.com> | 2021-12-29 09:01:06 +1100 |
commit | 9ef65574db3f0fd30b3782a8582212619edf4650 (patch) | |
tree | 70a85c69018d6bec09d42095245fde3428e7fb01 /pkg/test | |
parent | f89747451a609484bea0b2b5a004a6c3da66aaeb (diff) |
refactor to rename pull_request to hosting_service and apply SRP
Diffstat (limited to 'pkg/test')
-rw-r--r-- | pkg/test/log.go | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/pkg/test/log.go b/pkg/test/log.go new file mode 100644 index 000000000..3b166bb5d --- /dev/null +++ b/pkg/test/log.go @@ -0,0 +1,41 @@ +package test + +import ( + "fmt" + "testing" + + "github.com/sirupsen/logrus" + "github.com/stretchr/testify/assert" +) + +var ( + _ logrus.FieldLogger = &FakeFieldLogger{} +) + +// for now we're just tracking calls to the Error and Errorf methods +type FakeFieldLogger struct { + loggedErrors []string + *logrus.Entry +} + +func (self *FakeFieldLogger) Error(args ...interface{}) { + if len(args) != 1 { + panic("Expected exactly one argument to FakeFieldLogger.Error") + } + + switch arg := args[0].(type) { + case error: + self.loggedErrors = append(self.loggedErrors, arg.Error()) + case string: + self.loggedErrors = append(self.loggedErrors, arg) + } +} + +func (self *FakeFieldLogger) Errorf(format string, args ...interface{}) { + msg := fmt.Sprintf(format, args...) + self.loggedErrors = append(self.loggedErrors, msg) +} + +func (self *FakeFieldLogger) AssertErrors(t *testing.T, expectedErrors []string) { + assert.EqualValues(t, expectedErrors, self.loggedErrors) +} |