Fixed a bug with the Limit strategy

Rican7 · Jun 30, 2016 · 678601c · 678601c
1 parent 2f098aa
commit 678601c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/strategy/strategy.go b/strategy/strategy.go
@@ -21,7 +21,7 @@ type Strategy func(attempt uint) bool
 // make.
 func Limit(attemptLimit uint) Strategy {
 	return func(attempt uint) bool {
-		return (attempt < attemptLimit)
+		return (attempt <= attemptLimit)
 	}
 }
 

diff --git a/strategy/strategy_test.go b/strategy/strategy_test.go
@@ -18,7 +18,11 @@ func TestLimit(t *testing.T) {
 		t.Error("strategy expected to return true")
 	}
 
-	if strategy(3) {
+	if !strategy(3) {
+		t.Error("strategy expected to return true")
+	}
+
+	if strategy(4) {
 		t.Error("strategy expected to return false")
 	}
 }