From 18dde48c600f0b78e200bbd68ebe823517abaa00 Mon Sep 17 00:00:00 2001
From: Evan Huus <eapache@gmail.com>
Date: Sat, 13 Jan 2024 11:34:46 -0500
Subject: [PATCH] Make breaker tests run faster

Timing tests are finicky, but 10ms increments seems to still be plenty
of margin of error in practice, and means these tests don't have to take
14s of sleep() anymore.
---
 breaker/breaker_test.go | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/breaker/breaker_test.go b/breaker/breaker_test.go
index b41308d..01cdb25 100644
--- a/breaker/breaker_test.go
+++ b/breaker/breaker_test.go
@@ -21,20 +21,20 @@ func returnsSuccess() error {
 }
 
 func TestBreakerErrorExpiry(t *testing.T) {
-	breaker := New(2, 1, 1*time.Second)
+	breaker := New(2, 1, 10*time.Millisecond)
 
 	for i := 0; i < 3; i++ {
 		if err := breaker.Run(returnsError); err != errSomeError {
 			t.Error(err)
 		}
-		time.Sleep(1 * time.Second)
+		time.Sleep(10 * time.Millisecond)
 	}
 
 	for i := 0; i < 3; i++ {
 		if err := breaker.Go(returnsError); err != nil {
 			t.Error(err)
 		}
-		time.Sleep(1 * time.Second)
+		time.Sleep(10 * time.Millisecond)
 	}
 }
 
@@ -66,7 +66,7 @@ func TestBreakerPanicsCountAsErrors(t *testing.T) {
 }
 
 func TestBreakerStateTransitions(t *testing.T) {
-	breaker := New(3, 2, 1*time.Second)
+	breaker := New(3, 2, 10*time.Millisecond)
 
 	// three errors opens the breaker
 	for i := 0; i < 3; i++ {
@@ -83,7 +83,7 @@ func TestBreakerStateTransitions(t *testing.T) {
 	}
 
 	// wait for it to half-close
-	time.Sleep(2 * time.Second)
+	time.Sleep(20 * time.Millisecond)
 	// one success works, but is not enough to fully close
 	if err := breaker.Run(returnsSuccess); err != nil {
 		t.Error(err)
@@ -98,7 +98,7 @@ func TestBreakerStateTransitions(t *testing.T) {
 	}
 
 	// wait for it to half-close
-	time.Sleep(2 * time.Second)
+	time.Sleep(20 * time.Millisecond)
 	// two successes is enough to close it for good
 	for i := 0; i < 2; i++ {
 		if err := breaker.Run(returnsSuccess); err != nil {
@@ -116,7 +116,7 @@ func TestBreakerStateTransitions(t *testing.T) {
 }
 
 func TestBreakerAsyncStateTransitions(t *testing.T) {
-	breaker := New(3, 2, 1*time.Second)
+	breaker := New(3, 2, 10*time.Millisecond)
 
 	// three errors opens the breaker
 	for i := 0; i < 3; i++ {
@@ -136,7 +136,7 @@ func TestBreakerAsyncStateTransitions(t *testing.T) {
 	}
 
 	// wait for it to half-close
-	time.Sleep(2 * time.Second)
+	time.Sleep(20 * time.Millisecond)
 	// one success works, but is not enough to fully close
 	if err := breaker.Go(returnsSuccess); err != nil {
 		t.Error(err)
@@ -153,7 +153,7 @@ func TestBreakerAsyncStateTransitions(t *testing.T) {
 	}
 
 	// wait for it to half-close
-	time.Sleep(2 * time.Second)
+	time.Sleep(20 * time.Millisecond)
 	// two successes is enough to close it for good
 	for i := 0; i < 2; i++ {
 		if err := breaker.Go(returnsSuccess); err != nil {