diff options
Diffstat (limited to 'middleware/etcd/setup_test.go')
-rw-r--r-- | middleware/etcd/setup_test.go | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/middleware/etcd/setup_test.go b/middleware/etcd/setup_test.go index d7e6e344b..6e1eabdcd 100644 --- a/middleware/etcd/setup_test.go +++ b/middleware/etcd/setup_test.go @@ -28,17 +28,16 @@ func init() { func newEtcdMiddleware() *Etcd { ctxt, _ = context.WithTimeout(context.Background(), etcdTimeout) - etcdCfg := etcdc.Config{ - Endpoints: []string{"http://localhost:2379"}, - } - cli, _ := etcdc.New(etcdCfg) + endpoints := []string{"http://localhost:2379"} + client, _ := newEtcdClient(endpoints, "", "", "") + return &Etcd{ Proxy: proxy.New([]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: etcdc.NewKeysAPI(cli), + Client: client, } } |