add benchmarks

avast · Aug 3, 2023 · d090128 · d090128
1 parent 6d7600c
commit d090128
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/retry_test.go b/retry_test.go
@@ -553,6 +553,20 @@ func BenchmarkDo(b *testing.B) {
 	}
 }
 
+func BenchmarkDoWithData(b *testing.B) {
+	testError := errors.New("test error")
+
+	for i := 0; i < b.N; i++ {
+		_, _ = DoWithData(
+			func() (int, error) {
+				return 0, testError
+			},
+			Attempts(10),
+			Delay(0),
+		)
+	}
+}
+
 func BenchmarkDoNoErrors(b *testing.B) {
 	for i := 0; i < b.N; i++ {
 		_ = Do(
@@ -565,6 +579,18 @@ func BenchmarkDoNoErrors(b *testing.B) {
 	}
 }
 
+func BenchmarkDoWithDataNoErrors(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		_, _ = DoWithData(
+			func() (int, error) {
+				return 0, nil
+			},
+			Attempts(10),
+			Delay(0),
+		)
+	}
+}
+
 func TestIsRecoverable(t *testing.T) {
 	err := errors.New("err")
 	assert.True(t, IsRecoverable(err))