From 5f4281f98fd327f8705858a184c2274837143ac1 Mon Sep 17 00:00:00 2001
From: Chavez <matthew@el-chavez.me>
Date: Fri, 5 Dec 2014 10:50:23 -0800
Subject: [PATCH] consul: Server leave test fix

---
 consul/server_test.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/consul/server_test.go b/consul/server_test.go
index 8c61d2514..59c220b19 100644
--- a/consul/server_test.go
+++ b/consul/server_test.go
@@ -256,10 +256,12 @@ func TestServer_Leave(t *testing.T) {
 	}
 
 	// Should lose a peer
-	p1, _ = s1.raftPeers.Peers()
-	if len(p1) != 1 {
+	testutil.WaitForResult(func() (bool, error) {
+		p1, _ = s1.raftPeers.Peers()
+		return len(p1) == 1, nil
+	}, func(err error) {
 		t.Fatalf("should have 1 peer: %v", p1)
-	}
+	})
 }
 
 func TestServer_RPC(t *testing.T) {