ca8d97d673
Automatic merge from submit-queue (batch tested with PRs 53743, 53564). If you want to cherry-pick this change to another branch, please follow the instructions <a href="https://github.com/kubernetes/community/blob/master/contributors/devel/cherry-picks.md">here</a>. Polymorphic Scale Client This PR introduces a polymorphic scale client based on discovery information that's able to scale scalable resources in arbitrary group-versions, as long as they present the scale subresource in their discovery information. Currently, it supports `extensions/v1beta1.Scale` and `autoscaling/v1.Scale`, but supporting other versions of scale if/when we produce them should be fairly trivial. It also updates the HPA to use this client, meaning the HPA will now work on any scalable resource, not just things in the `extensions/v1beta1` API group. **Release note**: ```release-note Introduces a polymorphic scale client, allowing HorizontalPodAutoscalers to properly function on scalable resources in any API group. ``` Unblocks #29698 Unblocks #38756 Unblocks #49504 Fixes #38810
1304 lines
42 KiB
Go
1304 lines
42 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/api/core/v1"
|
|
apiequality "k8s.io/apimachinery/pkg/api/equality"
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
|
"k8s.io/apimachinery/pkg/api/resource"
|
|
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"
|
|
"k8s.io/client-go/rest/fake"
|
|
restclientwatch "k8s.io/client-go/rest/watch"
|
|
utiltesting "k8s.io/client-go/util/testing"
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/legacyscheme"
|
|
"k8s.io/kubernetes/pkg/api/testapi"
|
|
apitesting "k8s.io/kubernetes/pkg/api/testing"
|
|
)
|
|
|
|
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{
|
|
GroupVersion: legacyscheme.Registry.GroupOrDie(api.GroupName).GroupVersion,
|
|
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": "` + legacyscheme.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": "` + legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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 !apiequality.Semantic.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(), LegacyCategoryExpander, legacyscheme.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 !apiequality.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(), LegacyCategoryExpander, legacyscheme.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 !apiequality.Semantic.DeepEqual(&pods.Items[0], test.Objects()[0]) {
|
|
t.Errorf("unexpected object: \n%#v, expected: \n%#v", test.Objects()[0], &pods.Items[0])
|
|
}
|
|
if !apiequality.Semantic.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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 !apiequality.Semantic.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(legacyscheme.Registry.GroupOrDie(api.GroupName).GroupVersion.String())
|
|
b := NewBuilder(testapi.Default.RESTMapper(), LegacyCategoryExpander, legacyscheme.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 !apiequality.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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 !apiequality.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(), LegacyCategoryExpander, legacyscheme.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 !apiequality.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(), LegacyCategoryExpander, legacyscheme.Scheme, fakeClient(), testapi.Default.Codec()).
|
|
NamespaceParam("test").Stream(r, "STDIN").Flatten().
|
|
Do().Object()
|
|
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
|
|
expected := &v1.List{
|
|
Items: []runtime.RawExtension{
|
|
{Object: &pods.Items[0]},
|
|
{Object: &pods.Items[1]},
|
|
{Object: &svc.Items[0]},
|
|
},
|
|
}
|
|
if !apiequality.Semantic.DeepDerivative(expected, obj) {
|
|
t.Errorf("unexpected visited objects: %#v", obj)
|
|
}
|
|
}
|
|
|
|
func TestContinueOnErrorVisitor(t *testing.T) {
|
|
r, _, _ := streamTestData()
|
|
req := NewBuilder(testapi.Default.RESTMapper(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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(legacyscheme.Registry.GroupOrDie(api.GroupName).GroupVersion.String())
|
|
b := NewBuilder(testapi.Default.RESTMapper(), LegacyCategoryExpander, legacyscheme.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.(*v1.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(legacyscheme.Registry.GroupOrDie(api.GroupName).GroupVersion.String())
|
|
obj, err := NewBuilder(testapi.Default.RESTMapper(), LegacyCategoryExpander, legacyscheme.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.(*v1.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(), LegacyCategoryExpander, legacyscheme.Scheme, fakeClientWith("", t, map[string]string{
|
|
"/namespaces/test/services?fieldSelector=metadata.name%3Dredis-master&resourceVersion=12&watch=true": 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(), LegacyCategoryExpander, legacyscheme.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(), LegacyCategoryExpander, legacyscheme.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 !apiequality.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(), LegacyCategoryExpander, legacyscheme.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{"pod,all"},
|
|
expectedMultipleTypes: true,
|
|
},
|
|
{
|
|
args: []string{"all,rc,pod"},
|
|
expectedMultipleTypes: true,
|
|
},
|
|
{
|
|
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 MultipleTypesRequested to return %v for %s", test.expectedMultipleTypes, test.args)
|
|
}
|
|
}
|
|
}
|