/* Copyright 2014 The Kubernetes Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ package resource import ( "bytes" "fmt" "io" "io/ioutil" "net/http" "net/http/httptest" "os" "path/filepath" "reflect" "strings" "testing" "github.com/ghodss/yaml" "k8s.io/apimachinery/pkg/api/meta" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/apimachinery/pkg/runtime" "k8s.io/apimachinery/pkg/runtime/serializer/streaming" utilerrors "k8s.io/apimachinery/pkg/util/errors" "k8s.io/apimachinery/pkg/watch" restclientwatch "k8s.io/client-go/rest/watch" utiltesting "k8s.io/client-go/util/testing" "k8s.io/kubernetes/pkg/api" "k8s.io/kubernetes/pkg/api/resource" "k8s.io/kubernetes/pkg/api/testapi" apitesting "k8s.io/kubernetes/pkg/api/testing" "k8s.io/kubernetes/pkg/api/v1" "k8s.io/kubernetes/pkg/client/restclient/fake" ) func stringBody(body string) io.ReadCloser { return ioutil.NopCloser(bytes.NewReader([]byte(body))) } func watchBody(events ...watch.Event) string { buf := &bytes.Buffer{} codec := testapi.Default.Codec() enc := restclientwatch.NewEncoder(streaming.NewEncoder(buf, codec), codec) for _, e := range events { enc.Encode(&e) } return buf.String() } func fakeClient() ClientMapper { return ClientMapperFunc(func(*meta.RESTMapping) (RESTClient, error) { return &fake.RESTClient{}, nil }) } func fakeClientWith(testName string, t *testing.T, data map[string]string) ClientMapper { return ClientMapperFunc(func(*meta.RESTMapping) (RESTClient, error) { return &fake.RESTClient{ NegotiatedSerializer: testapi.Default.NegotiatedSerializer(), Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) { p := req.URL.Path q := req.URL.RawQuery if len(q) != 0 { p = p + "?" + q } body, ok := data[p] if !ok { t.Fatalf("%s: unexpected request: %s (%s)\n%#v", testName, p, req.URL, req) } header := http.Header{} header.Set("Content-Type", runtime.ContentTypeJSON) return &http.Response{ StatusCode: http.StatusOK, Header: header, Body: stringBody(body), }, nil }), }, nil }) } func testData() (*api.PodList, *api.ServiceList) { pods := &api.PodList{ ListMeta: metav1.ListMeta{ ResourceVersion: "15", }, Items: []api.Pod{ { ObjectMeta: metav1.ObjectMeta{Name: "foo", Namespace: "test", ResourceVersion: "10"}, Spec: apitesting.DeepEqualSafePodSpec(), }, { ObjectMeta: metav1.ObjectMeta{Name: "bar", Namespace: "test", ResourceVersion: "11"}, Spec: apitesting.DeepEqualSafePodSpec(), }, }, } svc := &api.ServiceList{ ListMeta: metav1.ListMeta{ ResourceVersion: "16", }, Items: []api.Service{ { ObjectMeta: metav1.ObjectMeta{Name: "baz", Namespace: "test", ResourceVersion: "12"}, Spec: api.ServiceSpec{ Type: "ClusterIP", SessionAffinity: "None", }, }, }, } return pods, svc } func streamTestData() (io.Reader, *api.PodList, *api.ServiceList) { pods, svc := testData() r, w := io.Pipe() go func() { defer w.Close() w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), pods))) w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), svc))) }() return r, pods, svc } func JSONToYAMLOrDie(in []byte) []byte { data, err := yaml.JSONToYAML(in) if err != nil { panic(err) } return data } func streamYAMLTestData() (io.Reader, *api.PodList, *api.ServiceList) { pods, svc := testData() r, w := io.Pipe() go func() { defer w.Close() w.Write(JSONToYAMLOrDie([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), pods)))) w.Write([]byte("\n---\n")) w.Write(JSONToYAMLOrDie([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), svc)))) }() return r, pods, svc } func streamTestObject(obj runtime.Object) io.Reader { r, w := io.Pipe() go func() { defer w.Close() w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), obj))) }() return r } type testVisitor struct { InjectErr error Infos []*Info } func (v *testVisitor) Handle(info *Info, err error) error { if err != nil { return err } v.Infos = append(v.Infos, info) return v.InjectErr } func (v *testVisitor) Objects() []runtime.Object { objects := []runtime.Object{} for i := range v.Infos { objects = append(objects, v.Infos[i].Object) } return objects } var aPod string = ` { "kind": "Pod", "apiVersion": "` + api.Registry.GroupOrDie(api.GroupName).GroupVersion.String() + `", "metadata": { "name": "busybox{id}", "labels": { "name": "busybox{id}" } }, "spec": { "containers": [ { "name": "busybox", "image": "busybox", "command": [ "sleep", "3600" ], "imagePullPolicy": "IfNotPresent" } ], "restartPolicy": "Always" } } ` var aRC string = ` { "kind": "ReplicationController", "apiVersion": "` + api.Registry.GroupOrDie(api.GroupName).GroupVersion.String() + `", "metadata": { "name": "busybox{id}", "labels": { "app": "busybox" } }, "spec": { "replicas": 1, "template": { "metadata": { "name": "busybox{id}", "labels": { "app": "busybox{id}" } }, "spec": { "containers": [ { "name": "busybox", "image": "busybox", "command": [ "sleep", "3600" ], "imagePullPolicy": "IfNotPresent" } ], "restartPolicy": "Always" } } } } ` func TestPathBuilderAndVersionedObjectNotDefaulted(t *testing.T) { b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{"../../../test/fixtures/pkg/kubectl/builder/kitten-rc.yaml"}}) test := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || !singleItemImplied || len(test.Infos) != 1 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } info := test.Infos[0] if info.Name != "update-demo-kitten" || info.Namespace != "" || info.Object == nil { t.Errorf("unexpected info: %#v", info) } version, ok := info.VersionedObject.(*v1.ReplicationController) // versioned object does not have defaulting applied if info.VersionedObject == nil || !ok || version.Spec.Replicas != nil { t.Errorf("unexpected versioned object: %#v", info.VersionedObject) } } func TestNodeBuilder(t *testing.T) { node := &api.Node{ ObjectMeta: metav1.ObjectMeta{Name: "node1", Namespace: "should-not-have", ResourceVersion: "10"}, Spec: api.NodeSpec{}, Status: api.NodeStatus{ Capacity: api.ResourceList{ api.ResourceCPU: resource.MustParse("1000m"), api.ResourceMemory: resource.MustParse("1Mi"), }, }, } r, w := io.Pipe() go func() { defer w.Close() w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), node))) }() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). NamespaceParam("test").Stream(r, "STDIN") test := &testVisitor{} err := b.Do().Visit(test.Handle) if err != nil || len(test.Infos) != 1 { t.Fatalf("unexpected response: %v %#v", err, test.Infos) } info := test.Infos[0] if info.Name != "node1" || info.Namespace != "" || info.Object == nil { t.Errorf("unexpected info: %#v", info) } } func createTestDir(t *testing.T, path string) { if err := os.MkdirAll(path, 0750); err != nil { t.Fatalf("error creating test dir: %v", err) } } func writeTestFile(t *testing.T, path string, contents string) { if err := ioutil.WriteFile(path, []byte(contents), 0644); err != nil { t.Fatalf("error creating test file %#v", err) } } func TestPathBuilderWithMultiple(t *testing.T) { // create test dirs tmpDir, err := utiltesting.MkTmpdir("recursive_test_multiple") if err != nil { t.Fatalf("error creating temp dir: %v", err) } createTestDir(t, fmt.Sprintf("%s/%s", tmpDir, "recursive/pod/pod_1")) createTestDir(t, fmt.Sprintf("%s/%s", tmpDir, "recursive/rc/rc_1")) createTestDir(t, fmt.Sprintf("%s/%s", tmpDir, "inode/hardlink")) defer os.RemoveAll(tmpDir) // create test files writeTestFile(t, fmt.Sprintf("%s/recursive/pod/busybox.json", tmpDir), strings.Replace(aPod, "{id}", "0", -1)) writeTestFile(t, fmt.Sprintf("%s/recursive/pod/pod_1/busybox.json", tmpDir), strings.Replace(aPod, "{id}", "1", -1)) writeTestFile(t, fmt.Sprintf("%s/recursive/rc/busybox.json", tmpDir), strings.Replace(aRC, "{id}", "0", -1)) writeTestFile(t, fmt.Sprintf("%s/recursive/rc/rc_1/busybox.json", tmpDir), strings.Replace(aRC, "{id}", "1", -1)) writeTestFile(t, fmt.Sprintf("%s/inode/hardlink/busybox.json", tmpDir), strings.Replace(aPod, "{id}", "0", -1)) if err := os.Link(fmt.Sprintf("%s/inode/hardlink/busybox.json", tmpDir), fmt.Sprintf("%s/inode/hardlink/busybox-link.json", tmpDir)); err != nil { t.Fatalf("error creating test file: %v", err) } tests := []struct { name string object runtime.Object recursive bool directory string expectedNames []string }{ {"pod", &api.Pod{}, false, "../../../examples/pod", []string{"nginx"}}, {"recursive-pod", &api.Pod{}, true, fmt.Sprintf("%s/recursive/pod", tmpDir), []string{"busybox0", "busybox1"}}, {"rc", &api.ReplicationController{}, false, "../../../examples/guestbook/legacy/redis-master-controller.yaml", []string{"redis-master"}}, {"recursive-rc", &api.ReplicationController{}, true, fmt.Sprintf("%s/recursive/rc", tmpDir), []string{"busybox0", "busybox1"}}, {"hardlink", &api.Pod{}, false, fmt.Sprintf("%s/inode/hardlink/busybox-link.json", tmpDir), []string{"busybox0"}}, {"hardlink", &api.Pod{}, true, fmt.Sprintf("%s/inode/hardlink/busybox-link.json", tmpDir), []string{"busybox0"}}, } for _, test := range tests { b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). FilenameParam(false, &FilenameOptions{Recursive: test.recursive, Filenames: []string{test.directory}}). NamespaceParam("test").DefaultNamespace() testVisitor := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(testVisitor.Handle) if err != nil { t.Fatalf("unexpected response: %v %t %#v %s", err, singleItemImplied, testVisitor.Infos, test.name) } info := testVisitor.Infos for i, v := range info { switch test.object.(type) { case *api.Pod: if _, ok := v.Object.(*api.Pod); !ok || v.Name != test.expectedNames[i] || v.Namespace != "test" { t.Errorf("unexpected info: %#v", v) } case *api.ReplicationController: if _, ok := v.Object.(*api.ReplicationController); !ok || v.Name != test.expectedNames[i] || v.Namespace != "test" { t.Errorf("unexpected info: %#v", v) } } } } } func TestPathBuilderWithMultipleInvalid(t *testing.T) { // create test dirs tmpDir, err := utiltesting.MkTmpdir("recursive_test_multiple_invalid") if err != nil { t.Fatalf("error creating temp dir: %v", err) } createTestDir(t, fmt.Sprintf("%s/%s", tmpDir, "inode/symlink/pod")) defer os.RemoveAll(tmpDir) // create test files writeTestFile(t, fmt.Sprintf("%s/inode/symlink/pod/busybox.json", tmpDir), strings.Replace(aPod, "{id}", "0", -1)) if err := os.Symlink(fmt.Sprintf("%s/inode/symlink/pod", tmpDir), fmt.Sprintf("%s/inode/symlink/pod-link", tmpDir)); err != nil { t.Fatalf("error creating test file: %v", err) } if err := os.Symlink(fmt.Sprintf("%s/inode/symlink/loop", tmpDir), fmt.Sprintf("%s/inode/symlink/loop", tmpDir)); err != nil { t.Fatalf("error creating test file: %v", err) } tests := []struct { name string recursive bool directory string }{ {"symlink", false, fmt.Sprintf("%s/inode/symlink/pod-link", tmpDir)}, {"symlink", true, fmt.Sprintf("%s/inode/symlink/pod-link", tmpDir)}, {"loop", false, fmt.Sprintf("%s/inode/symlink/loop", tmpDir)}, {"loop", true, fmt.Sprintf("%s/inode/symlink/loop", tmpDir)}, } for _, test := range tests { b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). FilenameParam(false, &FilenameOptions{Recursive: test.recursive, Filenames: []string{test.directory}}). NamespaceParam("test").DefaultNamespace() testVisitor := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(testVisitor.Handle) if err == nil { t.Fatalf("unexpected response: %v %t %#v %s", err, singleItemImplied, testVisitor.Infos, test.name) } } } func TestDirectoryBuilder(t *testing.T) { b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{"../../../examples/guestbook/legacy"}}). NamespaceParam("test").DefaultNamespace() test := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || singleItemImplied || len(test.Infos) < 3 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } found := false for _, info := range test.Infos { if info.Name == "redis-master" && info.Namespace == "test" && info.Object != nil { found = true break } } if !found { t.Errorf("unexpected responses: %#v", test.Infos) } } func TestNamespaceOverride(t *testing.T) { s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { w.WriteHeader(http.StatusOK) w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), &api.Pod{ObjectMeta: metav1.ObjectMeta{Namespace: "foo", Name: "test"}}))) })) defer s.Close() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{s.URL}}). NamespaceParam("test") test := &testVisitor{} err := b.Do().Visit(test.Handle) if err != nil || len(test.Infos) != 1 && test.Infos[0].Namespace != "foo" { t.Fatalf("unexpected response: %v %#v", err, test.Infos) } b = NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). FilenameParam(true, &FilenameOptions{Recursive: false, Filenames: []string{s.URL}}). NamespaceParam("test") test = &testVisitor{} err = b.Do().Visit(test.Handle) if err == nil { t.Fatalf("expected namespace error. got: %#v", test.Infos) } } func TestURLBuilder(t *testing.T) { s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { w.WriteHeader(http.StatusOK) w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), &api.Pod{ObjectMeta: metav1.ObjectMeta{Namespace: "foo", Name: "test"}}))) w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), &api.Pod{ObjectMeta: metav1.ObjectMeta{Namespace: "foo", Name: "test1"}}))) })) defer s.Close() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{s.URL}}). NamespaceParam("foo") test := &testVisitor{} err := b.Do().Visit(test.Handle) if err != nil || len(test.Infos) != 2 { t.Fatalf("unexpected response: %v %#v", err, test.Infos) } info := test.Infos[0] if info.Name != "test" || info.Namespace != "foo" || info.Object == nil { t.Errorf("unexpected info: %#v", info) } info = test.Infos[1] if info.Name != "test1" || info.Namespace != "foo" || info.Object == nil { t.Errorf("unexpected info: %#v", info) } } func TestURLBuilderRequireNamespace(t *testing.T) { s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { w.WriteHeader(http.StatusOK) w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), &api.Pod{ObjectMeta: metav1.ObjectMeta{Namespace: "foo", Name: "test"}}))) })) defer s.Close() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{s.URL}}). NamespaceParam("test").RequireNamespace() test := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err == nil || !singleItemImplied || len(test.Infos) != 0 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } } func TestResourceByName(t *testing.T) { pods, _ := testData() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods/foo": runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[0]), }), testapi.Default.Codec()). NamespaceParam("test") test := &testVisitor{} singleItemImplied := false if b.Do().Err() == nil { t.Errorf("unexpected non-error") } b.ResourceTypeOrNameArgs(true, "pods", "foo") err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || !singleItemImplied || len(test.Infos) != 1 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } if !reflect.DeepEqual(&pods.Items[0], test.Objects()[0]) { t.Errorf("unexpected object: %#v", test.Objects()[0]) } mapping, err := b.Do().ResourceMapping() if err != nil { t.Fatalf("unexpected error: %v", err) } if mapping.Resource != "pods" { t.Errorf("unexpected resource mapping: %#v", mapping) } } func TestMultipleResourceByTheSameName(t *testing.T) { pods, svcs := testData() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods/foo": runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[0]), "/namespaces/test/pods/baz": runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[1]), "/namespaces/test/services/foo": runtime.EncodeOrDie(testapi.Default.Codec(), &svcs.Items[0]), "/namespaces/test/services/baz": runtime.EncodeOrDie(testapi.Default.Codec(), &svcs.Items[0]), }), testapi.Default.Codec()). NamespaceParam("test") test := &testVisitor{} singleItemImplied := false if b.Do().Err() == nil { t.Errorf("unexpected non-error") } b.ResourceTypeOrNameArgs(true, "pods,services", "foo", "baz") err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || singleItemImplied || len(test.Infos) != 4 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } if !api.Semantic.DeepDerivative([]runtime.Object{&pods.Items[0], &pods.Items[1], &svcs.Items[0], &svcs.Items[0]}, test.Objects()) { t.Errorf("unexpected visited objects: %#v", test.Objects()) } if _, err := b.Do().ResourceMapping(); err == nil { t.Errorf("unexpected non-error") } } func TestResourceNames(t *testing.T) { pods, svc := testData() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods/foo": runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[0]), "/namespaces/test/services/baz": runtime.EncodeOrDie(testapi.Default.Codec(), &svc.Items[0]), }), testapi.Default.Codec()). NamespaceParam("test") test := &testVisitor{} if b.Do().Err() == nil { t.Errorf("unexpected non-error") } b.ResourceNames("pods", "foo", "services/baz") err := b.Do().Visit(test.Handle) if err != nil || len(test.Infos) != 2 { t.Fatalf("unexpected response: %v %#v", err, test.Infos) } if !reflect.DeepEqual(&pods.Items[0], test.Objects()[0]) { t.Errorf("unexpected object: \n%#v, expected: \n%#v", test.Objects()[0], &pods.Items[0]) } if !reflect.DeepEqual(&svc.Items[0], test.Objects()[1]) { t.Errorf("unexpected object: \n%#v, expected: \n%#v", test.Objects()[1], &svc.Items[0]) } } func TestResourceNamesWithoutResource(t *testing.T) { pods, svc := testData() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods/foo": runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[0]), "/namespaces/test/services/baz": runtime.EncodeOrDie(testapi.Default.Codec(), &svc.Items[0]), }), testapi.Default.Codec()). NamespaceParam("test") test := &testVisitor{} if b.Do().Err() == nil { t.Errorf("unexpected non-error") } b.ResourceNames("", "foo", "services/baz") err := b.Do().Visit(test.Handle) if err == nil || !strings.Contains(err.Error(), "must be RESOURCE/NAME") { t.Fatalf("unexpected response: %v", err) } } func TestResourceByNameWithoutRequireObject(t *testing.T) { b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{}), testapi.Default.Codec()). NamespaceParam("test") test := &testVisitor{} singleItemImplied := false if b.Do().Err() == nil { t.Errorf("unexpected non-error") } b.ResourceTypeOrNameArgs(true, "pods", "foo").RequireObject(false) err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || !singleItemImplied || len(test.Infos) != 1 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } if test.Infos[0].Name != "foo" { t.Errorf("unexpected name: %#v", test.Infos[0].Name) } if test.Infos[0].Object != nil { t.Errorf("unexpected object: %#v", test.Infos[0].Object) } mapping, err := b.Do().ResourceMapping() if err != nil { t.Fatalf("unexpected error: %v", err) } if mapping.GroupVersionKind.Kind != "Pod" || mapping.Resource != "pods" { t.Errorf("unexpected resource mapping: %#v", mapping) } } func TestResourceByNameAndEmptySelector(t *testing.T) { pods, _ := testData() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods/foo": runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[0]), }), testapi.Default.Codec()). NamespaceParam("test"). SelectorParam(""). ResourceTypeOrNameArgs(true, "pods", "foo") singleItemImplied := false infos, err := b.Do().IntoSingleItemImplied(&singleItemImplied).Infos() if err != nil || !singleItemImplied || len(infos) != 1 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, infos) } if !reflect.DeepEqual(&pods.Items[0], infos[0].Object) { t.Errorf("unexpected object: %#v", infos[0]) } mapping, err := b.Do().ResourceMapping() if err != nil { t.Fatalf("unexpected error: %v", err) } if mapping.Resource != "pods" { t.Errorf("unexpected resource mapping: %#v", mapping) } } func TestSelector(t *testing.T) { pods, svc := testData() labelKey := metav1.LabelSelectorQueryParam(api.Registry.GroupOrDie(api.GroupName).GroupVersion.String()) b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods?" + labelKey + "=a%3Db": runtime.EncodeOrDie(testapi.Default.Codec(), pods), "/namespaces/test/services?" + labelKey + "=a%3Db": runtime.EncodeOrDie(testapi.Default.Codec(), svc), }), testapi.Default.Codec()). SelectorParam("a=b"). NamespaceParam("test"). Flatten() test := &testVisitor{} singleItemImplied := false if b.Do().Err() == nil { t.Errorf("unexpected non-error") } b.ResourceTypeOrNameArgs(true, "pods,service") err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || singleItemImplied || len(test.Infos) != 3 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } if !api.Semantic.DeepDerivative([]runtime.Object{&pods.Items[0], &pods.Items[1], &svc.Items[0]}, test.Objects()) { t.Errorf("unexpected visited objects: %#v", test.Objects()) } if _, err := b.Do().ResourceMapping(); err == nil { t.Errorf("unexpected non-error") } } func TestSelectorRequiresKnownTypes(t *testing.T) { b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). SelectorParam("a=b"). NamespaceParam("test"). ResourceTypes("unknown") if b.Do().Err() == nil { t.Errorf("unexpected non-error") } } func TestSingleResourceType(t *testing.T) { b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). SelectorParam("a=b"). SingleResourceType(). ResourceTypeOrNameArgs(true, "pods,services") if b.Do().Err() == nil { t.Errorf("unexpected non-error") } } func TestResourceTuple(t *testing.T) { expectNoErr := func(err error) bool { return err == nil } expectErr := func(err error) bool { return err != nil } testCases := map[string]struct { args []string errFn func(error) bool }{ "valid": { args: []string{"pods/foo"}, errFn: expectNoErr, }, "valid multiple with name indirection": { args: []string{"pods/foo", "pod/bar"}, errFn: expectNoErr, }, "valid multiple with namespaced and non-namespaced types": { args: []string{"nodes/foo", "pod/bar"}, errFn: expectNoErr, }, "mixed arg types": { args: []string{"pods/foo", "bar"}, errFn: expectErr, }, /*"missing resource": { args: []string{"pods/foo2"}, errFn: expectNoErr, // not an error because resources are lazily visited },*/ "comma in resource": { args: []string{",pods/foo"}, errFn: expectErr, }, "multiple types in resource": { args: []string{"pods,services/foo"}, errFn: expectErr, }, "unknown resource type": { args: []string{"unknown/foo"}, errFn: expectErr, }, "leading slash": { args: []string{"/bar"}, errFn: expectErr, }, "trailing slash": { args: []string{"bar/"}, errFn: expectErr, }, } for k, testCase := range testCases { for _, requireObject := range []bool{true, false} { expectedRequests := map[string]string{} if requireObject { pods, _ := testData() expectedRequests = map[string]string{ "/namespaces/test/pods/foo": runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[0]), "/namespaces/test/pods/bar": runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[0]), "/nodes/foo": runtime.EncodeOrDie(testapi.Default.Codec(), &api.Node{ObjectMeta: metav1.ObjectMeta{Name: "foo"}}), } } b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith(k, t, expectedRequests), testapi.Default.Codec()). NamespaceParam("test").DefaultNamespace(). ResourceTypeOrNameArgs(true, testCase.args...).RequireObject(requireObject) r := b.Do() if !testCase.errFn(r.Err()) { t.Errorf("%s: unexpected error: %v", k, r.Err()) } if r.Err() != nil { continue } switch { case (r.singleItemImplied && len(testCase.args) != 1), (!r.singleItemImplied && len(testCase.args) == 1): t.Errorf("%s: result had unexpected singleItemImplied value", k) } info, err := r.Infos() if err != nil { // test error continue } if len(info) != len(testCase.args) { t.Errorf("%s: unexpected number of infos returned: %#v", k, info) } } } } func TestStream(t *testing.T) { r, pods, rc := streamTestData() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). NamespaceParam("test").Stream(r, "STDIN").Flatten() test := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || singleItemImplied || len(test.Infos) != 3 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } if !api.Semantic.DeepDerivative([]runtime.Object{&pods.Items[0], &pods.Items[1], &rc.Items[0]}, test.Objects()) { t.Errorf("unexpected visited objects: %#v", test.Objects()) } } func TestYAMLStream(t *testing.T) { r, pods, rc := streamYAMLTestData() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). NamespaceParam("test").Stream(r, "STDIN").Flatten() test := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || singleItemImplied || len(test.Infos) != 3 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } if !api.Semantic.DeepDerivative([]runtime.Object{&pods.Items[0], &pods.Items[1], &rc.Items[0]}, test.Objects()) { t.Errorf("unexpected visited objects: %#v", test.Objects()) } } func TestMultipleObject(t *testing.T) { r, pods, svc := streamTestData() obj, err := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). NamespaceParam("test").Stream(r, "STDIN").Flatten(). Do().Object() if err != nil { t.Fatalf("unexpected error: %v", err) } expected := &api.List{ Items: []runtime.Object{ &pods.Items[0], &pods.Items[1], &svc.Items[0], }, } if !api.Semantic.DeepDerivative(expected, obj) { t.Errorf("unexpected visited objects: %#v", obj) } } func TestContinueOnErrorVisitor(t *testing.T) { r, _, _ := streamTestData() req := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). ContinueOnError(). NamespaceParam("test").Stream(r, "STDIN").Flatten(). Do() count := 0 testErr := fmt.Errorf("test error") err := req.Visit(func(_ *Info, _ error) error { count++ if count > 1 { return testErr } return nil }) if err == nil { t.Fatalf("unexpected error: %v", err) } if count != 3 { t.Fatalf("did not visit all infos: %d", count) } agg, ok := err.(utilerrors.Aggregate) if !ok { t.Fatalf("unexpected error: %v", err) } if len(agg.Errors()) != 2 || agg.Errors()[0] != testErr || agg.Errors()[1] != testErr { t.Fatalf("unexpected error: %v", err) } } func TestSingleItemImpliedObject(t *testing.T) { obj, err := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). NamespaceParam("test").DefaultNamespace(). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{"../../../examples/guestbook/legacy/redis-master-controller.yaml"}}). Flatten(). Do().Object() if err != nil { t.Fatalf("unexpected error: %v", err) } rc, ok := obj.(*api.ReplicationController) if !ok { t.Fatalf("unexpected object: %#v", obj) } if rc.Name != "redis-master" || rc.Namespace != "test" { t.Errorf("unexpected controller: %#v", rc) } } func TestSingleItemImpliedObjectNoExtension(t *testing.T) { obj, err := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). NamespaceParam("test").DefaultNamespace(). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{"../../../examples/pod"}}). Flatten(). Do().Object() if err != nil { t.Fatalf("unexpected error: %v", err) } pod, ok := obj.(*api.Pod) if !ok { t.Fatalf("unexpected object: %#v", obj) } if pod.Name != "nginx" || pod.Namespace != "test" { t.Errorf("unexpected pod: %#v", pod) } } func TestSingleItemImpliedRootScopedObject(t *testing.T) { node := &api.Node{ObjectMeta: metav1.ObjectMeta{Name: "test"}, Spec: api.NodeSpec{ExternalID: "test"}} r := streamTestObject(node) infos, err := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). NamespaceParam("test").DefaultNamespace(). Stream(r, "STDIN"). Flatten(). Do().Infos() if err != nil || len(infos) != 1 { t.Fatalf("unexpected error: %v", err) } if infos[0].Namespace != "" { t.Errorf("namespace should be empty: %#v", infos[0]) } n, ok := infos[0].Object.(*api.Node) if !ok { t.Fatalf("unexpected object: %#v", infos[0].Object) } if n.Name != "test" || n.Namespace != "" { t.Errorf("unexpected object: %#v", n) } } func TestListObject(t *testing.T) { pods, _ := testData() labelKey := metav1.LabelSelectorQueryParam(api.Registry.GroupOrDie(api.GroupName).GroupVersion.String()) b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods?" + labelKey + "=a%3Db": runtime.EncodeOrDie(testapi.Default.Codec(), pods), }), testapi.Default.Codec()). SelectorParam("a=b"). NamespaceParam("test"). ResourceTypeOrNameArgs(true, "pods"). Flatten() obj, err := b.Do().Object() if err != nil { t.Fatalf("unexpected error: %v", err) } list, ok := obj.(*api.List) if !ok { t.Fatalf("unexpected object: %#v", obj) } if list.ResourceVersion != pods.ResourceVersion || len(list.Items) != 2 { t.Errorf("unexpected list: %#v", list) } mapping, err := b.Do().ResourceMapping() if err != nil { t.Fatalf("unexpected error: %v", err) } if mapping.Resource != "pods" { t.Errorf("unexpected resource mapping: %#v", mapping) } } func TestListObjectWithDifferentVersions(t *testing.T) { pods, svc := testData() labelKey := metav1.LabelSelectorQueryParam(api.Registry.GroupOrDie(api.GroupName).GroupVersion.String()) obj, err := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods?" + labelKey + "=a%3Db": runtime.EncodeOrDie(testapi.Default.Codec(), pods), "/namespaces/test/services?" + labelKey + "=a%3Db": runtime.EncodeOrDie(testapi.Default.Codec(), svc), }), testapi.Default.Codec()). SelectorParam("a=b"). NamespaceParam("test"). ResourceTypeOrNameArgs(true, "pods,services"). Flatten(). Do().Object() if err != nil { t.Fatalf("unexpected error: %v", err) } list, ok := obj.(*api.List) if !ok { t.Fatalf("unexpected object: %#v", obj) } // resource version differs between type lists, so it's not possible to get a single version. if list.ResourceVersion != "" || len(list.Items) != 3 { t.Errorf("unexpected list: %#v", list) } } func TestWatch(t *testing.T) { _, svc := testData() w, err := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/watch/namespaces/test/services/redis-master?resourceVersion=12": watchBody(watch.Event{ Type: watch.Added, Object: &svc.Items[0], }), }), testapi.Default.Codec()). NamespaceParam("test").DefaultNamespace(). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{"../../../examples/guestbook/redis-master-service.yaml"}}).Flatten(). Do().Watch("12") if err != nil { t.Fatalf("unexpected error: %v", err) } defer w.Stop() ch := w.ResultChan() select { case obj := <-ch: if obj.Type != watch.Added { t.Fatalf("unexpected watch event %#v", obj) } service, ok := obj.Object.(*api.Service) if !ok { t.Fatalf("unexpected object: %#v", obj) } if service.Name != "baz" || service.ResourceVersion != "12" { t.Errorf("unexpected service: %#v", service) } } } func TestWatchMultipleError(t *testing.T) { _, err := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). NamespaceParam("test").DefaultNamespace(). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{"../../../examples/guestbook/legacy/redis-master-controller.yaml"}}).Flatten(). FilenameParam(false, &FilenameOptions{Recursive: false, Filenames: []string{"../../../examples/guestbook/legacy/redis-master-controller.yaml"}}).Flatten(). Do().Watch("") if err == nil { t.Fatalf("unexpected non-error") } } func TestLatest(t *testing.T) { r, _, _ := streamTestData() newPod := &api.Pod{ ObjectMeta: metav1.ObjectMeta{Name: "foo", Namespace: "test", ResourceVersion: "13"}, } newPod2 := &api.Pod{ ObjectMeta: metav1.ObjectMeta{Name: "bar", Namespace: "test", ResourceVersion: "14"}, } newSvc := &api.Service{ ObjectMeta: metav1.ObjectMeta{Name: "baz", Namespace: "test", ResourceVersion: "15"}, } b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClientWith("", t, map[string]string{ "/namespaces/test/pods/foo": runtime.EncodeOrDie(testapi.Default.Codec(), newPod), "/namespaces/test/pods/bar": runtime.EncodeOrDie(testapi.Default.Codec(), newPod2), "/namespaces/test/services/baz": runtime.EncodeOrDie(testapi.Default.Codec(), newSvc), }), testapi.Default.Codec()). NamespaceParam("other").Stream(r, "STDIN").Flatten().Latest() test := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err != nil || singleItemImplied || len(test.Infos) != 3 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } if !api.Semantic.DeepDerivative([]runtime.Object{newPod, newPod2, newSvc}, test.Objects()) { t.Errorf("unexpected visited objects: %#v", test.Objects()) } } func TestReceiveMultipleErrors(t *testing.T) { pods, svc := testData() r, w := io.Pipe() go func() { defer w.Close() w.Write([]byte(`{}`)) w.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), &pods.Items[0]))) }() r2, w2 := io.Pipe() go func() { defer w2.Close() w2.Write([]byte(`{}`)) w2.Write([]byte(runtime.EncodeOrDie(testapi.Default.Codec(), &svc.Items[0]))) }() b := NewBuilder(testapi.Default.RESTMapper(), api.Scheme, fakeClient(), testapi.Default.Codec()). Stream(r, "1").Stream(r2, "2"). ContinueOnError() test := &testVisitor{} singleItemImplied := false err := b.Do().IntoSingleItemImplied(&singleItemImplied).Visit(test.Handle) if err == nil || singleItemImplied || len(test.Infos) != 2 { t.Fatalf("unexpected response: %v %t %#v", err, singleItemImplied, test.Infos) } errs, ok := err.(utilerrors.Aggregate) if !ok { t.Fatalf("unexpected error: %v", reflect.TypeOf(err)) } if len(errs.Errors()) != 2 { t.Errorf("unexpected errors %v", errs) } } func TestHasNames(t *testing.T) { basename := filepath.Base(os.Args[0]) tests := []struct { args []string expectedHasName bool expectedError error }{ { args: []string{""}, expectedHasName: false, expectedError: nil, }, { args: []string{"rc"}, expectedHasName: false, expectedError: nil, }, { args: []string{"rc,pod,svc"}, expectedHasName: false, expectedError: nil, }, { args: []string{"rc/foo"}, expectedHasName: true, expectedError: nil, }, { args: []string{"rc", "foo"}, expectedHasName: true, expectedError: nil, }, { args: []string{"rc,pod,svc", "foo"}, expectedHasName: true, expectedError: nil, }, { args: []string{"rc/foo", "rc/bar", "rc/zee"}, expectedHasName: true, expectedError: nil, }, { args: []string{"rc/foo", "bar"}, expectedHasName: false, expectedError: fmt.Errorf("there is no need to specify a resource type as a separate argument when passing arguments in resource/name form (e.g. '" + basename + " get resource/' instead of '" + basename + " get resource resource/'"), }, } for _, test := range tests { hasNames, err := HasNames(test.args) if !reflect.DeepEqual(test.expectedError, err) { t.Errorf("expected HasName to error:\n%s\tgot:\n%s", test.expectedError, err) } if hasNames != test.expectedHasName { t.Errorf("expected HasName to return %v for %s", test.expectedHasName, test.args) } } } func TestMultipleTypesRequested(t *testing.T) { tests := []struct { args []string expectedMultipleTypes bool }{ { args: []string{""}, expectedMultipleTypes: false, }, { args: []string{"all"}, expectedMultipleTypes: true, }, { args: []string{"rc"}, expectedMultipleTypes: false, }, { args: []string{"rc,pod,svc"}, expectedMultipleTypes: true, }, { args: []string{"rc/foo"}, expectedMultipleTypes: false, }, { args: []string{"rc/foo", "rc/bar"}, expectedMultipleTypes: false, }, { args: []string{"rc", "foo"}, expectedMultipleTypes: false, }, { args: []string{"rc,pod,svc", "foo"}, expectedMultipleTypes: true, }, { args: []string{"rc,secrets"}, expectedMultipleTypes: true, }, { args: []string{"rc/foo", "rc/bar", "svc/svc"}, expectedMultipleTypes: true, }, } for _, test := range tests { hasMultipleTypes := MultipleTypesRequested(test.args) if hasMultipleTypes != test.expectedMultipleTypes { t.Errorf("expected HasName to return %v for %s", test.expectedMultipleTypes, test.args) } } }