From d64c83ad4f6d085cdbc387b5df9cf6331ea15803 Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Thu, 11 Feb 2016 11:07:18 -0800
Subject: [PATCH] Fix panic when Ack occurs at delivery limit

---
 nomad/eval_broker.go      |  2 +-
 nomad/eval_broker_test.go | 44 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 45 insertions(+), 1 deletion(-)

diff --git a/nomad/eval_broker.go b/nomad/eval_broker.go
index ead6f86bf71c..97c76eafaec5 100644
--- a/nomad/eval_broker.go
+++ b/nomad/eval_broker.go
@@ -448,7 +448,7 @@ func (b *EvalBroker) Ack(evalID, token string) error {
 	// Update the stats
 	b.stats.TotalUnacked -= 1
 	queue := unack.Eval.Type
-	if b.evals[evalID] >= b.deliveryLimit {
+	if b.evals[evalID] > b.deliveryLimit {
 		queue = failedQueue
 	}
 	bySched := b.stats.ByScheduler[queue]
diff --git a/nomad/eval_broker_test.go b/nomad/eval_broker_test.go
index 25faadc858df..b918b4e726c0 100644
--- a/nomad/eval_broker_test.go
+++ b/nomad/eval_broker_test.go
@@ -748,6 +748,50 @@ func TestEvalBroker_DeliveryLimit(t *testing.T) {
 	}
 }
 
+func TestEvalBroker_AckAtDeliveryLimit(t *testing.T) {
+	b := testBroker(t, 0)
+	b.SetEnabled(true)
+
+	eval := mock.Eval()
+	err := b.Enqueue(eval)
+	if err != nil {
+		t.Fatalf("err: %v", err)
+	}
+
+	for i := 0; i < 3; i++ {
+		// Dequeue should work
+		out, token, err := b.Dequeue(defaultSched, time.Second)
+		if err != nil {
+			t.Fatalf("err: %v", err)
+		}
+		if out != eval {
+			t.Fatalf("bad : %#v", out)
+		}
+
+		if i == 2 {
+			b.Ack(eval.ID, token)
+		} else {
+			// Nack with wrong token should fail
+			err = b.Nack(eval.ID, token)
+			if err != nil {
+				t.Fatalf("err: %v", err)
+			}
+		}
+	}
+
+	// Check the stats
+	stats := b.Stats()
+	if stats.TotalReady != 0 {
+		t.Fatalf("bad: %#v", stats)
+	}
+	if stats.TotalUnacked != 0 {
+		t.Fatalf("bad: %#v", stats)
+	}
+	if _, ok := stats.ByScheduler[failedQueue]; ok {
+		t.Fatalf("bad: %#v", stats)
+	}
+}
+
 // Ensure fairness between schedulers
 func TestEvalBroker_Wait(t *testing.T) {
 	b := testBroker(t, 0)