diff --git a/pkg/api/rest/resttest/resttest.go b/pkg/api/rest/resttest/resttest.go index fe8ab7ad870..c04567636fe 100644 --- a/pkg/api/rest/resttest/resttest.go +++ b/pkg/api/rest/resttest/resttest.go @@ -34,6 +34,8 @@ import ( "k8s.io/kubernetes/pkg/runtime" "k8s.io/kubernetes/pkg/types" "k8s.io/kubernetes/pkg/util/wait" + + "golang.org/x/net/context" ) type Tester struct { @@ -1233,6 +1235,7 @@ func (t *Tester) testWatchFields(obj runtime.Object, emitFn EmitFunc, fieldsPass for _, field := range fieldsPass { for _, action := range actions { options := &api.ListOptions{FieldSelector: field.AsSelector(), ResourceVersion: "1"} + ctx = context.WithValue(context.WithValue(ctx, "field", field), "action", action) watcher, err := t.storage.(rest.Watcher).Watch(ctx, options) if err != nil { t.Errorf("unexpected error: %v, %v", err, action) @@ -1257,6 +1260,7 @@ func (t *Tester) testWatchFields(obj runtime.Object, emitFn EmitFunc, fieldsPass for _, field := range fieldsFail { for _, action := range actions { options := &api.ListOptions{FieldSelector: field.AsSelector(), ResourceVersion: "1"} + ctx = context.WithValue(context.WithValue(ctx, "field", field), "action", action) watcher, err := t.storage.(rest.Watcher).Watch(ctx, options) if err != nil { t.Errorf("unexpected error: %v", err) @@ -1277,11 +1281,12 @@ func (t *Tester) testWatchFields(obj runtime.Object, emitFn EmitFunc, fieldsPass } func (t *Tester) testWatchLabels(obj runtime.Object, emitFn EmitFunc, labelsPass, labelsFail []labels.Set, actions []string) { - ctx := t.TestContext() + ctx := t.TestContext().(context.Context) for _, label := range labelsPass { for _, action := range actions { options := &api.ListOptions{LabelSelector: label.AsSelector(), ResourceVersion: "1"} + ctx = context.WithValue(context.WithValue(ctx, "label", label), "action", action) watcher, err := t.storage.(rest.Watcher).Watch(ctx, options) if err != nil { t.Errorf("unexpected error: %v", err) @@ -1305,6 +1310,7 @@ func (t *Tester) testWatchLabels(obj runtime.Object, emitFn EmitFunc, labelsPass for _, label := range labelsFail { for _, action := range actions { options := &api.ListOptions{LabelSelector: label.AsSelector(), ResourceVersion: "1"} + ctx = context.WithValue(context.WithValue(ctx, "label", label), "action", action) watcher, err := t.storage.(rest.Watcher).Watch(ctx, options) if err != nil { t.Errorf("unexpected error: %v", err)