test policy delay for startup error

hashicorp · Jul 12, 2016 · f884324 · f884324
1 parent f255464
commit f884324
Showing 1 changed file with 27 additions and 2 deletions.
diff --git a/client/restarts_test.go b/client/restarts_test.go
@@ -94,9 +94,9 @@ func TestClient_RestartTracker_ZeroAttempts(t *testing.T) {
 	}
 }
 
-func TestClient_RestartTracker_StartError_Recoverable(t *testing.T) {
+func TestClient_RestartTracker_StartError_Recoverable_Fail(t *testing.T) {
 	t.Parallel()
-	p := testPolicy(true, structs.RestartPolicyModeDelay)
+	p := testPolicy(true, structs.RestartPolicyModeFail)
 	rt := newRestartTracker(p, structs.JobTypeSystem)
 	recErr := cstructs.NewRecoverableError(fmt.Errorf("foo"), true)
 	for i := 0; i < p.Attempts; i++ {
@@ -114,3 +114,28 @@ func TestClient_RestartTracker_StartError_Recoverable(t *testing.T) {
 		t.Fatalf("NextRestart() returned %v; want %v", state, structs.TaskNotRestarting)
 	}
 }
+
+func TestClient_RestartTracker_StartError_Recoverable_Delay(t *testing.T) {
+        t.Parallel()
+        p := testPolicy(true, structs.RestartPolicyModeDelay)
+        rt := newRestartTracker(p, structs.JobTypeSystem)
+        recErr := cstructs.NewRecoverableError(fmt.Errorf("foo"), true)
+        for i := 0; i < p.Attempts; i++ {
+                state, when := rt.SetStartError(recErr).GetState()
+                if state != structs.TaskRestarting {
+                        t.Fatalf("NextRestart() returned %v, want %v", state, structs.TaskRestarting)
+                }
+                if !withinJitter(p.Delay, when) {
+                        t.Fatalf("NextRestart() returned %v; want %v+jitter", when, p.Delay)
+                }
+        }
+
+        // Next restart should cause delay
+        state, when := rt.SetStartError(recErr).GetState()
+        if state != structs.TaskRestarting {
+                t.Fatalf("NextRestart() returned %v; want %v", state, structs.TaskRestarting)
+        }
+        if !(when > p.Delay && when <= p.Interval) {
+                t.Fatalf("NextRestart() returned %v; want > %v and <= %v", when, p.Delay, p.Interval)
+        }
+}