mirror of
				https://github.com/optim-enterprises-bv/kubernetes.git
				synced 2025-11-04 04:08:16 +00:00 
			
		
		
		
	This reverts commit93be3f7735. Revert "update vendor kube-openapi" This reverts commit006f80e35d.
		
			
				
	
	
		
			1097 lines
		
	
	
		
			33 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			1097 lines
		
	
	
		
			33 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 cmd
 | 
						|
 | 
						|
import (
 | 
						|
	"bytes"
 | 
						|
	encjson "encoding/json"
 | 
						|
	"io"
 | 
						|
	"io/ioutil"
 | 
						|
	"net/http"
 | 
						|
	"reflect"
 | 
						|
	"strings"
 | 
						|
	"testing"
 | 
						|
 | 
						|
	apiequality "k8s.io/apimachinery/pkg/api/equality"
 | 
						|
	"k8s.io/apimachinery/pkg/api/meta"
 | 
						|
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 | 
						|
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
 | 
						|
	"k8s.io/apimachinery/pkg/runtime"
 | 
						|
	"k8s.io/apimachinery/pkg/runtime/schema"
 | 
						|
	"k8s.io/apimachinery/pkg/runtime/serializer/json"
 | 
						|
	"k8s.io/apimachinery/pkg/runtime/serializer/streaming"
 | 
						|
	"k8s.io/apimachinery/pkg/watch"
 | 
						|
	restclient "k8s.io/client-go/rest"
 | 
						|
	"k8s.io/client-go/rest/fake"
 | 
						|
	restclientwatch "k8s.io/client-go/rest/watch"
 | 
						|
	"k8s.io/kubernetes/pkg/api"
 | 
						|
	"k8s.io/kubernetes/pkg/api/testapi"
 | 
						|
	apitesting "k8s.io/kubernetes/pkg/api/testing"
 | 
						|
	cmdtesting "k8s.io/kubernetes/pkg/kubectl/cmd/testing"
 | 
						|
	"k8s.io/kubernetes/pkg/kubectl/cmd/util/openapi"
 | 
						|
)
 | 
						|
 | 
						|
func testData() (*api.PodList, *api.ServiceList, *api.ReplicationControllerList) {
 | 
						|
	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{
 | 
						|
					SessionAffinity: "None",
 | 
						|
					Type:            api.ServiceTypeClusterIP,
 | 
						|
				},
 | 
						|
			},
 | 
						|
		},
 | 
						|
	}
 | 
						|
	rc := &api.ReplicationControllerList{
 | 
						|
		ListMeta: metav1.ListMeta{
 | 
						|
			ResourceVersion: "17",
 | 
						|
		},
 | 
						|
		Items: []api.ReplicationController{
 | 
						|
			{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{Name: "rc1", Namespace: "test", ResourceVersion: "18"},
 | 
						|
				Spec: api.ReplicationControllerSpec{
 | 
						|
					Replicas: 1,
 | 
						|
				},
 | 
						|
			},
 | 
						|
		},
 | 
						|
	}
 | 
						|
	return pods, svc, rc
 | 
						|
}
 | 
						|
 | 
						|
