nomad: testing status endpoint

testutil/wait.go

@@ -1,6 +1,9 @@
 package testutil
 
-import "time"
+import (
+	"testing"
+	"time"
+)
 
 type testFn func() (bool, error)
 type errorFn func(error)
@@ -22,3 +25,16 @@ func WaitForResult(test testFn, error errorFn) {
 		}
 	}
 }
+
+type rpcFn func(string, interface{}, interface{}) error
+
+func WaitForLeader(t *testing.T, rpc rpcFn) {
+	WaitForResult(func() (bool, error) {
+		args := struct{}{}
+		var leader string
+		err := rpc("Status.Leader", args, &leader)
+		return leader != "", err
+	}, func(err error) {
+		t.Fatalf("failed to find leader: %v", err)
+	})
+}