From 82145ee2d84229bcd673fc9749f604f346bc6a61 Mon Sep 17 00:00:00 2001
From: Armon Dadgar <armon.dadgar@gmail.com>
Date: Wed, 5 Aug 2015 16:46:07 -0700
Subject: [PATCH] nomad: increase test reliability

---
 nomad/eval_endpoint_test.go | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/nomad/eval_endpoint_test.go b/nomad/eval_endpoint_test.go
index 255ee25a0..96c89b846 100644
--- a/nomad/eval_endpoint_test.go
+++ b/nomad/eval_endpoint_test.go
@@ -86,7 +86,12 @@ func TestEvalEndpoint_Ack(t *testing.T) {
 	s1 := testServer(t, nil)
 	defer s1.Shutdown()
 	codec := rpcClient(t, s1)
-	testutil.WaitForLeader(t, s1.RPC)
+
+	testutil.WaitForResult(func() (bool, error) {
+		return s1.evalBroker.Enabled(), nil
+	}, func(err error) {
+		t.Fatalf("should enable eval broker")
+	})
 
 	// Create the register request
 	eval1 := mockEval()
@@ -119,7 +124,12 @@ func TestEvalEndpoint_Nack(t *testing.T) {
 	s1 := testServer(t, nil)
 	defer s1.Shutdown()
 	codec := rpcClient(t, s1)
-	testutil.WaitForLeader(t, s1.RPC)
+
+	testutil.WaitForResult(func() (bool, error) {
+		return s1.evalBroker.Enabled(), nil
+	}, func(err error) {
+		t.Fatalf("should enable eval broker")
+	})
 
 	// Create the register request
 	eval1 := mockEval()