mirror of https://github.com/hashicorp/consul
Ryan Uber
10 years ago
12 changed files with 228 additions and 203 deletions
@ -0,0 +1,156 @@
|
||||
package testutil |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"fmt" |
||||
"io/ioutil" |
||||
"net/http" |
||||
"os" |
||||
"os/exec" |
||||
"sync/atomic" |
||||
"testing" |
||||
) |
||||
|
||||
var offset uint64 |
||||
|
||||
type TestPortConfig struct { |
||||
DNS int `json:"dns,omitempty"` |
||||
HTTP int `json:"http,omitempty"` |
||||
RPC int `json:"rpc,omitempty"` |
||||
SerfLan int `json:"serf_lan,omitempty"` |
||||
SerfWan int `json:"serf_wan,omitempty"` |
||||
Server int `json:"server,omitempty"` |
||||
} |
||||
|
||||
type TestAddressConfig struct { |
||||
HTTP string `json:"http,omitempty"` |
||||
} |
||||
|
||||
type TestServerConfig struct { |
||||
Bootstrap bool `json:"bootstrap,omitempty"` |
||||
Server bool `json:"server,omitempty"` |
||||
DataDir string `json:"data_dir,omitempty"` |
||||
LogLevel string `json:"log_level,omitempty"` |
||||
Addresses *TestAddressConfig `json:"addresses,omitempty"` |
||||
Ports *TestPortConfig `json:"ports,omitempty"` |
||||
} |
||||
|
||||
type ServerConfigCallback func(c *TestServerConfig) |
||||
|
||||
func defaultServerConfig() *TestServerConfig { |
||||
idx := int(atomic.AddUint64(&offset, 1)) |
||||
|
||||
return &TestServerConfig{ |
||||
Bootstrap: true, |
||||
Server: true, |
||||
LogLevel: "debug", |
||||
Ports: &TestPortConfig{ |
||||
DNS: 19000 + idx, |
||||
HTTP: 18800 + idx, |
||||
RPC: 18600 + idx, |
||||
SerfLan: 18200 + idx, |
||||
SerfWan: 18400 + idx, |
||||
Server: 18000 + idx, |
||||
}, |
||||
} |
||||
} |
||||
|
||||
type TestServer struct { |
||||
pid int |
||||
dataDir string |
||||
config *TestServerConfig |
||||
} |
||||
|
||||
func NewTestServer(t *testing.T) *TestServer { |
||||
return NewTestServerConfig(t, nil) |
||||
} |
||||
|
||||
func NewTestServerConfig(t *testing.T, cb ServerConfigCallback) *TestServer { |
||||
if path, err := exec.LookPath("consul"); err != nil || path == "" { |
||||
t.Log("consul not found on $PATH, skipping") |
||||
t.SkipNow() |
||||
} |
||||
|
||||
dataDir, err := ioutil.TempDir("", "consul") |
||||
if err != nil { |
||||
t.Fatalf("err: %s", err) |
||||
} |
||||
|
||||
configFile, err := ioutil.TempFile("", "consul") |
||||
if err != nil { |
||||
t.Fatalf("err: %s", err) |
||||
} |
||||
defer os.Remove(configFile.Name()) |
||||
|
||||
consulConfig := defaultServerConfig() |
||||
consulConfig.DataDir = dataDir |
||||
|
||||
if cb != nil { |
||||
cb(consulConfig) |
||||
} |
||||
|
||||
configContent, err := json.Marshal(consulConfig) |
||||
if err != nil { |
||||
t.Fatalf("err: %s", err) |
||||
} |
||||
|
||||
if _, err := configFile.Write(configContent); err != nil { |
||||
t.Fatalf("err: %s", err) |
||||
} |
||||
configFile.Close() |
||||
|
||||
// Start the server
|
||||
cmd := exec.Command("consul", "agent", "-config-file", configFile.Name()) |
||||
cmd.Stdout = os.Stdout |
||||
cmd.Stderr = os.Stderr |
||||
if err := cmd.Start(); err != nil { |
||||
t.Fatalf("err: %s", err) |
||||
} |
||||
|
||||
server := &TestServer{ |
||||
config: consulConfig, |
||||
pid: cmd.Process.Pid, |
||||
dataDir: dataDir, |
||||
} |
||||
|
||||
if err := server.waitForLeader(); err != nil { |
||||
t.Fatalf("err: %s", err) |
||||
} |
||||
return server |
||||
} |
||||
|
||||
func (s *TestServer) Stop() { |
||||
defer os.RemoveAll(s.dataDir) |
||||
|
||||
cmd := exec.Command("kill", "-9", fmt.Sprintf("%d", s.pid)) |
||||
if err := cmd.Run(); err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
func (s *TestServer) waitForLeader() error { |
||||
url := fmt.Sprintf("http://127.0.0.1:%d/v1/catalog/nodes", s.config.Ports.HTTP) |
||||
|
||||
WaitForResult(func() (bool, error) { |
||||
resp, err := http.Get(url) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
resp.Body.Close() |
||||
|
||||
// Ensure we have a leader and a node registeration
|
||||
if leader := resp.Header.Get("X-Consul-KnownLeader"); leader != "true" { |
||||
fmt.Println(leader) |
||||
return false, fmt.Errorf("Consul leader status: %#v", leader) |
||||
} |
||||
if resp.Header.Get("X-Consul-Index") == "0" { |
||||
return false, fmt.Errorf("Consul index is 0") |
||||
} |
||||
|
||||
return true, nil |
||||
}, func(err error) { |
||||
panic(err) |
||||
}) |
||||
|
||||
return nil |
||||
} |
||||
Loading…
Reference in new issue