1292 lines
40 KiB
Go
1292 lines
40 KiB
Go
/*
|
|
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/<resource_name>' instead of '" + basename + " get resource resource/<resource_name>'"),
|
|
},
|
|
}
|
|
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)
|
|
}
|
|
}
|
|
}
|