Merge pull request #1097 from bcwaldon/remove-code

Remove unnecessary parameter from NewRegistry
This commit is contained in:
Daniel Smith 2014-08-28 17:37:05 -07:00
commit e30d229a34
3 changed files with 42 additions and 44 deletions

View File

@ -67,11 +67,11 @@ func New(c *Config) *Master {
etcdClient := goetcd.NewClient(c.EtcdServers) etcdClient := goetcd.NewClient(c.EtcdServers)
minionRegistry := makeMinionRegistry(c) minionRegistry := makeMinionRegistry(c)
m := &Master{ m := &Master{
podRegistry: etcd.NewRegistry(etcdClient, minionRegistry), podRegistry: etcd.NewRegistry(etcdClient),
controllerRegistry: etcd.NewRegistry(etcdClient, minionRegistry), controllerRegistry: etcd.NewRegistry(etcdClient),
serviceRegistry: etcd.NewRegistry(etcdClient, minionRegistry), serviceRegistry: etcd.NewRegistry(etcdClient),
endpointRegistry: etcd.NewRegistry(etcdClient, minionRegistry), endpointRegistry: etcd.NewRegistry(etcdClient),
bindingRegistry: etcd.NewRegistry(etcdClient, minionRegistry), bindingRegistry: etcd.NewRegistry(etcdClient),
minionRegistry: minionRegistry, minionRegistry: minionRegistry,
client: c.Client, client: c.Client,
} }

View File

@ -23,7 +23,6 @@ import (
"github.com/GoogleCloudPlatform/kubernetes/pkg/apiserver" "github.com/GoogleCloudPlatform/kubernetes/pkg/apiserver"
"github.com/GoogleCloudPlatform/kubernetes/pkg/constraint" "github.com/GoogleCloudPlatform/kubernetes/pkg/constraint"
"github.com/GoogleCloudPlatform/kubernetes/pkg/labels" "github.com/GoogleCloudPlatform/kubernetes/pkg/labels"
"github.com/GoogleCloudPlatform/kubernetes/pkg/registry/minion"
"github.com/GoogleCloudPlatform/kubernetes/pkg/tools" "github.com/GoogleCloudPlatform/kubernetes/pkg/tools"
"github.com/GoogleCloudPlatform/kubernetes/pkg/watch" "github.com/GoogleCloudPlatform/kubernetes/pkg/watch"
@ -41,7 +40,7 @@ type Registry struct {
} }
// NewRegistry creates an etcd registry. // NewRegistry creates an etcd registry.
func NewRegistry(client tools.EtcdClient, machines minion.Registry) *Registry { func NewRegistry(client tools.EtcdClient) *Registry {
registry := &Registry{ registry := &Registry{
EtcdHelper: tools.EtcdHelper{ EtcdHelper: tools.EtcdHelper{
client, client,

View File

@ -23,15 +23,14 @@ import (
"github.com/GoogleCloudPlatform/kubernetes/pkg/api" "github.com/GoogleCloudPlatform/kubernetes/pkg/api"
"github.com/GoogleCloudPlatform/kubernetes/pkg/apiserver" "github.com/GoogleCloudPlatform/kubernetes/pkg/apiserver"
"github.com/GoogleCloudPlatform/kubernetes/pkg/labels" "github.com/GoogleCloudPlatform/kubernetes/pkg/labels"
"github.com/GoogleCloudPlatform/kubernetes/pkg/registry/minion"
"github.com/GoogleCloudPlatform/kubernetes/pkg/registry/registrytest" "github.com/GoogleCloudPlatform/kubernetes/pkg/registry/registrytest"
"github.com/GoogleCloudPlatform/kubernetes/pkg/tools" "github.com/GoogleCloudPlatform/kubernetes/pkg/tools"
"github.com/coreos/go-etcd/etcd" "github.com/coreos/go-etcd/etcd"
) )
func NewTestEtcdRegistry(client tools.EtcdClient, machines []string) *Registry { func NewTestEtcdRegistry(client tools.EtcdClient) *Registry {
registry := NewRegistry(client, minion.NewRegistry(machines)) registry := NewRegistry(client)
registry.manifestFactory = &BasicManifestFactory{ registry.manifestFactory = &BasicManifestFactory{
serviceRegistry: &registrytest.ServiceRegistry{}, serviceRegistry: &registrytest.ServiceRegistry{},
} }
@ -41,7 +40,7 @@ func NewTestEtcdRegistry(client tools.EtcdClient, machines []string) *Registry {
func TestEtcdGetPod(t *testing.T) { func TestEtcdGetPod(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
fakeClient.Set("/registry/pods/foo", api.EncodeOrDie(api.Pod{JSONBase: api.JSONBase{ID: "foo"}}), 0) fakeClient.Set("/registry/pods/foo", api.EncodeOrDie(api.Pod{JSONBase: api.JSONBase{ID: "foo"}}), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
pod, err := registry.GetPod("foo") pod, err := registry.GetPod("foo")
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -60,7 +59,7 @@ func TestEtcdGetPodNotFound(t *testing.T) {
}, },
E: tools.EtcdErrorNotFound, E: tools.EtcdErrorNotFound,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
_, err := registry.GetPod("foo") _, err := registry.GetPod("foo")
if err == nil { if err == nil {
t.Errorf("Unexpected non-error.") t.Errorf("Unexpected non-error.")
@ -77,7 +76,7 @@ func TestEtcdCreatePod(t *testing.T) {
E: tools.EtcdErrorNotFound, E: tools.EtcdErrorNotFound,
} }
fakeClient.Set("/registry/hosts/machine/kubelet", api.EncodeOrDie(&api.ContainerManifestList{}), 0) fakeClient.Set("/registry/hosts/machine/kubelet", api.EncodeOrDie(&api.ContainerManifestList{}), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreatePod(api.Pod{ err := registry.CreatePod(api.Pod{
JSONBase: api.JSONBase{ JSONBase: api.JSONBase{
ID: "foo", ID: "foo",
@ -137,7 +136,7 @@ func TestEtcdCreatePodAlreadyExisting(t *testing.T) {
}, },
E: nil, E: nil,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreatePod(api.Pod{ err := registry.CreatePod(api.Pod{
JSONBase: api.JSONBase{ JSONBase: api.JSONBase{
ID: "foo", ID: "foo",
@ -163,7 +162,7 @@ func TestEtcdCreatePodWithContainersError(t *testing.T) {
}, },
E: tools.EtcdErrorValueRequired, E: tools.EtcdErrorValueRequired,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreatePod(api.Pod{ err := registry.CreatePod(api.Pod{
JSONBase: api.JSONBase{ JSONBase: api.JSONBase{
ID: "foo", ID: "foo",
@ -203,7 +202,7 @@ func TestEtcdCreatePodWithContainersNotFound(t *testing.T) {
}, },
E: tools.EtcdErrorNotFound, E: tools.EtcdErrorNotFound,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreatePod(api.Pod{ err := registry.CreatePod(api.Pod{
JSONBase: api.JSONBase{ JSONBase: api.JSONBase{
ID: "foo", ID: "foo",
@ -268,7 +267,7 @@ func TestEtcdCreatePodWithExistingContainers(t *testing.T) {
{ID: "bar"}, {ID: "bar"},
}, },
}), 0) }), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreatePod(api.Pod{ err := registry.CreatePod(api.Pod{
JSONBase: api.JSONBase{ JSONBase: api.JSONBase{
ID: "foo", ID: "foo",
@ -333,7 +332,7 @@ func TestEtcdDeletePod(t *testing.T) {
{ID: "foo"}, {ID: "foo"},
}, },
}), 0) }), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.DeletePod("foo") err := registry.DeletePod("foo")
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -370,7 +369,7 @@ func TestEtcdDeletePodMultipleContainers(t *testing.T) {
{ID: "bar"}, {ID: "bar"},
}, },
}), 0) }), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.DeletePod("foo") err := registry.DeletePod("foo")
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -407,7 +406,7 @@ func TestEtcdEmptyListPods(t *testing.T) {
}, },
E: nil, E: nil,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
pods, err := registry.ListPods(labels.Everything()) pods, err := registry.ListPods(labels.Everything())
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -425,7 +424,7 @@ func TestEtcdListPodsNotFound(t *testing.T) {
R: &etcd.Response{}, R: &etcd.Response{},
E: tools.EtcdErrorNotFound, E: tools.EtcdErrorNotFound,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
pods, err := registry.ListPods(labels.Everything()) pods, err := registry.ListPods(labels.Everything())
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -460,7 +459,7 @@ func TestEtcdListPods(t *testing.T) {
}, },
E: nil, E: nil,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
pods, err := registry.ListPods(labels.Everything()) pods, err := registry.ListPods(labels.Everything())
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -482,7 +481,7 @@ func TestEtcdListControllersNotFound(t *testing.T) {
R: &etcd.Response{}, R: &etcd.Response{},
E: tools.EtcdErrorNotFound, E: tools.EtcdErrorNotFound,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
controllers, err := registry.ListControllers() controllers, err := registry.ListControllers()
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -500,7 +499,7 @@ func TestEtcdListServicesNotFound(t *testing.T) {
R: &etcd.Response{}, R: &etcd.Response{},
E: tools.EtcdErrorNotFound, E: tools.EtcdErrorNotFound,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
services, err := registry.ListServices() services, err := registry.ListServices()
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -529,7 +528,7 @@ func TestEtcdListControllers(t *testing.T) {
}, },
E: nil, E: nil,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
controllers, err := registry.ListControllers() controllers, err := registry.ListControllers()
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -543,7 +542,7 @@ func TestEtcdListControllers(t *testing.T) {
func TestEtcdGetController(t *testing.T) { func TestEtcdGetController(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
fakeClient.Set("/registry/controllers/foo", api.EncodeOrDie(api.ReplicationController{JSONBase: api.JSONBase{ID: "foo"}}), 0) fakeClient.Set("/registry/controllers/foo", api.EncodeOrDie(api.ReplicationController{JSONBase: api.JSONBase{ID: "foo"}}), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
ctrl, err := registry.GetController("foo") ctrl, err := registry.GetController("foo")
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -562,7 +561,7 @@ func TestEtcdGetControllerNotFound(t *testing.T) {
}, },
E: tools.EtcdErrorNotFound, E: tools.EtcdErrorNotFound,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
ctrl, err := registry.GetController("foo") ctrl, err := registry.GetController("foo")
if ctrl != nil { if ctrl != nil {
t.Errorf("Unexpected non-nil controller: %#v", ctrl) t.Errorf("Unexpected non-nil controller: %#v", ctrl)
@ -574,7 +573,7 @@ func TestEtcdGetControllerNotFound(t *testing.T) {
func TestEtcdDeleteController(t *testing.T) { func TestEtcdDeleteController(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.DeleteController("foo") err := registry.DeleteController("foo")
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -591,7 +590,7 @@ func TestEtcdDeleteController(t *testing.T) {
func TestEtcdCreateController(t *testing.T) { func TestEtcdCreateController(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreateController(api.ReplicationController{ err := registry.CreateController(api.ReplicationController{
JSONBase: api.JSONBase{ JSONBase: api.JSONBase{
ID: "foo", ID: "foo",
@ -620,7 +619,7 @@ func TestEtcdCreateControllerAlreadyExisting(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
fakeClient.Set("/registry/controllers/foo", api.EncodeOrDie(api.ReplicationController{JSONBase: api.JSONBase{ID: "foo"}}), 0) fakeClient.Set("/registry/controllers/foo", api.EncodeOrDie(api.ReplicationController{JSONBase: api.JSONBase{ID: "foo"}}), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreateController(api.ReplicationController{ err := registry.CreateController(api.ReplicationController{
JSONBase: api.JSONBase{ JSONBase: api.JSONBase{
ID: "foo", ID: "foo",
@ -636,7 +635,7 @@ func TestEtcdUpdateController(t *testing.T) {
fakeClient.TestIndex = true fakeClient.TestIndex = true
resp, _ := fakeClient.Set("/registry/controllers/foo", api.EncodeOrDie(api.ReplicationController{JSONBase: api.JSONBase{ID: "foo"}}), 0) resp, _ := fakeClient.Set("/registry/controllers/foo", api.EncodeOrDie(api.ReplicationController{JSONBase: api.JSONBase{ID: "foo"}}), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.UpdateController(api.ReplicationController{ err := registry.UpdateController(api.ReplicationController{
JSONBase: api.JSONBase{ID: "foo", ResourceVersion: resp.Node.ModifiedIndex}, JSONBase: api.JSONBase{ID: "foo", ResourceVersion: resp.Node.ModifiedIndex},
DesiredState: api.ReplicationControllerState{ DesiredState: api.ReplicationControllerState{
@ -671,7 +670,7 @@ func TestEtcdListServices(t *testing.T) {
}, },
E: nil, E: nil,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
services, err := registry.ListServices() services, err := registry.ListServices()
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -684,7 +683,7 @@ func TestEtcdListServices(t *testing.T) {
func TestEtcdCreateService(t *testing.T) { func TestEtcdCreateService(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreateService(api.Service{ err := registry.CreateService(api.Service{
JSONBase: api.JSONBase{ID: "foo"}, JSONBase: api.JSONBase{ID: "foo"},
}) })
@ -711,7 +710,7 @@ func TestEtcdCreateService(t *testing.T) {
func TestEtcdCreateServiceAlreadyExisting(t *testing.T) { func TestEtcdCreateServiceAlreadyExisting(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
fakeClient.Set("/registry/services/specs/foo", api.EncodeOrDie(api.Service{JSONBase: api.JSONBase{ID: "foo"}}), 0) fakeClient.Set("/registry/services/specs/foo", api.EncodeOrDie(api.Service{JSONBase: api.JSONBase{ID: "foo"}}), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.CreateService(api.Service{ err := registry.CreateService(api.Service{
JSONBase: api.JSONBase{ID: "foo"}, JSONBase: api.JSONBase{ID: "foo"},
}) })
@ -723,7 +722,7 @@ func TestEtcdCreateServiceAlreadyExisting(t *testing.T) {
func TestEtcdGetService(t *testing.T) { func TestEtcdGetService(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
fakeClient.Set("/registry/services/specs/foo", api.EncodeOrDie(api.Service{JSONBase: api.JSONBase{ID: "foo"}}), 0) fakeClient.Set("/registry/services/specs/foo", api.EncodeOrDie(api.Service{JSONBase: api.JSONBase{ID: "foo"}}), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
service, err := registry.GetService("foo") service, err := registry.GetService("foo")
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -742,7 +741,7 @@ func TestEtcdGetServiceNotFound(t *testing.T) {
}, },
E: tools.EtcdErrorNotFound, E: tools.EtcdErrorNotFound,
} }
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
_, err := registry.GetService("foo") _, err := registry.GetService("foo")
if err == nil { if err == nil {
t.Errorf("Unexpected non-error.") t.Errorf("Unexpected non-error.")
@ -751,7 +750,7 @@ func TestEtcdGetServiceNotFound(t *testing.T) {
func TestEtcdDeleteService(t *testing.T) { func TestEtcdDeleteService(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
err := registry.DeleteService("foo") err := registry.DeleteService("foo")
if err != nil { if err != nil {
t.Errorf("unexpected error: %v", err) t.Errorf("unexpected error: %v", err)
@ -775,7 +774,7 @@ func TestEtcdUpdateService(t *testing.T) {
fakeClient.TestIndex = true fakeClient.TestIndex = true
resp, _ := fakeClient.Set("/registry/services/specs/foo", api.EncodeOrDie(api.Service{JSONBase: api.JSONBase{ID: "foo"}}), 0) resp, _ := fakeClient.Set("/registry/services/specs/foo", api.EncodeOrDie(api.Service{JSONBase: api.JSONBase{ID: "foo"}}), 0)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
testService := api.Service{ testService := api.Service{
JSONBase: api.JSONBase{ID: "foo", ResourceVersion: resp.Node.ModifiedIndex}, JSONBase: api.JSONBase{ID: "foo", ResourceVersion: resp.Node.ModifiedIndex},
Labels: map[string]string{ Labels: map[string]string{
@ -805,7 +804,7 @@ func TestEtcdUpdateService(t *testing.T) {
func TestEtcdGetEndpoints(t *testing.T) { func TestEtcdGetEndpoints(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
endpoints := &api.Endpoints{ endpoints := &api.Endpoints{
JSONBase: api.JSONBase{ID: "foo"}, JSONBase: api.JSONBase{ID: "foo"},
Endpoints: []string{"127.0.0.1:34855"}, Endpoints: []string{"127.0.0.1:34855"},
@ -826,7 +825,7 @@ func TestEtcdGetEndpoints(t *testing.T) {
func TestEtcdUpdateEndpoints(t *testing.T) { func TestEtcdUpdateEndpoints(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
fakeClient.TestIndex = true fakeClient.TestIndex = true
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
endpoints := api.Endpoints{ endpoints := api.Endpoints{
JSONBase: api.JSONBase{ID: "foo"}, JSONBase: api.JSONBase{ID: "foo"},
Endpoints: []string{"baz", "bar"}, Endpoints: []string{"baz", "bar"},
@ -852,7 +851,7 @@ func TestEtcdUpdateEndpoints(t *testing.T) {
func TestEtcdWatchServices(t *testing.T) { func TestEtcdWatchServices(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
watching, err := registry.WatchServices( watching, err := registry.WatchServices(
labels.Everything(), labels.Everything(),
labels.SelectorFromSet(labels.Set{"ID": "foo"}), labels.SelectorFromSet(labels.Set{"ID": "foo"}),
@ -879,7 +878,7 @@ func TestEtcdWatchServices(t *testing.T) {
func TestEtcdWatchServicesBadSelector(t *testing.T) { func TestEtcdWatchServicesBadSelector(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
_, err := registry.WatchServices( _, err := registry.WatchServices(
labels.Everything(), labels.Everything(),
labels.SelectorFromSet(labels.Set{"Field.Selector": "foo"}), labels.SelectorFromSet(labels.Set{"Field.Selector": "foo"}),
@ -901,7 +900,7 @@ func TestEtcdWatchServicesBadSelector(t *testing.T) {
func TestEtcdWatchEndpoints(t *testing.T) { func TestEtcdWatchEndpoints(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
watching, err := registry.WatchEndpoints( watching, err := registry.WatchEndpoints(
labels.Everything(), labels.Everything(),
labels.SelectorFromSet(labels.Set{"ID": "foo"}), labels.SelectorFromSet(labels.Set{"ID": "foo"}),
@ -928,7 +927,7 @@ func TestEtcdWatchEndpoints(t *testing.T) {
func TestEtcdWatchEndpointsBadSelector(t *testing.T) { func TestEtcdWatchEndpointsBadSelector(t *testing.T) {
fakeClient := tools.NewFakeEtcdClient(t) fakeClient := tools.NewFakeEtcdClient(t)
registry := NewTestEtcdRegistry(fakeClient, []string{"machine"}) registry := NewTestEtcdRegistry(fakeClient)
_, err := registry.WatchEndpoints( _, err := registry.WatchEndpoints(
labels.Everything(), labels.Everything(),
labels.SelectorFromSet(labels.Set{"Field.Selector": "foo"}), labels.SelectorFromSet(labels.Set{"Field.Selector": "foo"}),