summaryrefslogtreecommitdiff
path: root/vendor/github.com/golang/mock/gomock/call_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/golang/mock/gomock/call_test.go')
-rw-r--r--vendor/github.com/golang/mock/gomock/call_test.go47
1 files changed, 47 insertions, 0 deletions
diff --git a/vendor/github.com/golang/mock/gomock/call_test.go b/vendor/github.com/golang/mock/gomock/call_test.go
new file mode 100644
index 0000000..3ae7263
--- /dev/null
+++ b/vendor/github.com/golang/mock/gomock/call_test.go
@@ -0,0 +1,47 @@
+package gomock
+
+import "testing"
+
+type mockTestReporter struct {
+ errorCalls int
+ fatalCalls int
+}
+
+func (o *mockTestReporter) Errorf(format string, args ...interface{}) {
+ o.errorCalls++
+}
+
+func (o *mockTestReporter) Fatalf(format string, args ...interface{}) {
+ o.fatalCalls++
+}
+
+func TestCall_After(t *testing.T) {
+ t.Run("SelfPrereqCallsFatalf", func(t *testing.T) {
+ tr1 := &mockTestReporter{}
+
+ c := &Call{t: tr1}
+ c.After(c)
+
+ if tr1.fatalCalls != 1 {
+ t.Errorf("number of fatal calls == %v, want 1", tr1.fatalCalls)
+ }
+ })
+
+ t.Run("LoopInCallOrderCallsFatalf", func(t *testing.T) {
+ tr1 := &mockTestReporter{}
+ tr2 := &mockTestReporter{}
+
+ c1 := &Call{t: tr1}
+ c2 := &Call{t: tr2}
+ c1.After(c2)
+ c2.After(c1)
+
+ if tr1.errorCalls != 0 || tr1.fatalCalls != 0 {
+ t.Error("unexpected errors")
+ }
+
+ if tr2.fatalCalls != 1 {
+ t.Errorf("number of fatal calls == %v, want 1", tr2.fatalCalls)
+ }
+ })
+}