mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
tests(tests/functional): add tests for Discovery
This tests a variety of failure cases for the Discovery service including: - Initial leader failures - Discovery service failures - Positive tests for discovery working flawlessly
This commit is contained in:
Brandon Philips
parent
ff6090836c
commit
2822b9c579
245
tests/functional/discovery_test.go
Normal file
245
tests/functional/discovery_test.go
Normal file
@ -0,0 +1,245 @@
|
||||
package test
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"net/url"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/coreos/etcd/third_party/github.com/stretchr/testify/assert"
|
||||
|
||||
etcdtest "github.com/coreos/etcd/tests"
|
||||
"github.com/coreos/etcd/server"
|
||||
goetcd "github.com/coreos/etcd/third_party/github.com/coreos/go-etcd/etcd"
|
||||
)
|
||||
|
||||
type garbageHandler struct {
|
||||
t *testing.T
|
||||
success bool
|
||||
}
|
||||
|
||||
func (g *garbageHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
fmt.Fprintln(w, "Hello, client")
|
||||
println("HI")
|
||||
if r.URL.String() != "/v2/keys/_etcd/registry/1/node1" {
|
||||
g.t.Fatalf("Unexpected web request")
|
||||
}
|
||||
g.success = true
|
||||
}
|
||||
|
||||
// TestDiscoveryDownNoBackupPeers ensures that etcd stops if it is started with a
|
||||
// bad discovery URL and no backups.
|
||||
func TestDiscoveryDownNoBackupPeers(t *testing.T) {
|
||||
g := garbageHandler{t: t}
|
||||
ts := httptest.NewServer(&g)
|
||||
defer ts.Close()
|
||||
|
||||
discover := ts.URL + "/v2/keys/_etcd/registry/1"
|
||||
proc, err := startServer([]string{"-discovery", discover})
|
||||
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
defer stopServer(proc)
|
||||
|
||||
client := http.Client{}
|
||||
err = assertServerNotUp(client, "http")
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
|
||||
if !g.success {
|
||||
t.Fatal("Discovery server never called")
|
||||
}
|
||||
}
|
||||
|
||||
// TestDiscoveryDownWithBackupPeers ensures that etcd runs if it is started with a
|
||||
// bad discovery URL and a peer list.
|
||||
func TestDiscoveryDownWithBackupPeers(t *testing.T) {
|
||||
etcdtest.RunServer(func(s *server.Server) {
|
||||
g := garbageHandler{t: t}
|
||||
ts := httptest.NewServer(&g)
|
||||
defer ts.Close()
|
||||
|
||||
discover := ts.URL + "/v2/keys/_etcd/registry/1"
|
||||
u, ok := s.PeerURL("ETCDTEST")
|
||||
if !ok {
|
||||
t.Fatalf("Couldn't find the URL")
|
||||
}
|
||||
proc, err := startServer([]string{"-discovery", discover, "-peers", u})
|
||||
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
defer stopServer(proc)
|
||||
|
||||
client := http.Client{}
|
||||
err = assertServerFunctional(client, "http")
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
|
||||
if !g.success {
|
||||
t.Fatal("Discovery server never called")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// TestDiscoveryFirstPeer ensures that etcd starts as the leader if it
|
||||
// registers as the first peer.
|
||||
func TestDiscoveryFirstPeer(t *testing.T) {
|
||||
etcdtest.RunServer(func(s *server.Server) {
|
||||
v := url.Values{}
|
||||
v.Set("value", "init")
|
||||
resp, err := etcdtest.PutForm(fmt.Sprintf("%s%s", s.URL(), "/v2/keys/_etcd/registry/2/_state"), v)
|
||||
assert.Equal(t, resp.StatusCode, http.StatusCreated)
|
||||
|
||||
proc, err := startServer([]string{"-discovery", s.URL() + "/v2/keys/_etcd/registry/2"})
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
defer stopServer(proc)
|
||||
|
||||
client := http.Client{}
|
||||
err = assertServerFunctional(client, "http")
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// TestDiscoverySecondPeerFirstDown ensures that etcd stops if it is started with a
|
||||
// correct discovery URL but no active machines are found.
|
||||
func TestDiscoverySecondPeerFirstDown(t *testing.T) {
|
||||
etcdtest.RunServer(func(s *server.Server) {
|
||||
v := url.Values{}
|
||||
v.Set("value", "started")
|
||||
resp, err := etcdtest.PutForm(fmt.Sprintf("%s%s", s.URL(), "/v2/keys/_etcd/registry/2/_state"), v)
|
||||
assert.Equal(t, resp.StatusCode, http.StatusCreated)
|
||||
|
||||
proc, err := startServer([]string{"-discovery", s.URL() + "/v2/keys/_etcd/registry/2"})
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
defer stopServer(proc)
|
||||
|
||||
client := http.Client{}
|
||||
err = assertServerNotUp(client, "http")
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// TestDiscoverySecondPeerFirstNoResponse ensures that if the first etcd
|
||||
// machine stops after heartbeating that the second machine fails too.
|
||||
func TestDiscoverySecondPeerFirstNoResponse(t *testing.T) {
|
||||
etcdtest.RunServer(func(s *server.Server) {
|
||||
v := url.Values{}
|
||||
v.Set("value", "started")
|
||||
resp, err := etcdtest.PutForm(fmt.Sprintf("%s%s", s.URL(), "/v2/keys/_etcd/registry/2/_state"), v)
|
||||
assert.Equal(t, resp.StatusCode, http.StatusCreated)
|
||||
|
||||
v = url.Values{}
|
||||
v.Set("value", "http://127.0.0.1:49151")
|
||||
resp, err = etcdtest.PutForm(fmt.Sprintf("%s%s", s.URL(), "/v2/keys/_etcd/registry/2/ETCDTEST"), v)
|
||||
assert.Equal(t, resp.StatusCode, http.StatusCreated)
|
||||
|
||||
proc, err := startServer([]string{"-discovery", s.URL() + "/v2/keys/_etcd/registry/2"})
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
defer stopServer(proc)
|
||||
|
||||
// TODO(bp): etcd will take 30 seconds to shutdown, figure this
|
||||
// out instead
|
||||
time.Sleep(35 * time.Second)
|
||||
|
||||
client := http.Client{}
|
||||
_, err = client.Get("/")
|
||||
if err != nil && strings.Contains(err.Error(), "connection reset by peer") {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// TestDiscoverySecondPeerUp ensures that a second peer joining a discovery
|
||||
// cluster works.
|
||||
func TestDiscoverySecondPeerUp(t *testing.T) {
|
||||
etcdtest.RunServer(func(s *server.Server) {
|
||||
v := url.Values{}
|
||||
v.Set("value", "started")
|
||||
resp, err := etcdtest.PutForm(fmt.Sprintf("%s%s", s.URL(), "/v2/keys/_etcd/registry/3/_state"), v)
|
||||
assert.Equal(t, resp.StatusCode, http.StatusCreated)
|
||||
|
||||
u, ok := s.PeerURL("ETCDTEST")
|
||||
if !ok {
|
||||
t.Fatalf("Couldn't find the URL")
|
||||
}
|
||||
|
||||
wc := goetcd.NewClient([]string{s.URL()})
|
||||
_, err = wc.Set("test", "0", 0)
|
||||
|
||||
if err != nil {
|
||||
t.Fatalf("Couldn't set a test key on the leader %v", err)
|
||||
}
|
||||
|
||||
receiver := make(chan *goetcd.Response)
|
||||
stop := make(chan bool)
|
||||
|
||||
go wc.Watch("_etcd/registry/3/node1", 0, false, receiver, stop)
|
||||
|
||||
v = url.Values{}
|
||||
v.Set("value", u)
|
||||
resp, err = etcdtest.PutForm(fmt.Sprintf("%s%s", s.URL(), "/v2/keys/_etcd/registry/3/ETCDTEST"), v)
|
||||
assert.Equal(t, resp.StatusCode, http.StatusCreated)
|
||||
|
||||
proc, err := startServer([]string{"-discovery", s.URL() + "/v2/keys/_etcd/registry/3"})
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
defer stopServer(proc)
|
||||
|
||||
// Test to ensure the machine registered iteslf
|
||||
watchResp := <-receiver
|
||||
if watchResp.Node.Value != "http://127.0.0.1:7001" {
|
||||
t.Fatalf("Second peer didn't register! %s", watchResp.Node.Value)
|
||||
}
|
||||
|
||||
// TODO(bp): need to have a better way of knowing a machine is up
|
||||
time.Sleep(1 * time.Second)
|
||||
|
||||
etcdc := goetcd.NewClient(nil)
|
||||
_, err = etcdc.Set("foobar", "baz", 0)
|
||||
if err != nil {
|
||||
t.Fatal(err.Error())
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func assertServerNotUp(client http.Client, scheme string) error {
|
||||
path := fmt.Sprintf("%s://127.0.0.1:4001/v2/keys/foo", scheme)
|
||||
fields := url.Values(map[string][]string{"value": []string{"bar"}})
|
||||
|
||||
for i := 0; i < 10; i++ {
|
||||
time.Sleep(1 * time.Second)
|
||||
|
||||
_, err := client.PostForm(path, fields)
|
||||
if err == nil {
|
||||
return errors.New("Expected error during POST, got nil")
|
||||
} else {
|
||||
errString := err.Error()
|
||||
if strings.Contains(errString, "connection refused") {
|
||||
return nil
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user