aboutsummaryrefslogtreecommitdiff
path: root/middleware/etcd/setup_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'middleware/etcd/setup_test.go')
-rw-r--r--middleware/etcd/setup_test.go85
1 files changed, 0 insertions, 85 deletions
diff --git a/middleware/etcd/setup_test.go b/middleware/etcd/setup_test.go
index d29f34bc5..833e2ba4c 100644
--- a/middleware/etcd/setup_test.go
+++ b/middleware/etcd/setup_test.go
@@ -1,95 +1,12 @@
-// +build etcd
-
package etcd
import (
- "encoding/json"
- "sort"
"strings"
"testing"
- "time"
-
- "github.com/coredns/coredns/middleware/etcd/msg"
- "github.com/coredns/coredns/middleware/pkg/dnsrecorder"
- "github.com/coredns/coredns/middleware/pkg/singleflight"
- "github.com/coredns/coredns/middleware/pkg/tls"
- "github.com/coredns/coredns/middleware/proxy"
- "github.com/coredns/coredns/middleware/test"
- etcdc "github.com/coreos/etcd/client"
"github.com/mholt/caddy"
- "golang.org/x/net/context"
)
-func init() {
- ctxt, _ = context.WithTimeout(context.Background(), etcdTimeout)
-}
-
-func newEtcdMiddleware() *Etcd {
- ctxt, _ = context.WithTimeout(context.Background(), etcdTimeout)
-
- endpoints := []string{"http://localhost:2379"}
- tlsc, _ := tls.NewTLSConfigFromArgs()
- client, _ := newEtcdClient(endpoints, tlsc)
-
- return &Etcd{
- Proxy: proxy.NewLookup([]string{"8.8.8.8:53"}),
- PathPrefix: "skydns",
- Ctx: context.Background(),
- Inflight: &singleflight.Group{},
- Zones: []string{"skydns.test.", "skydns_extra.test.", "in-addr.arpa."},
- Client: client,
- }
-}
-
-func set(t *testing.T, e *Etcd, k string, ttl time.Duration, m *msg.Service) {
- b, err := json.Marshal(m)
- if err != nil {
- t.Fatal(err)
- }
- path, _ := msg.PathWithWildcard(k, e.PathPrefix)
- e.Client.Set(ctxt, path, string(b), &etcdc.SetOptions{TTL: ttl})
-}
-
-func delete(t *testing.T, e *Etcd, k string) {
- path, _ := msg.PathWithWildcard(k, e.PathPrefix)
- e.Client.Delete(ctxt, path, &etcdc.DeleteOptions{Recursive: false})
-}
-
-func TestLookup(t *testing.T) {
- etc := newEtcdMiddleware()
- for _, serv := range services {
- set(t, etc, serv.Key, 0, serv)
- defer delete(t, etc, serv.Key)
- }
-
- for _, tc := range dnsTestCases {
- m := tc.Msg()
-
- rec := dnsrecorder.New(&test.ResponseWriter{})
- etc.ServeDNS(ctxt, rec, m)
-
- resp := rec.Msg
- sort.Sort(test.RRSet(resp.Answer))
- sort.Sort(test.RRSet(resp.Ns))
- sort.Sort(test.RRSet(resp.Extra))
-
- if !test.Header(t, tc, resp) {
- t.Logf("%v\n", resp)
- continue
- }
- if !test.Section(t, tc, test.Answer, resp.Answer) {
- t.Logf("%v\n", resp)
- }
- if !test.Section(t, tc, test.Ns, resp.Ns) {
- t.Logf("%v\n", resp)
- }
- if !test.Section(t, tc, test.Extra, resp.Extra) {
- t.Logf("%v\n", resp)
- }
- }
-}
-
func TestSetupEtcd(t *testing.T) {
tests := []struct {
input string
@@ -145,5 +62,3 @@ func TestSetupEtcd(t *testing.T) {
}
}
}
-
-var ctxt context.Context