testutil: key leader wait on bootstrap flag

This commit is contained in:
Ryan Uber 2015-05-08 18:16:35 -07:00
parent 952fedf3d6
commit 2fb1bc931e
2 changed files with 4 additions and 10 deletions

View File

@ -29,10 +29,9 @@ func TestMain(t *testing.T) {
defer srv1.Stop()
// Create a secondary server, passing in configuration
// to avoid bootstrapping or waiting for a leader.
// to avoid bootstrapping as we are forming a cluster.
srv2 := testutil.NewTestServerConfig(t, func(c *testutil.TestServerConfig) {
c.Bootstrap = false
c.NoLeaderWait = true
})
defer srv2.Stop()

View File

@ -59,11 +59,6 @@ type TestServerConfig struct {
Bind string `json:"bind_addr,omitempty"`
Addresses *TestAddressConfig `json:"addresses,omitempty"`
Ports *TestPortConfig `json:"ports,omitempty"`
// NoLeaderWait is a special config option used to instruct
// the test harness not to wait for a leader. Useful for
// bootstrapping a multi-node cluster for testing.
NoLeaderWait bool `json:"-"`
}
// ServerConfigCallback is a function interface which can be
@ -207,10 +202,10 @@ func NewTestServerConfig(t *testing.T, cb ServerConfigCallback) *TestServer {
}
// Wait for the server to be ready
if consulConfig.NoLeaderWait {
server.waitForAPI()
} else {
if consulConfig.Bootstrap {
server.waitForLeader()
} else {
server.waitForAPI()
}
return server