func testComponentStatusData() *api.ComponentStatusList {
 | 
						|
	good := api.ComponentStatus{
 | 
						|
		Conditions: []api.ComponentCondition{
 | 
						|
			{Type: api.ComponentHealthy, Status: api.ConditionTrue, Message: "ok"},
 | 
						|
		},
 | 
						|
		ObjectMeta: metav1.ObjectMeta{Name: "servergood"},
 | 
						|
	}
 | 
						|
 | 
						|
	bad := api.ComponentStatus{
 | 
						|
		Conditions: []api.ComponentCondition{
 | 
						|
			{Type: api.ComponentHealthy, Status: api.ConditionFalse, Message: "", Error: "bad status: 500"},
 | 
						|
		},
 | 
						|
		ObjectMeta: metav1.ObjectMeta{Name: "serverbad"},
 | 
						|
	}
 | 
						|
 | 
						|
	unknown := api.ComponentStatus{
 | 
						|
		Conditions: []api.ComponentCondition{
 | 
						|
			{Type: api.ComponentHealthy, Status: api.ConditionUnknown, Message: "", Error: "fizzbuzz error"},
 | 
						|
		},
 | 
						|
		ObjectMeta: metav1.ObjectMeta{Name: "serverunknown"},
 | 
						|
	}
 | 
						|
 | 
						|
	return &api.ComponentStatusList{
 | 
						|
		Items: []api.ComponentStatus{good, bad, unknown},
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
// Verifies that schemas that are not in the master tree of Kubernetes can be retrieved via Get.
 | 
						|
func TestGetUnknownSchemaObject(t *testing.T) {
 | 
						|
	f, tf, _, _ := cmdtesting.NewAPIFactory()
 | 
						|
	_, _, codec, _ := cmdtesting.NewTestFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, cmdtesting.NewInternalType("", "", "foo"))},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	tf.ClientConfig = &restclient.Config{ContentConfig: restclient.ContentConfig{GroupVersion: &api.Registry.GroupOrDie(api.GroupName).GroupVersion}}
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Run(cmd, []string{"type", "foo"})
 | 
						|
 | 
						|
	expected := []runtime.Object{cmdtesting.NewInternalType("", "", "foo")}
 | 
						|
	actual := tf.Printer.(*testPrinter).Objects
 | 
						|
	if len(actual) != len(expected) {
 | 
						|
		t.Fatalf("expected: %#v, but actual: %#v", expected, actual)
 | 
						|
	}
 | 
						|
	for i, obj := range actual {
 | 
						|
		expectedJSON := runtime.EncodeOrDie(codec, expected[i])
 | 
						|
		expectedMap := map[string]interface{}{}
 | 
						|
		if err := encjson.Unmarshal([]byte(expectedJSON), &expectedMap); err != nil {
 | 
						|
			t.Fatal(err)
 | 
						|
		}
 | 
						|
 | 
						|
		actualJSON := runtime.EncodeOrDie(api.Codecs.LegacyCodec(), obj)
 | 
						|
		actualMap := map[string]interface{}{}
 | 
						|
		if err := encjson.Unmarshal([]byte(actualJSON), &actualMap); err != nil {
 | 
						|
			t.Fatal(err)
 | 
						|
		}
 | 
						|
 | 
						|
		if !reflect.DeepEqual(expectedMap, actualMap) {
 | 
						|
			t.Errorf("expectedMap: %#v, but actualMap: %#v", expectedMap, actualMap)
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
// Verifies that schemas that are not in the master tree of Kubernetes can be retrieved via Get.
 | 
						|
func TestGetSchemaObject(t *testing.T) {
 | 
						|
	f, tf, _, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Mapper = testapi.Default.RESTMapper()
 | 
						|
	tf.Typer = api.Scheme
 | 
						|
	codec := testapi.Default.Codec()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &api.ReplicationController{ObjectMeta: metav1.ObjectMeta{Name: "foo"}})},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	tf.ClientConfig = &restclient.Config{ContentConfig: restclient.ContentConfig{GroupVersion: &schema.GroupVersion{Version: "v1"}}}
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.Run(cmd, []string{"replicationcontrollers", "foo"})
 | 
						|
 | 
						|
	if !strings.Contains(buf.String(), "\"foo\"") {
 | 
						|
		t.Errorf("unexpected output: %s", buf.String())
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetObjectsWithOpenAPIOutputFormatPresent(t *testing.T) {
 | 
						|
	pods, _, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	// overide the openAPISchema function to return custom output
 | 
						|
	// for Pod type.
 | 
						|
	tf.OpenAPISchemaFunc = testOpenAPISchemaData
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &pods.Items[0])},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Flags().Set(useOpenAPIPrintColumnFlagLabel, "true")
 | 
						|
	cmd.Run(cmd, []string{"pods", "foo"})
 | 
						|
 | 
						|
	expected := []runtime.Object{&pods.Items[0]}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
type FakeResources struct {
 | 
						|
	resources map[schema.GroupVersionKind]openapi.Schema
 | 
						|
}
 | 
						|
 | 
						|
func (f FakeResources) LookupResource(s schema.GroupVersionKind) openapi.Schema {
 | 
						|
	return f.resources[s]
 | 
						|
}
 | 
						|
 | 
						|
var _ openapi.Resources = &FakeResources{}
 | 
						|
 | 
						|
func testOpenAPISchemaData() (openapi.Resources, error) {
 | 
						|
	return &FakeResources{
 | 
						|
		resources: map[schema.GroupVersionKind]openapi.Schema{
 | 
						|
			{
 | 
						|
				Version: "v1",
 | 
						|
				Kind:    "Pod",
 | 
						|
			}: &openapi.Primitive{
 | 
						|
				BaseSchema: openapi.BaseSchema{
 | 
						|
					Extensions: map[string]interface{}{
 | 
						|
						"x-kubernetes-print-columns": "custom-columns=NAME:.metadata.name,RSRC:.metadata.resourceVersion",
 | 
						|
					},
 | 
						|
				},
 | 
						|
			},
 | 
						|
		},
 | 
						|
	}, nil
 | 
						|
}
 | 
						|
 | 
						|
func TestGetObjects(t *testing.T) {
 | 
						|
	pods, _, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &pods.Items[0])},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Run(cmd, []string{"pods", "foo"})
 | 
						|
 | 
						|
	expected := []runtime.Object{&pods.Items[0]}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetObjectsFiltered(t *testing.T) {
 | 
						|
	initTestErrorHandler(t)
 | 
						|
 | 
						|
	pods, _, _ := testData()
 | 
						|
	pods.Items[0].Status.Phase = api.PodFailed
 | 
						|
	first := &pods.Items[0]
 | 
						|
	second := &pods.Items[1]
 | 
						|
 | 
						|
	testCases := []struct {
 | 
						|
		args           []string
 | 
						|
		resp           runtime.Object
 | 
						|
		flags          map[string]string
 | 
						|
		expect         []runtime.Object
 | 
						|
		genericPrinter bool
 | 
						|
	}{
 | 
						|
		{args: []string{"pods", "foo"}, resp: first, expect: []runtime.Object{first}, genericPrinter: true},
 | 
						|
		{args: []string{"pods", "foo"}, flags: map[string]string{"show-all": "false"}, resp: first, expect: []runtime.Object{first}, genericPrinter: true},
 | 
						|
		{args: []string{"pods"}, flags: map[string]string{"show-all": "true"}, resp: pods, expect: []runtime.Object{first, second}},
 | 
						|
		{args: []string{"pods/foo"}, resp: first, expect: []runtime.Object{first}, genericPrinter: true},
 | 
						|
		{args: []string{"pods"}, flags: map[string]string{"output": "yaml"}, resp: pods, expect: []runtime.Object{second}},
 | 
						|
		{args: []string{}, flags: map[string]string{"filename": "../../../examples/storage/cassandra/cassandra-controller.yaml"}, resp: pods, expect: []runtime.Object{first, second}},
 | 
						|
 | 
						|
		{args: []string{"pods"}, resp: pods, expect: []runtime.Object{second}},
 | 
						|
		{args: []string{"pods"}, flags: map[string]string{"show-all": "true", "output": "yaml"}, resp: pods, expect: []runtime.Object{first, second}},
 | 
						|
		{args: []string{"pods"}, flags: map[string]string{"show-all": "false"}, resp: pods, expect: []runtime.Object{second}},
 | 
						|
	}
 | 
						|
 | 
						|
	for i, test := range testCases {
 | 
						|
		t.Logf("%d", i)
 | 
						|
		f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
		tf.Printer = &testPrinter{GenericPrinter: test.genericPrinter}
 | 
						|
		tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
			APIRegistry:          api.Registry,
 | 
						|
			NegotiatedSerializer: unstructuredSerializer,
 | 
						|
			Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, test.resp)},
 | 
						|
		}
 | 
						|
		tf.Namespace = "test"
 | 
						|
		buf := bytes.NewBuffer([]byte{})
 | 
						|
		errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
		cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
		cmd.SetOutput(buf)
 | 
						|
		for k, v := range test.flags {
 | 
						|
			cmd.Flags().Lookup(k).Value.Set(v)
 | 
						|
		}
 | 
						|
		cmd.Run(cmd, test.args)
 | 
						|
 | 
						|
		verifyObjects(t, test.expect, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
		if len(buf.String()) == 0 {
 | 
						|
			t.Errorf("%d: unexpected empty output", i)
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetObjectIgnoreNotFound(t *testing.T) {
 | 
						|
	initTestErrorHandler(t)
 | 
						|
 | 
						|
	ns := &api.NamespaceList{
 | 
						|
		ListMeta: metav1.ListMeta{
 | 
						|
			ResourceVersion: "1",
 | 
						|
		},
 | 
						|
		Items: []api.Namespace{
 | 
						|
			{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{Name: "testns", Namespace: "test", ResourceVersion: "11"},
 | 
						|
				Spec:       api.NamespaceSpec{},
 | 
						|
			},
 | 
						|
		},
 | 
						|
	}
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{GenericPrinter: true}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			switch p, m := req.URL.Path, req.Method; {
 | 
						|
			case p == "/namespaces/test/pods/nonexistentpod" && m == "GET":
 | 
						|
				return &http.Response{StatusCode: 404, Header: defaultHeader(), Body: stringBody("")}, nil
 | 
						|
			case p == "/api/v1/namespaces/test" && m == "GET":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &ns.Items[0])}, nil
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Flags().Set("ignore-not-found", "true")
 | 
						|
	cmd.Flags().Set("output", "yaml")
 | 
						|
	cmd.Run(cmd, []string{"pods", "nonexistentpod"})
 | 
						|
 | 
						|
	if buf.String() != "" {
 | 
						|
		t.Errorf("unexpected output: %s", buf.String())
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetSortedObjects(t *testing.T) {
 | 
						|
	pods := &api.PodList{
 | 
						|
		ListMeta: metav1.ListMeta{
 | 
						|
			ResourceVersion: "15",
 | 
						|
		},
 | 
						|
		Items: []api.Pod{
 | 
						|
			{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{Name: "c", Namespace: "test", ResourceVersion: "10"},
 | 
						|
				Spec:       apitesting.DeepEqualSafePodSpec(),
 | 
						|
			},
 | 
						|
			{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{Name: "b", Namespace: "test", ResourceVersion: "11"},
 | 
						|
				Spec:       apitesting.DeepEqualSafePodSpec(),
 | 
						|
			},
 | 
						|
			{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{Name: "a", Namespace: "test", ResourceVersion: "9"},
 | 
						|
				Spec:       apitesting.DeepEqualSafePodSpec(),
 | 
						|
			},
 | 
						|
		},
 | 
						|
	}
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, pods)},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	tf.ClientConfig = &restclient.Config{ContentConfig: restclient.ContentConfig{GroupVersion: &schema.GroupVersion{Version: "v1"}}}
 | 
						|
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	// sorting with metedata.name
 | 
						|
	cmd.Flags().Set("sort-by", ".metadata.name")
 | 
						|
	cmd.Run(cmd, []string{"pods"})
 | 
						|
 | 
						|
	// expect sorted: a,b,c
 | 
						|
	expected := []runtime.Object{&pods.Items[2], &pods.Items[1], &pods.Items[0]}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func verifyObjects(t *testing.T, expected, actual []runtime.Object) {
 | 
						|
	var actualObj runtime.Object
 | 
						|
	var err error
 | 
						|
 | 
						|
	if len(actual) != len(expected) {
 | 
						|
		t.Fatalf("expected %d, but actual %d", len(expected), len(actual))
 | 
						|
	}
 | 
						|
	for i, obj := range actual {
 | 
						|
		switch obj.(type) {
 | 
						|
		case runtime.Unstructured, *runtime.Unknown:
 | 
						|
			actualObj, err = runtime.Decode(
 | 
						|
				api.Codecs.UniversalDecoder(),
 | 
						|
				[]byte(runtime.EncodeOrDie(api.Codecs.LegacyCodec(), obj)))
 | 
						|
		default:
 | 
						|
			actualObj = obj
 | 
						|
			err = nil
 | 
						|
		}
 | 
						|
 | 
						|
		if err != nil {
 | 
						|
			t.Fatal(err)
 | 
						|
		}
 | 
						|
		if !apiequality.Semantic.DeepEqual(expected[i], actualObj) {
 | 
						|
			t.Errorf("expected object: %#v, but actualObj:%#v\n", expected[i], actualObj)
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetObjectsIdentifiedByFile(t *testing.T) {
 | 
						|
	pods, _, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{GenericPrinter: true}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &pods.Items[0])},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Flags().Set("filename", "../../../examples/storage/cassandra/cassandra-controller.yaml")
 | 
						|
	cmd.Run(cmd, []string{})
 | 
						|
 | 
						|
	expected := []runtime.Object{&pods.Items[0]}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetListObjects(t *testing.T) {
 | 
						|
	pods, _, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, pods)},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Run(cmd, []string{"pods"})
 | 
						|
 | 
						|
	expected, err := extractResourceList([]runtime.Object{pods})
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func extractResourceList(objs []runtime.Object) ([]runtime.Object, error) {
 | 
						|
	finalObjs := []runtime.Object{}
 | 
						|
	for _, obj := range objs {
 | 
						|
		items, err := meta.ExtractList(obj)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		finalObjs = append(finalObjs, items...)
 | 
						|
	}
 | 
						|
	return finalObjs, nil
 | 
						|
}
 | 
						|
 | 
						|
func TestGetAllListObjects(t *testing.T) {
 | 
						|
	pods, _, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, pods)},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Flags().Set("show-all", "true")
 | 
						|
	cmd.Run(cmd, []string{"pods"})
 | 
						|
 | 
						|
	expected, err := extractResourceList([]runtime.Object{pods})
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetListComponentStatus(t *testing.T) {
 | 
						|
	statuses := testComponentStatusData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, statuses)},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Run(cmd, []string{"componentstatuses"})
 | 
						|
 | 
						|
	expected, err := extractResourceList([]runtime.Object{statuses})
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetMultipleTypeObjects(t *testing.T) {
 | 
						|
	pods, svc, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/namespaces/test/pods":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, pods)}, nil
 | 
						|
			case "/namespaces/test/services":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, svc)}, nil
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Run(cmd, []string{"pods,services"})
 | 
						|
 | 
						|
	expected, err := extractResourceList([]runtime.Object{pods, svc})
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetMultipleTypeObjectsAsList(t *testing.T) {
 | 
						|
	pods, svc, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{GenericPrinter: true}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/namespaces/test/pods":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, pods)}, nil
 | 
						|
			case "/namespaces/test/services":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, svc)}, nil
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	tf.ClientConfig = &restclient.Config{ContentConfig: restclient.ContentConfig{GroupVersion: &api.Registry.GroupOrDie(api.GroupName).GroupVersion}}
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	cmd.Flags().Set("output", "json")
 | 
						|
	cmd.Run(cmd, []string{"pods,services"})
 | 
						|
 | 
						|
	actual := tf.Printer.(*testPrinter).Objects
 | 
						|
	fn := func(obj runtime.Object) unstructured.Unstructured {
 | 
						|
		data, err := runtime.Encode(api.Codecs.LegacyCodec(schema.GroupVersion{Version: "v1"}), obj)
 | 
						|
		if err != nil {
 | 
						|
			panic(err)
 | 
						|
		}
 | 
						|
		out := &unstructured.Unstructured{Object: make(map[string]interface{})}
 | 
						|
		if err := encjson.Unmarshal(data, &out.Object); err != nil {
 | 
						|
			panic(err)
 | 
						|
		}
 | 
						|
		return *out
 | 
						|
	}
 | 
						|
 | 
						|
	expected := &unstructured.UnstructuredList{
 | 
						|
		Object: map[string]interface{}{"kind": "List", "apiVersion": "v1", "metadata": map[string]interface{}{"selfLink": "", "resourceVersion": ""}},
 | 
						|
		Items: []unstructured.Unstructured{
 | 
						|
			fn(&pods.Items[0]),
 | 
						|
			fn(&pods.Items[1]),
 | 
						|
			fn(&svc.Items[0]),
 | 
						|
		},
 | 
						|
	}
 | 
						|
	actualBytes, err := encjson.Marshal(actual[0])
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	expectedBytes, err := encjson.Marshal(expected)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	if string(actualBytes) != string(expectedBytes) {
 | 
						|
		t.Errorf("expectedBytes: %s,but actualBytes: %s", expectedBytes, actualBytes)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetMultipleTypeObjectsWithSelector(t *testing.T) {
 | 
						|
	pods, svc, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			if req.URL.Query().Get(metav1.LabelSelectorQueryParam(api.Registry.GroupOrDie(api.GroupName).GroupVersion.String())) != "a=b" {
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
			}
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/namespaces/test/pods":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, pods)}, nil
 | 
						|
			case "/namespaces/test/services":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, svc)}, nil
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	cmd.Flags().Set("selector", "a=b")
 | 
						|
	cmd.Run(cmd, []string{"pods,services"})
 | 
						|
 | 
						|
	expected, err := extractResourceList([]runtime.Object{pods, svc})
 | 
						|
	if err != nil {
 | 
						|
		t.Fatal(err)
 | 
						|
	}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetMultipleTypeObjectsWithDirectReference(t *testing.T) {
 | 
						|
	_, svc, _ := testData()
 | 
						|
	node := &api.Node{
 | 
						|
		ObjectMeta: metav1.ObjectMeta{
 | 
						|
			Name: "foo",
 | 
						|
		},
 | 
						|
		Spec: api.NodeSpec{
 | 
						|
			ExternalID: "ext",
 | 
						|
		},
 | 
						|
	}
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/nodes/foo":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, node)}, nil
 | 
						|
			case "/namespaces/test/services/bar":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &svc.Items[0])}, nil
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	cmd.Run(cmd, []string{"services/bar", "node/foo"})
 | 
						|
 | 
						|
	expected := []runtime.Object{&svc.Items[0], node}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestGetByFormatForcesFlag(t *testing.T) {
 | 
						|
	pods, _, _ := testData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{GenericPrinter: true}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Resp:                 &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &pods.Items[0])},
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
	cmd.Flags().Lookup("output").Value.Set("yaml")
 | 
						|
	cmd.Run(cmd, []string{"pods"})
 | 
						|
 | 
						|
	showAllFlag, _ := cmd.Flags().GetBool("show-all")
 | 
						|
	if showAllFlag {
 | 
						|
		t.Error("expected showAll to not be true when getting resource")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func watchTestData() ([]api.Pod, []watch.Event) {
 | 
						|
	pods := []api.Pod{
 | 
						|
		{
 | 
						|
			ObjectMeta: metav1.ObjectMeta{
 | 
						|
				Name:            "bar",
 | 
						|
				Namespace:       "test",
 | 
						|
				ResourceVersion: "9",
 | 
						|
			},
 | 
						|
			Spec: apitesting.DeepEqualSafePodSpec(),
 | 
						|
		},
 | 
						|
		{
 | 
						|
			ObjectMeta: metav1.ObjectMeta{
 | 
						|
				Name:            "foo",
 | 
						|
				Namespace:       "test",
 | 
						|
				ResourceVersion: "10",
 | 
						|
			},
 | 
						|
			Spec: apitesting.DeepEqualSafePodSpec(),
 | 
						|
		},
 | 
						|
	}
 | 
						|
	events := []watch.Event{
 | 
						|
		// current state events
 | 
						|
		{
 | 
						|
			Type: watch.Added,
 | 
						|
			Object: &api.Pod{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{
 | 
						|
					Name:            "bar",
 | 
						|
					Namespace:       "test",
 | 
						|
					ResourceVersion: "9",
 | 
						|
				},
 | 
						|
				Spec: apitesting.DeepEqualSafePodSpec(),
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Type: watch.Added,
 | 
						|
			Object: &api.Pod{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{
 | 
						|
					Name:            "foo",
 | 
						|
					Namespace:       "test",
 | 
						|
					ResourceVersion: "10",
 | 
						|
				},
 | 
						|
				Spec: apitesting.DeepEqualSafePodSpec(),
 | 
						|
			},
 | 
						|
		},
 | 
						|
		// resource events
 | 
						|
		{
 | 
						|
			Type: watch.Modified,
 | 
						|
			Object: &api.Pod{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{
 | 
						|
					Name:            "foo",
 | 
						|
					Namespace:       "test",
 | 
						|
					ResourceVersion: "11",
 | 
						|
				},
 | 
						|
				Spec: apitesting.DeepEqualSafePodSpec(),
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Type: watch.Deleted,
 | 
						|
			Object: &api.Pod{
 | 
						|
				ObjectMeta: metav1.ObjectMeta{
 | 
						|
					Name:            "foo",
 | 
						|
					Namespace:       "test",
 | 
						|
					ResourceVersion: "12",
 | 
						|
				},
 | 
						|
				Spec: apitesting.DeepEqualSafePodSpec(),
 | 
						|
			},
 | 
						|
		},
 | 
						|
	}
 | 
						|
	return pods, events
 | 
						|
}
 | 
						|
 | 
						|
func TestWatchSelector(t *testing.T) {
 | 
						|
	pods, events := watchTestData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	podList := &api.PodList{
 | 
						|
		Items: pods,
 | 
						|
		ListMeta: metav1.ListMeta{
 | 
						|
			ResourceVersion: "10",
 | 
						|
		},
 | 
						|
	}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			if req.URL.Query().Get(metav1.LabelSelectorQueryParam(api.Registry.GroupOrDie(api.GroupName).GroupVersion.String())) != "a=b" {
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
			}
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/namespaces/test/pods":
 | 
						|
				if req.URL.Query().Get("watch") == "true" {
 | 
						|
					return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: watchBody(codec, events[2:])}, nil
 | 
						|
				} else {
 | 
						|
					return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, podList)}, nil
 | 
						|
				}
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	cmd.Flags().Set("watch", "true")
 | 
						|
	cmd.Flags().Set("selector", "a=b")
 | 
						|
	cmd.Run(cmd, []string{"pods"})
 | 
						|
 | 
						|
	expected := []runtime.Object{&pods[0], &pods[1], events[2].Object, events[3].Object}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestWatchResource(t *testing.T) {
 | 
						|
	pods, events := watchTestData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/namespaces/test/pods/foo":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &pods[1])}, nil
 | 
						|
			case "/namespaces/test/pods":
 | 
						|
				if req.URL.Query().Get("watch") == "true" && req.URL.Query().Get("fieldSelector") == "metadata.name=foo" {
 | 
						|
					return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: watchBody(codec, events[1:])}, nil
 | 
						|
				}
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	cmd.Flags().Set("watch", "true")
 | 
						|
	cmd.Run(cmd, []string{"pods", "foo"})
 | 
						|
 | 
						|
	expected := []runtime.Object{&pods[1], events[2].Object, events[3].Object}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestWatchResourceIdentifiedByFile(t *testing.T) {
 | 
						|
	pods, events := watchTestData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/namespaces/test/replicationcontrollers/cassandra":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &pods[1])}, nil
 | 
						|
			case "/namespaces/test/replicationcontrollers":
 | 
						|
				if req.URL.Query().Get("watch") == "true" && req.URL.Query().Get("fieldSelector") == "metadata.name=cassandra" {
 | 
						|
					return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: watchBody(codec, events[1:])}, nil
 | 
						|
				}
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	cmd.Flags().Set("watch", "true")
 | 
						|
	cmd.Flags().Set("filename", "../../../examples/storage/cassandra/cassandra-controller.yaml")
 | 
						|
	cmd.Run(cmd, []string{})
 | 
						|
 | 
						|
	expected := []runtime.Object{&pods[1], events[2].Object, events[3].Object}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestWatchOnlyResource(t *testing.T) {
 | 
						|
	pods, events := watchTestData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/namespaces/test/pods/foo":
 | 
						|
				return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, &pods[1])}, nil
 | 
						|
			case "/namespaces/test/pods":
 | 
						|
				if req.URL.Query().Get("watch") == "true" && req.URL.Query().Get("fieldSelector") == "metadata.name=foo" {
 | 
						|
					return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: watchBody(codec, events[1:])}, nil
 | 
						|
				}
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	cmd.Flags().Set("watch-only", "true")
 | 
						|
	cmd.Run(cmd, []string{"pods", "foo"})
 | 
						|
 | 
						|
	expected := []runtime.Object{events[2].Object, events[3].Object}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestWatchOnlyList(t *testing.T) {
 | 
						|
	pods, events := watchTestData()
 | 
						|
 | 
						|
	f, tf, codec, _ := cmdtesting.NewAPIFactory()
 | 
						|
	tf.Printer = &testPrinter{}
 | 
						|
	podList := &api.PodList{
 | 
						|
		Items: pods,
 | 
						|
		ListMeta: metav1.ListMeta{
 | 
						|
			ResourceVersion: "10",
 | 
						|
		},
 | 
						|
	}
 | 
						|
	tf.UnstructuredClient = &fake.RESTClient{
 | 
						|
		APIRegistry:          api.Registry,
 | 
						|
		NegotiatedSerializer: unstructuredSerializer,
 | 
						|
		Client: fake.CreateHTTPClient(func(req *http.Request) (*http.Response, error) {
 | 
						|
			switch req.URL.Path {
 | 
						|
			case "/namespaces/test/pods":
 | 
						|
				if req.URL.Query().Get("watch") == "true" {
 | 
						|
					return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: watchBody(codec, events[2:])}, nil
 | 
						|
				} else {
 | 
						|
					return &http.Response{StatusCode: 200, Header: defaultHeader(), Body: objBody(codec, podList)}, nil
 | 
						|
				}
 | 
						|
			default:
 | 
						|
				t.Fatalf("request url: %#v,and request: %#v", req.URL, req)
 | 
						|
				return nil, nil
 | 
						|
			}
 | 
						|
		}),
 | 
						|
	}
 | 
						|
	tf.Namespace = "test"
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	errBuf := bytes.NewBuffer([]byte{})
 | 
						|
 | 
						|
	cmd := NewCmdGet(f, buf, errBuf)
 | 
						|
	cmd.SetOutput(buf)
 | 
						|
 | 
						|
	cmd.Flags().Set("watch-only", "true")
 | 
						|
	cmd.Run(cmd, []string{"pods"})
 | 
						|
 | 
						|
	expected := []runtime.Object{events[2].Object, events[3].Object}
 | 
						|
	verifyObjects(t, expected, tf.Printer.(*testPrinter).Objects)
 | 
						|
 | 
						|
	if len(buf.String()) == 0 {
 | 
						|
		t.Error("unexpected empty output")
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func watchBody(codec runtime.Codec, events []watch.Event) io.ReadCloser {
 | 
						|
	buf := bytes.NewBuffer([]byte{})
 | 
						|
	enc := restclientwatch.NewEncoder(streaming.NewEncoder(buf, codec), codec)
 | 
						|
	for i := range events {
 | 
						|
		enc.Encode(&events[i])
 | 
						|
	}
 | 
						|
	return json.Framer.NewFrameReader(ioutil.NopCloser(buf))
 | 
						|
}
 |