test fixes and delete

hashicorp · May 25, 2016 · bd079a2 · bd079a2
1 parent 8456f77
commit bd079a2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/nomad/blocked_evals.go b/nomad/blocked_evals.go
@@ -316,15 +316,17 @@ func (b *BlockedEvals) UnblockFailed() {
 	}
 
 	var unblock []*structs.Evaluation
-	for _, eval := range b.captured {
+	for id, eval := range b.captured {
 		if eval.TriggeredBy == structs.EvalTriggerMaxPlans {
 			unblock = append(unblock, eval)
+			delete(b.captured, id)
 		}
 	}
 
-	for _, eval := range b.escaped {
+	for id, eval := range b.escaped {
 		if eval.TriggeredBy == structs.EvalTriggerMaxPlans {
 			unblock = append(unblock, eval)
+			delete(b.escaped, id)
 		}
 	}
 

diff --git a/nomad/worker_test.go b/nomad/worker_test.go
@@ -466,13 +466,8 @@ func TestWorker_ReblockEval(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	// Enqueue the eval
-	testutil.WaitForResult(func() (bool, error) {
-		err := s1.evalBroker.Enqueue(eval1)
-		return err == nil, err
-	}, func(err error) {
-		t.Fatalf("err: %v", err)
-	})
+	// Enqueue the eval and then dequeue
+	s1.evalBroker.Enqueue(eval1)
 	evalOut, token, err := s1.evalBroker.Dequeue([]string{eval1.Type}, time.Second)
 	if err != nil {
 		t.Fatalf("err: %v", err)