add strict verify only for non-verified methods (#84)

ovechkin-dm · Oct 28, 2024 · 5450d08 · 5450d08
1 parent 52470ce
commit 5450d08
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/registry/handler.go b/registry/handler.go
@@ -352,7 +352,9 @@ func (h *invocationHandler) TearDown() {
 	if h.ctx.cfg.StrictVerify {
 		for _, m := range h.methods {
 			for _, mm := range m.methodMatches {
-				mm.verifiers = append(mm.verifiers, matchers.AtLeastOnce())
+				if len(mm.verifiers) == 0 {
+					mm.verifiers = append(mm.verifiers, matchers.AtLeastOnce())
+				}
 			}
 		}
 		h.VerifyNoMoreInteractions(true)

diff --git a/tests/verify/verify_test.go b/tests/verify/verify_test.go
@@ -187,3 +187,9 @@ func TestStrictVerifyUnverifiedStub(t *testing.T) {
 	r.TriggerCleanup()
 	r.AssertError()
 }
+
+func TestVerifyNeverInReturner(t *testing.T) {
+	SetUp(t, mockopts.StrictVerify())
+	m := Mock[iface]()
+	WhenSingle(m.Foo(12)).Verify(Never())
+}