52baf68d50
Signed-off-by: Michał Żyłowski <michal.zylowski@intel.com>
1835 lines
51 KiB
Go
1835 lines
51 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 kubectl
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
yamlserializer "k8s.io/apimachinery/pkg/runtime/serializer/yaml"
|
|
"k8s.io/apimachinery/pkg/util/diff"
|
|
"k8s.io/apimachinery/pkg/util/sets"
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/testapi"
|
|
"k8s.io/kubernetes/pkg/api/v1"
|
|
"k8s.io/kubernetes/pkg/apis/batch"
|
|
"k8s.io/kubernetes/pkg/apis/extensions"
|
|
"k8s.io/kubernetes/pkg/apis/policy"
|
|
kubectltesting "k8s.io/kubernetes/pkg/kubectl/testing"
|
|
"k8s.io/kubernetes/pkg/util/intstr"
|
|
|
|
"github.com/ghodss/yaml"
|
|
)
|
|
|
|
func init() {
|
|
api.Scheme.AddKnownTypes(testapi.Default.InternalGroupVersion(), &kubectltesting.TestStruct{})
|
|
api.Scheme.AddKnownTypes(api.Registry.GroupOrDie(api.GroupName).GroupVersion, &kubectltesting.TestStruct{})
|
|
}
|
|
|
|
var testData = kubectltesting.TestStruct{
|
|
Key: "testValue",
|
|
Map: map[string]int{"TestSubkey": 1},
|
|
StringList: []string{"a", "b", "c"},
|
|
IntList: []int{1, 2, 3},
|
|
}
|
|
|
|
func TestVersionedPrinter(t *testing.T) {
|
|
original := &kubectltesting.TestStruct{Key: "value"}
|
|
p := NewVersionedPrinter(
|
|
ResourcePrinterFunc(func(obj runtime.Object, w io.Writer) error {
|
|
if obj == original {
|
|
t.Fatalf("object should not be identical: %#v", obj)
|
|
}
|
|
if obj.(*kubectltesting.TestStruct).Key != "value" {
|
|
t.Fatalf("object was not converted: %#v", obj)
|
|
}
|
|
return nil
|
|
}),
|
|
api.Scheme,
|
|
api.Registry.GroupOrDie(api.GroupName).GroupVersion,
|
|
)
|
|
if err := p.PrintObj(original, nil); err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestPrintDefault(t *testing.T) {
|
|
printer, found, err := GetPrinter("", "", false, false)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %#v", err)
|
|
}
|
|
if found {
|
|
t.Errorf("no printer should have been found: %#v / %v", printer, err)
|
|
}
|
|
}
|
|
|
|
type TestPrintType struct {
|
|
Data string
|
|
}
|
|
|
|
func (obj *TestPrintType) GetObjectKind() schema.ObjectKind { return schema.EmptyObjectKind }
|
|
|
|
type TestUnknownType struct{}
|
|
|
|
func (obj *TestUnknownType) GetObjectKind() schema.ObjectKind { return schema.EmptyObjectKind }
|
|
|
|
func TestPrinter(t *testing.T) {
|
|
//test inputs
|
|
simpleTest := &TestPrintType{"foo"}
|
|
podTest := &api.Pod{ObjectMeta: metav1.ObjectMeta{Name: "foo"}}
|
|
podListTest := &api.PodList{
|
|
Items: []api.Pod{
|
|
{ObjectMeta: metav1.ObjectMeta{Name: "foo"}},
|
|
{ObjectMeta: metav1.ObjectMeta{Name: "bar"}},
|
|
},
|
|
}
|
|
emptyListTest := &api.PodList{}
|
|
testapi, err := api.Scheme.ConvertToVersion(podTest, api.Registry.GroupOrDie(api.GroupName).GroupVersion)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
|
|
printerTests := []struct {
|
|
Name string
|
|
Format string
|
|
FormatArgument string
|
|
Input runtime.Object
|
|
OutputVersions []schema.GroupVersion
|
|
Expect string
|
|
}{
|
|
{"test json", "json", "", simpleTest, nil, "{\n \"Data\": \"foo\"\n}\n"},
|
|
{"test yaml", "yaml", "", simpleTest, nil, "Data: foo\n"},
|
|
{"test template", "template", "{{if .id}}{{.id}}{{end}}{{if .metadata.name}}{{.metadata.name}}{{end}}",
|
|
podTest, []schema.GroupVersion{v1.SchemeGroupVersion}, "foo"},
|
|
{"test jsonpath", "jsonpath", "{.metadata.name}", podTest, []schema.GroupVersion{v1.SchemeGroupVersion}, "foo"},
|
|
{"test jsonpath list", "jsonpath", "{.items[*].metadata.name}", podListTest, []schema.GroupVersion{v1.SchemeGroupVersion}, "foo bar"},
|
|
{"test jsonpath empty list", "jsonpath", "{.items[*].metadata.name}", emptyListTest, []schema.GroupVersion{v1.SchemeGroupVersion}, ""},
|
|
{"test name", "name", "", podTest, []schema.GroupVersion{v1.SchemeGroupVersion}, "pod/foo\n"},
|
|
{"emits versioned objects", "template", "{{.kind}}", testapi, []schema.GroupVersion{v1.SchemeGroupVersion}, "Pod"},
|
|
}
|
|
for _, test := range printerTests {
|
|
buf := bytes.NewBuffer([]byte{})
|
|
printer, generic, err := GetPrinter(test.Format, test.FormatArgument, false, true)
|
|
if err != nil {
|
|
t.Errorf("in %s, unexpected error: %#v", test.Name, err)
|
|
}
|
|
if generic && len(test.OutputVersions) > 0 {
|
|
printer = NewVersionedPrinter(printer, api.Scheme, test.OutputVersions...)
|
|
}
|
|
if err := printer.PrintObj(test.Input, buf); err != nil {
|
|
t.Errorf("in %s, unexpected error: %#v", test.Name, err)
|
|
}
|
|
if buf.String() != test.Expect {
|
|
t.Errorf("in %s, expect %q, got %q", test.Name, test.Expect, buf.String())
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func TestBadPrinter(t *testing.T) {
|
|
badPrinterTests := []struct {
|
|
Name string
|
|
Format string
|
|
FormatArgument string
|
|
Error error
|
|
}{
|
|
{"empty template", "template", "", fmt.Errorf("template format specified but no template given")},
|
|
{"bad template", "template", "{{ .Name", fmt.Errorf("error parsing template {{ .Name, template: output:1: unclosed action\n")},
|
|
{"bad templatefile", "templatefile", "", fmt.Errorf("templatefile format specified but no template file given")},
|
|
{"bad jsonpath", "jsonpath", "{.Name", fmt.Errorf("error parsing jsonpath {.Name, unclosed action\n")},
|
|
}
|
|
for _, test := range badPrinterTests {
|
|
_, _, err := GetPrinter(test.Format, test.FormatArgument, false, false)
|
|
if err == nil || err.Error() != test.Error.Error() {
|
|
t.Errorf("in %s, expect %s, got %s", test.Name, test.Error, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func testPrinter(t *testing.T, printer ResourcePrinter, unmarshalFunc func(data []byte, v interface{}) error) {
|
|
buf := bytes.NewBuffer([]byte{})
|
|
|
|
err := printer.PrintObj(&testData, buf)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var poutput kubectltesting.TestStruct
|
|
// Verify that given function runs without error.
|
|
err = unmarshalFunc(buf.Bytes(), &poutput)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Use real decode function to undo the versioning process.
|
|
poutput = kubectltesting.TestStruct{}
|
|
s := yamlserializer.NewDecodingSerializer(testapi.Default.Codec())
|
|
if err := runtime.DecodeInto(s, buf.Bytes(), &poutput); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(testData, poutput) {
|
|
t.Errorf("Test data and unmarshaled data are not equal: %v", diff.ObjectDiff(poutput, testData))
|
|
}
|
|
|
|
obj := &api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo"},
|
|
}
|
|
buf.Reset()
|
|
printer.PrintObj(obj, buf)
|
|
var objOut api.Pod
|
|
// Verify that given function runs without error.
|
|
err = unmarshalFunc(buf.Bytes(), &objOut)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %#v", err)
|
|
}
|
|
// Use real decode function to undo the versioning process.
|
|
objOut = api.Pod{}
|
|
if err := runtime.DecodeInto(s, buf.Bytes(), &objOut); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(obj, &objOut) {
|
|
t.Errorf("Unexpected inequality:\n%v", diff.ObjectDiff(obj, &objOut))
|
|
}
|
|
}
|
|
|
|
func TestYAMLPrinter(t *testing.T) {
|
|
testPrinter(t, &YAMLPrinter{}, yaml.Unmarshal)
|
|
}
|
|
|
|
func TestJSONPrinter(t *testing.T) {
|
|
testPrinter(t, &JSONPrinter{}, json.Unmarshal)
|
|
}
|
|
|
|
func TestFormatResourceName(t *testing.T) {
|
|
tests := []struct {
|
|
kind, name string
|
|
want string
|
|
}{
|
|
{"", "", ""},
|
|
{"", "name", "name"},
|
|
{"kind", "", "kind/"}, // should not happen in practice
|
|
{"kind", "name", "kind/name"},
|
|
}
|
|
for _, tt := range tests {
|
|
if got := formatResourceName(tt.kind, tt.name, true); got != tt.want {
|
|
t.Errorf("formatResourceName(%q, %q) = %q, want %q", tt.kind, tt.name, got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func PrintCustomType(obj *TestPrintType, w io.Writer, options PrintOptions) error {
|
|
data := obj.Data
|
|
kind := options.Kind
|
|
if options.WithKind {
|
|
data = kind + "/" + data
|
|
}
|
|
_, err := fmt.Fprintf(w, "%s", data)
|
|
return err
|
|
}
|
|
|
|
func ErrorPrintHandler(obj *TestPrintType, w io.Writer, options PrintOptions) error {
|
|
return fmt.Errorf("ErrorPrintHandler error")
|
|
}
|
|
|
|
func TestCustomTypePrinting(t *testing.T) {
|
|
columns := []string{"Data"}
|
|
printer := NewHumanReadablePrinter(PrintOptions{})
|
|
printer.Handler(columns, nil, PrintCustomType)
|
|
|
|
obj := TestPrintType{"test object"}
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(&obj, buffer)
|
|
if err != nil {
|
|
t.Fatalf("An error occurred printing the custom type: %#v", err)
|
|
}
|
|
expectedOutput := "Data\ntest object"
|
|
if buffer.String() != expectedOutput {
|
|
t.Errorf("The data was not printed as expected. Expected:\n%s\nGot:\n%s", expectedOutput, buffer.String())
|
|
}
|
|
}
|
|
|
|
func TestCustomTypePrintingWithKind(t *testing.T) {
|
|
columns := []string{"Data"}
|
|
printer := NewHumanReadablePrinter(PrintOptions{})
|
|
printer.Handler(columns, nil, PrintCustomType)
|
|
printer.EnsurePrintWithKind("test")
|
|
|
|
obj := TestPrintType{"test object"}
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(&obj, buffer)
|
|
if err != nil {
|
|
t.Fatalf("An error occurred printing the custom type: %#v", err)
|
|
}
|
|
expectedOutput := "Data\ntest/test object"
|
|
if buffer.String() != expectedOutput {
|
|
t.Errorf("The data was not printed as expected. Expected:\n%s\nGot:\n%s", expectedOutput, buffer.String())
|
|
}
|
|
}
|
|
|
|
func TestPrintHandlerError(t *testing.T) {
|
|
columns := []string{"Data"}
|
|
printer := NewHumanReadablePrinter(PrintOptions{})
|
|
printer.Handler(columns, nil, ErrorPrintHandler)
|
|
obj := TestPrintType{"test object"}
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(&obj, buffer)
|
|
if err == nil || err.Error() != "ErrorPrintHandler error" {
|
|
t.Errorf("Did not get the expected error: %#v", err)
|
|
}
|
|
}
|
|
|
|
func TestUnknownTypePrinting(t *testing.T) {
|
|
printer := NewHumanReadablePrinter(PrintOptions{})
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(&TestUnknownType{}, buffer)
|
|
if err == nil {
|
|
t.Errorf("An error was expected from printing unknown type")
|
|
}
|
|
}
|
|
|
|
func TestTemplatePanic(t *testing.T) {
|
|
tmpl := `{{and ((index .currentState.info "foo").state.running.startedAt) .currentState.info.net.state.running.startedAt}}`
|
|
printer, err := NewTemplatePrinter([]byte(tmpl))
|
|
if err != nil {
|
|
t.Fatalf("tmpl fail: %v", err)
|
|
}
|
|
buffer := &bytes.Buffer{}
|
|
err = printer.PrintObj(&api.Pod{}, buffer)
|
|
if err == nil {
|
|
t.Fatalf("expected that template to crash")
|
|
}
|
|
if buffer.String() == "" {
|
|
t.Errorf("no debugging info was printed")
|
|
}
|
|
}
|
|
|
|
func TestNamePrinter(t *testing.T) {
|
|
tests := map[string]struct {
|
|
obj runtime.Object
|
|
expect string
|
|
}{
|
|
"singleObject": {
|
|
&api.Pod{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "Pod",
|
|
},
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo",
|
|
},
|
|
},
|
|
"pod/foo\n"},
|
|
"List": {
|
|
&v1.List{
|
|
TypeMeta: metav1.TypeMeta{
|
|
Kind: "List",
|
|
},
|
|
Items: []runtime.RawExtension{
|
|
{
|
|
Raw: []byte(`{"kind": "Pod", "apiVersion": "v1", "metadata": { "name": "foo"}}`),
|
|
},
|
|
{
|
|
Raw: []byte(`{"kind": "Pod", "apiVersion": "v1", "metadata": { "name": "bar"}}`),
|
|
},
|
|
},
|
|
},
|
|
"pod/foo\npod/bar\n"},
|
|
}
|
|
printer, _, _ := GetPrinter("name", "", false, false)
|
|
for name, item := range tests {
|
|
buff := &bytes.Buffer{}
|
|
err := printer.PrintObj(item.obj, buff)
|
|
if err != nil {
|
|
t.Errorf("%v: unexpected err: %v", name, err)
|
|
continue
|
|
}
|
|
got := buff.String()
|
|
if item.expect != got {
|
|
t.Errorf("%v: expected %v, got %v", name, item.expect, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTemplateStrings(t *testing.T) {
|
|
// This unit tests the "exists" function as well as the template from update.sh
|
|
table := map[string]struct {
|
|
pod api.Pod
|
|
expect string
|
|
}{
|
|
"nilInfo": {api.Pod{}, "false"},
|
|
"emptyInfo": {api.Pod{Status: api.PodStatus{ContainerStatuses: []api.ContainerStatus{}}}, "false"},
|
|
"fooExists": {
|
|
api.Pod{
|
|
Status: api.PodStatus{
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{
|
|
Name: "foo",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"false",
|
|
},
|
|
"barExists": {
|
|
api.Pod{
|
|
Status: api.PodStatus{
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{
|
|
Name: "bar",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"false",
|
|
},
|
|
"bothExist": {
|
|
api.Pod{
|
|
Status: api.PodStatus{
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{
|
|
Name: "foo",
|
|
},
|
|
{
|
|
Name: "bar",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"false",
|
|
},
|
|
"barValid": {
|
|
api.Pod{
|
|
Status: api.PodStatus{
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{
|
|
Name: "foo",
|
|
},
|
|
{
|
|
Name: "bar",
|
|
State: api.ContainerState{
|
|
Running: &api.ContainerStateRunning{
|
|
StartedAt: metav1.Time{},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"false",
|
|
},
|
|
"bothValid": {
|
|
api.Pod{
|
|
Status: api.PodStatus{
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{
|
|
Name: "foo",
|
|
State: api.ContainerState{
|
|
Running: &api.ContainerStateRunning{
|
|
StartedAt: metav1.Time{},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "bar",
|
|
State: api.ContainerState{
|
|
Running: &api.ContainerStateRunning{
|
|
StartedAt: metav1.Time{},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"true",
|
|
},
|
|
}
|
|
// The point of this test is to verify that the below template works.
|
|
tmpl := `{{if (exists . "status" "containerStatuses")}}{{range .status.containerStatuses}}{{if (and (eq .name "foo") (exists . "state" "running"))}}true{{end}}{{end}}{{end}}`
|
|
p, err := NewTemplatePrinter([]byte(tmpl))
|
|
if err != nil {
|
|
t.Fatalf("tmpl fail: %v", err)
|
|
}
|
|
|
|
printer := NewVersionedPrinter(p, api.Scheme, api.Registry.GroupOrDie(api.GroupName).GroupVersion)
|
|
|
|
for name, item := range table {
|
|
buffer := &bytes.Buffer{}
|
|
err = printer.PrintObj(&item.pod, buffer)
|
|
if err != nil {
|
|
t.Errorf("%v: unexpected err: %v", name, err)
|
|
continue
|
|
}
|
|
actual := buffer.String()
|
|
if len(actual) == 0 {
|
|
actual = "false"
|
|
}
|
|
if e := item.expect; e != actual {
|
|
t.Errorf("%v: expected %v, got %v", name, e, actual)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrinters(t *testing.T) {
|
|
om := func(name string) metav1.ObjectMeta { return metav1.ObjectMeta{Name: name} }
|
|
|
|
var (
|
|
err error
|
|
templatePrinter ResourcePrinter
|
|
templatePrinter2 ResourcePrinter
|
|
jsonpathPrinter ResourcePrinter
|
|
)
|
|
|
|
templatePrinter, err = NewTemplatePrinter([]byte("{{.name}}"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
templatePrinter = NewVersionedPrinter(templatePrinter, api.Scheme, v1.SchemeGroupVersion)
|
|
|
|
templatePrinter2, err = NewTemplatePrinter([]byte("{{len .items}}"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
templatePrinter2 = NewVersionedPrinter(templatePrinter2, api.Scheme, v1.SchemeGroupVersion)
|
|
|
|
jsonpathPrinter, err = NewJSONPathPrinter("{.metadata.name}")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
jsonpathPrinter = NewVersionedPrinter(jsonpathPrinter, api.Scheme, v1.SchemeGroupVersion)
|
|
|
|
printers := map[string]ResourcePrinter{
|
|
"humanReadable": NewHumanReadablePrinter(PrintOptions{
|
|
NoHeaders: true,
|
|
}),
|
|
"humanReadableHeaders": NewHumanReadablePrinter(PrintOptions{}),
|
|
"json": &JSONPrinter{},
|
|
"yaml": &YAMLPrinter{},
|
|
"template": templatePrinter,
|
|
"template2": templatePrinter2,
|
|
"jsonpath": jsonpathPrinter,
|
|
"name": &NamePrinter{
|
|
Typer: api.Scheme,
|
|
Decoder: api.Codecs.UniversalDecoder(),
|
|
},
|
|
}
|
|
objects := map[string]runtime.Object{
|
|
"pod": &api.Pod{ObjectMeta: om("pod")},
|
|
"emptyPodList": &api.PodList{},
|
|
"nonEmptyPodList": &api.PodList{Items: []api.Pod{{}}},
|
|
"endpoints": &api.Endpoints{
|
|
Subsets: []api.EndpointSubset{{
|
|
Addresses: []api.EndpointAddress{{IP: "127.0.0.1"}, {IP: "localhost"}},
|
|
Ports: []api.EndpointPort{{Port: 8080}},
|
|
}}},
|
|
}
|
|
// map of printer name to set of objects it should fail on.
|
|
expectedErrors := map[string]sets.String{
|
|
"template2": sets.NewString("pod", "emptyPodList", "endpoints"),
|
|
"jsonpath": sets.NewString("emptyPodList", "nonEmptyPodList", "endpoints"),
|
|
}
|
|
|
|
for pName, p := range printers {
|
|
for oName, obj := range objects {
|
|
b := &bytes.Buffer{}
|
|
if err := p.PrintObj(obj, b); err != nil {
|
|
if set, found := expectedErrors[pName]; found && set.Has(oName) {
|
|
// expected error
|
|
continue
|
|
}
|
|
t.Errorf("printer '%v', object '%v'; error: '%v'", pName, oName, err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrintEventsResultSorted(t *testing.T) {
|
|
// Arrange
|
|
printer := NewHumanReadablePrinter(PrintOptions{})
|
|
|
|
obj := api.EventList{
|
|
Items: []api.Event{
|
|
{
|
|
Source: api.EventSource{Component: "kubelet"},
|
|
Message: "Item 1",
|
|
FirstTimestamp: metav1.NewTime(time.Date(2014, time.January, 15, 0, 0, 0, 0, time.UTC)),
|
|
LastTimestamp: metav1.NewTime(time.Date(2014, time.January, 15, 0, 0, 0, 0, time.UTC)),
|
|
Count: 1,
|
|
Type: api.EventTypeNormal,
|
|
},
|
|
{
|
|
Source: api.EventSource{Component: "scheduler"},
|
|
Message: "Item 2",
|
|
FirstTimestamp: metav1.NewTime(time.Date(1987, time.June, 17, 0, 0, 0, 0, time.UTC)),
|
|
LastTimestamp: metav1.NewTime(time.Date(1987, time.June, 17, 0, 0, 0, 0, time.UTC)),
|
|
Count: 1,
|
|
Type: api.EventTypeNormal,
|
|
},
|
|
{
|
|
Source: api.EventSource{Component: "kubelet"},
|
|
Message: "Item 3",
|
|
FirstTimestamp: metav1.NewTime(time.Date(2002, time.December, 25, 0, 0, 0, 0, time.UTC)),
|
|
LastTimestamp: metav1.NewTime(time.Date(2002, time.December, 25, 0, 0, 0, 0, time.UTC)),
|
|
Count: 1,
|
|
Type: api.EventTypeNormal,
|
|
},
|
|
},
|
|
}
|
|
buffer := &bytes.Buffer{}
|
|
|
|
// Act
|
|
err := printer.PrintObj(&obj, buffer)
|
|
|
|
// Assert
|
|
if err != nil {
|
|
t.Fatalf("An error occurred printing the EventList: %#v", err)
|
|
}
|
|
out := buffer.String()
|
|
VerifyDatesInOrder(out, "\n" /* rowDelimiter */, " " /* columnDelimiter */, t)
|
|
}
|
|
|
|
func TestPrintNodeStatus(t *testing.T) {
|
|
printer := NewHumanReadablePrinter(PrintOptions{})
|
|
table := []struct {
|
|
node api.Node
|
|
status string
|
|
}{
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo1"},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: api.NodeReady, Status: api.ConditionTrue}}},
|
|
},
|
|
status: "Ready",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo2"},
|
|
Spec: api.NodeSpec{Unschedulable: true},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: api.NodeReady, Status: api.ConditionTrue}}},
|
|
},
|
|
status: "Ready,SchedulingDisabled",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo3"},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{
|
|
{Type: api.NodeReady, Status: api.ConditionTrue},
|
|
{Type: api.NodeReady, Status: api.ConditionTrue}}},
|
|
},
|
|
status: "Ready",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo4"},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: api.NodeReady, Status: api.ConditionFalse}}},
|
|
},
|
|
status: "NotReady",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo5"},
|
|
Spec: api.NodeSpec{Unschedulable: true},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: api.NodeReady, Status: api.ConditionFalse}}},
|
|
},
|
|
status: "NotReady,SchedulingDisabled",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo6"},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: "InvalidValue", Status: api.ConditionTrue}}},
|
|
},
|
|
status: "Unknown",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo7"},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{}}},
|
|
},
|
|
status: "Unknown",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo8"},
|
|
Spec: api.NodeSpec{Unschedulable: true},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: "InvalidValue", Status: api.ConditionTrue}}},
|
|
},
|
|
status: "Unknown,SchedulingDisabled",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo9"},
|
|
Spec: api.NodeSpec{Unschedulable: true},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{}}},
|
|
},
|
|
status: "Unknown,SchedulingDisabled",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo10",
|
|
Labels: map[string]string{"kubernetes.io/role": "master"},
|
|
},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: api.NodeReady, Status: api.ConditionTrue}}},
|
|
},
|
|
status: "Ready,master",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo11",
|
|
Labels: map[string]string{"kubernetes.io/role": "node"},
|
|
},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: api.NodeReady, Status: api.ConditionTrue}}},
|
|
},
|
|
status: "Ready,node",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "foo12",
|
|
Labels: map[string]string{"kubeadm.alpha.kubernetes.io/role": "node"},
|
|
},
|
|
Status: api.NodeStatus{Conditions: []api.NodeCondition{{Type: api.NodeReady, Status: api.ConditionTrue}}},
|
|
},
|
|
status: "Ready,node",
|
|
},
|
|
}
|
|
|
|
for _, test := range table {
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(&test.node, buffer)
|
|
if err != nil {
|
|
t.Fatalf("An error occurred printing Node: %#v", err)
|
|
}
|
|
if !contains(strings.Fields(buffer.String()), test.status) {
|
|
t.Fatalf("Expect printing node %s with status %#v, got: %#v", test.node.Name, test.status, buffer.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrintNodeOSImage(t *testing.T) {
|
|
printer := NewHumanReadablePrinter(PrintOptions{
|
|
ColumnLabels: []string{},
|
|
Wide: true,
|
|
})
|
|
|
|
table := []struct {
|
|
node api.Node
|
|
osImage string
|
|
}{
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo1"},
|
|
Status: api.NodeStatus{
|
|
NodeInfo: api.NodeSystemInfo{OSImage: "fake-os-image"},
|
|
Addresses: []api.NodeAddress{{Type: api.NodeExternalIP, Address: "1.1.1.1"}},
|
|
},
|
|
},
|
|
osImage: "fake-os-image",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo2"},
|
|
Status: api.NodeStatus{
|
|
NodeInfo: api.NodeSystemInfo{KernelVersion: "fake-kernel-version"},
|
|
Addresses: []api.NodeAddress{{Type: api.NodeExternalIP, Address: "1.1.1.1"}},
|
|
},
|
|
},
|
|
osImage: "<unknown>",
|
|
},
|
|
}
|
|
|
|
for _, test := range table {
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(&test.node, buffer)
|
|
if err != nil {
|
|
t.Fatalf("An error occurred printing Node: %#v", err)
|
|
}
|
|
if !contains(strings.Fields(buffer.String()), test.osImage) {
|
|
t.Fatalf("Expect printing node %s with os image %#v, got: %#v", test.node.Name, test.osImage, buffer.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrintNodeKernelVersion(t *testing.T) {
|
|
printer := NewHumanReadablePrinter(PrintOptions{
|
|
ColumnLabels: []string{},
|
|
Wide: true,
|
|
})
|
|
|
|
table := []struct {
|
|
node api.Node
|
|
kernelVersion string
|
|
}{
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo1"},
|
|
Status: api.NodeStatus{
|
|
NodeInfo: api.NodeSystemInfo{KernelVersion: "fake-kernel-version"},
|
|
Addresses: []api.NodeAddress{{Type: api.NodeExternalIP, Address: "1.1.1.1"}},
|
|
},
|
|
},
|
|
kernelVersion: "fake-kernel-version",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo2"},
|
|
Status: api.NodeStatus{
|
|
NodeInfo: api.NodeSystemInfo{OSImage: "fake-os-image"},
|
|
Addresses: []api.NodeAddress{{Type: api.NodeExternalIP, Address: "1.1.1.1"}},
|
|
},
|
|
},
|
|
kernelVersion: "<unknown>",
|
|
},
|
|
}
|
|
|
|
for _, test := range table {
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(&test.node, buffer)
|
|
if err != nil {
|
|
t.Fatalf("An error occurred printing Node: %#v", err)
|
|
}
|
|
if !contains(strings.Fields(buffer.String()), test.kernelVersion) {
|
|
t.Fatalf("Expect printing node %s with kernel version %#v, got: %#v", test.node.Name, test.kernelVersion, buffer.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrintNodeExternalIP(t *testing.T) {
|
|
printer := NewHumanReadablePrinter(PrintOptions{
|
|
Wide: true,
|
|
})
|
|
table := []struct {
|
|
node api.Node
|
|
externalIP string
|
|
}{
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo1"},
|
|
Status: api.NodeStatus{Addresses: []api.NodeAddress{{Type: api.NodeExternalIP, Address: "1.1.1.1"}}},
|
|
},
|
|
externalIP: "1.1.1.1",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo2"},
|
|
Status: api.NodeStatus{Addresses: []api.NodeAddress{{Type: api.NodeInternalIP, Address: "1.1.1.1"}}},
|
|
},
|
|
externalIP: "<none>",
|
|
},
|
|
{
|
|
node: api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "foo3"},
|
|
Status: api.NodeStatus{Addresses: []api.NodeAddress{
|
|
{Type: api.NodeLegacyHostIP, Address: "1.1.1.1"},
|
|
{Type: api.NodeExternalIP, Address: "2.2.2.2"},
|
|
{Type: api.NodeInternalIP, Address: "3.3.3.3"},
|
|
{Type: api.NodeExternalIP, Address: "4.4.4.4"},
|
|
}},
|
|
},
|
|
externalIP: "2.2.2.2",
|
|
},
|
|
}
|
|
|
|
for _, test := range table {
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(&test.node, buffer)
|
|
if err != nil {
|
|
t.Fatalf("An error occurred printing Node: %#v", err)
|
|
}
|
|
if !contains(strings.Fields(buffer.String()), test.externalIP) {
|
|
t.Fatalf("Expect printing node %s with external ip %#v, got: %#v", test.node.Name, test.externalIP, buffer.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
func contains(fields []string, field string) bool {
|
|
for _, v := range fields {
|
|
if v == field {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func TestPrintHunmanReadableIngressWithColumnLabels(t *testing.T) {
|
|
ingress := extensions.Ingress{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test1",
|
|
CreationTimestamp: metav1.Time{Time: time.Now().AddDate(-10, 0, 0)},
|
|
Labels: map[string]string{
|
|
"app_name": "kubectl_test_ingress",
|
|
},
|
|
},
|
|
Spec: extensions.IngressSpec{
|
|
Backend: &extensions.IngressBackend{
|
|
ServiceName: "svc",
|
|
ServicePort: intstr.FromInt(93),
|
|
},
|
|
},
|
|
Status: extensions.IngressStatus{
|
|
LoadBalancer: api.LoadBalancerStatus{
|
|
Ingress: []api.LoadBalancerIngress{
|
|
{
|
|
IP: "2.3.4.5",
|
|
Hostname: "localhost.localdomain",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
buff := bytes.Buffer{}
|
|
printIngress(&ingress, &buff, PrintOptions{
|
|
ColumnLabels: []string{"app_name"},
|
|
})
|
|
output := string(buff.Bytes())
|
|
appName := ingress.ObjectMeta.Labels["app_name"]
|
|
if !strings.Contains(output, appName) {
|
|
t.Errorf("expected to container app_name label value %s, but doesn't %s", appName, output)
|
|
}
|
|
}
|
|
|
|
func TestPrintHumanReadableService(t *testing.T) {
|
|
tests := []api.Service{
|
|
{
|
|
Spec: api.ServiceSpec{
|
|
ClusterIP: "1.2.3.4",
|
|
Type: "LoadBalancer",
|
|
Ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
},
|
|
},
|
|
},
|
|
Status: api.ServiceStatus{
|
|
LoadBalancer: api.LoadBalancerStatus{
|
|
Ingress: []api.LoadBalancerIngress{
|
|
{
|
|
IP: "2.3.4.5",
|
|
},
|
|
{
|
|
IP: "3.4.5.6",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Spec: api.ServiceSpec{
|
|
ClusterIP: "1.3.4.5",
|
|
Ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
},
|
|
{
|
|
Port: 8090,
|
|
Protocol: "UDP",
|
|
},
|
|
{
|
|
Port: 8000,
|
|
Protocol: "TCP",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Spec: api.ServiceSpec{
|
|
ClusterIP: "1.4.5.6",
|
|
Type: "LoadBalancer",
|
|
Ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
},
|
|
{
|
|
Port: 8090,
|
|
Protocol: "UDP",
|
|
},
|
|
{
|
|
Port: 8000,
|
|
Protocol: "TCP",
|
|
},
|
|
},
|
|
},
|
|
Status: api.ServiceStatus{
|
|
LoadBalancer: api.LoadBalancerStatus{
|
|
Ingress: []api.LoadBalancerIngress{
|
|
{
|
|
IP: "2.3.4.5",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Spec: api.ServiceSpec{
|
|
ClusterIP: "1.5.6.7",
|
|
Type: "LoadBalancer",
|
|
Ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
},
|
|
{
|
|
Port: 8090,
|
|
Protocol: "UDP",
|
|
},
|
|
{
|
|
Port: 8000,
|
|
Protocol: "TCP",
|
|
},
|
|
},
|
|
},
|
|
Status: api.ServiceStatus{
|
|
LoadBalancer: api.LoadBalancerStatus{
|
|
Ingress: []api.LoadBalancerIngress{
|
|
{
|
|
IP: "2.3.4.5",
|
|
},
|
|
{
|
|
IP: "3.4.5.6",
|
|
},
|
|
{
|
|
IP: "5.6.7.8",
|
|
Hostname: "host5678",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, svc := range tests {
|
|
for _, wide := range []bool{false, true} {
|
|
buff := bytes.Buffer{}
|
|
printService(&svc, &buff, PrintOptions{Wide: wide})
|
|
output := string(buff.Bytes())
|
|
ip := svc.Spec.ClusterIP
|
|
if !strings.Contains(output, ip) {
|
|
t.Errorf("expected to contain ClusterIP %s, but doesn't: %s", ip, output)
|
|
}
|
|
|
|
for n, ingress := range svc.Status.LoadBalancer.Ingress {
|
|
ip = ingress.IP
|
|
// For non-wide output, we only guarantee the first IP to be printed
|
|
if (n == 0 || wide) && !strings.Contains(output, ip) {
|
|
t.Errorf("expected to contain ingress ip %s with wide=%v, but doesn't: %s", ip, wide, output)
|
|
}
|
|
}
|
|
|
|
for _, port := range svc.Spec.Ports {
|
|
portSpec := fmt.Sprintf("%d/%s", port.Port, port.Protocol)
|
|
if !strings.Contains(output, portSpec) {
|
|
t.Errorf("expected to contain port: %s, but doesn't: %s", portSpec, output)
|
|
}
|
|
}
|
|
// Each service should print on one line
|
|
if 1 != strings.Count(output, "\n") {
|
|
t.Errorf("expected a single newline, found %d", strings.Count(output, "\n"))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrintHumanReadableWithNamespace(t *testing.T) {
|
|
namespaceName := "testnamespace"
|
|
name := "test"
|
|
table := []struct {
|
|
obj runtime.Object
|
|
isNamespaced bool
|
|
}{
|
|
{
|
|
obj: &api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.ReplicationController{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
Spec: api.ReplicationControllerSpec{
|
|
Replicas: 2,
|
|
Template: &api.PodTemplateSpec{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Labels: map[string]string{
|
|
"name": "foo",
|
|
"type": "production",
|
|
},
|
|
},
|
|
Spec: api.PodSpec{
|
|
Containers: []api.Container{
|
|
{
|
|
Image: "foo/bar",
|
|
TerminationMessagePath: api.TerminationMessagePathDefault,
|
|
ImagePullPolicy: api.PullIfNotPresent,
|
|
},
|
|
},
|
|
RestartPolicy: api.RestartPolicyAlways,
|
|
DNSPolicy: api.DNSDefault,
|
|
NodeSelector: map[string]string{
|
|
"baz": "blah",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.Service{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
Spec: api.ServiceSpec{
|
|
ClusterIP: "1.2.3.4",
|
|
Ports: []api.ServicePort{
|
|
{
|
|
Port: 80,
|
|
Protocol: "TCP",
|
|
},
|
|
},
|
|
},
|
|
Status: api.ServiceStatus{
|
|
LoadBalancer: api.LoadBalancerStatus{
|
|
Ingress: []api.LoadBalancerIngress{
|
|
{
|
|
IP: "2.3.4.5",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.Endpoints{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
Subsets: []api.EndpointSubset{{
|
|
Addresses: []api.EndpointAddress{{IP: "127.0.0.1"}, {IP: "localhost"}},
|
|
Ports: []api.EndpointPort{{Port: 8080}},
|
|
},
|
|
}},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.Namespace{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name},
|
|
},
|
|
isNamespaced: false,
|
|
},
|
|
{
|
|
obj: &api.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.ServiceAccount{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
Secrets: []api.ObjectReference{},
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.Node{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name},
|
|
Status: api.NodeStatus{},
|
|
},
|
|
isNamespaced: false,
|
|
},
|
|
{
|
|
obj: &api.PersistentVolume{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
Spec: api.PersistentVolumeSpec{},
|
|
},
|
|
isNamespaced: false,
|
|
},
|
|
{
|
|
obj: &api.PersistentVolumeClaim{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
Spec: api.PersistentVolumeClaimSpec{},
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.Event{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
Source: api.EventSource{Component: "kubelet"},
|
|
Message: "Item 1",
|
|
FirstTimestamp: metav1.NewTime(time.Date(2014, time.January, 15, 0, 0, 0, 0, time.UTC)),
|
|
LastTimestamp: metav1.NewTime(time.Date(2014, time.January, 15, 0, 0, 0, 0, time.UTC)),
|
|
Count: 1,
|
|
Type: api.EventTypeNormal,
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.LimitRange{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.ResourceQuota{
|
|
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: namespaceName},
|
|
},
|
|
isNamespaced: true,
|
|
},
|
|
{
|
|
obj: &api.ComponentStatus{
|
|
Conditions: []api.ComponentCondition{
|
|
{Type: api.ComponentHealthy, Status: api.ConditionTrue, Message: "ok", Error: ""},
|
|
},
|
|
},
|
|
isNamespaced: false,
|
|
},
|
|
}
|
|
|
|
for _, test := range table {
|
|
if test.isNamespaced {
|
|
// Expect output to include namespace when requested.
|
|
printer := NewHumanReadablePrinter(PrintOptions{
|
|
WithNamespace: true,
|
|
})
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(test.obj, buffer)
|
|
if err != nil {
|
|
t.Fatalf("An error occurred printing object: %#v", err)
|
|
}
|
|
matched := contains(strings.Fields(buffer.String()), fmt.Sprintf("%s", namespaceName))
|
|
if !matched {
|
|
t.Errorf("Expect printing object to contain namespace: %#v", test.obj)
|
|
}
|
|
} else {
|
|
// Expect error when trying to get all namespaces for un-namespaced object.
|
|
printer := NewHumanReadablePrinter(PrintOptions{
|
|
WithNamespace: true,
|
|
})
|
|
buffer := &bytes.Buffer{}
|
|
err := printer.PrintObj(test.obj, buffer)
|
|
if err == nil {
|
|
t.Errorf("Expected error when printing un-namespaced type")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrintPod(t *testing.T) {
|
|
tests := []struct {
|
|
pod api.Pod
|
|
expect string
|
|
}{
|
|
{
|
|
// Test name, num of containers, restarts, container ready status
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test1"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test1\t1/2\tpodPhase\t6\t",
|
|
},
|
|
{
|
|
// Test container error overwrites pod phase
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test2"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{State: api.ContainerState{Waiting: &api.ContainerStateWaiting{Reason: "ContainerWaitingReason"}}, RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test2\t1/2\tContainerWaitingReason\t6\t",
|
|
},
|
|
{
|
|
// Test the same as the above but with Terminated state and the first container overwrites the rest
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test3"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{State: api.ContainerState{Waiting: &api.ContainerStateWaiting{Reason: "ContainerWaitingReason"}}, RestartCount: 3},
|
|
{State: api.ContainerState{Terminated: &api.ContainerStateTerminated{Reason: "ContainerTerminatedReason"}}, RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test3\t0/2\tContainerWaitingReason\t6\t",
|
|
},
|
|
{
|
|
// Test ready is not enough for reporting running
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test4"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{Ready: true, RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test4\t1/2\tpodPhase\t6\t",
|
|
},
|
|
{
|
|
// Test ready is not enough for reporting running
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test5"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Reason: "OutOfDisk",
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{Ready: true, RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test5\t1/2\tOutOfDisk\t6\t",
|
|
},
|
|
}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
printer := HumanReadablePrinter{}
|
|
for _, test := range tests {
|
|
printer.printPod(&test.pod, buf, PrintOptions{ShowAll: true})
|
|
// We ignore time
|
|
if !strings.HasPrefix(buf.String(), test.expect) {
|
|
t.Fatalf("Expected: %s, got: %s", test.expect, buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
func TestPrintNonTerminatedPod(t *testing.T) {
|
|
tests := []struct {
|
|
pod api.Pod
|
|
expect string
|
|
}{
|
|
{
|
|
// Test pod phase Running should be printed
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test1"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: api.PodRunning,
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test1\t1/2\tRunning\t6\t",
|
|
},
|
|
{
|
|
// Test pod phase Pending should be printed
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test2"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: api.PodPending,
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test2\t1/2\tPending\t6\t",
|
|
},
|
|
{
|
|
// Test pod phase Unknown should be printed
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test3"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: api.PodUnknown,
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test3\t1/2\tUnknown\t6\t",
|
|
},
|
|
{
|
|
// Test pod phase Succeeded shouldn't be printed
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test4"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: api.PodSucceeded,
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"",
|
|
},
|
|
{
|
|
// Test pod phase Failed shouldn't be printed
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test5"},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: api.PodFailed,
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{Ready: true, RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"",
|
|
},
|
|
}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
printer := HumanReadablePrinter{}
|
|
for _, test := range tests {
|
|
printer.printPod(&test.pod, buf, PrintOptions{})
|
|
// We ignore time
|
|
if !strings.HasPrefix(buf.String(), test.expect) {
|
|
t.Fatalf("Expected: %s, got: %s", test.expect, buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
func TestPrintPodWithLabels(t *testing.T) {
|
|
tests := []struct {
|
|
pod api.Pod
|
|
labelColumns []string
|
|
startsWith string
|
|
endsWith string
|
|
}{
|
|
{
|
|
// Test name, num of containers, restarts, container ready status
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test1",
|
|
Labels: map[string]string{"col1": "asd", "COL2": "zxc"},
|
|
},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
[]string{"col1", "COL2"},
|
|
"test1\t1/2\tpodPhase\t6\t",
|
|
"\tasd\tzxc\n",
|
|
},
|
|
{
|
|
// Test name, num of containers, restarts, container ready status
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test1",
|
|
Labels: map[string]string{"col1": "asd", "COL2": "zxc"},
|
|
},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
[]string{},
|
|
"test1\t1/2\tpodPhase\t6\t",
|
|
"\n",
|
|
},
|
|
}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
printer := HumanReadablePrinter{}
|
|
for _, test := range tests {
|
|
printer.printPod(&test.pod, buf, PrintOptions{ColumnLabels: test.labelColumns})
|
|
// We ignore time
|
|
if !strings.HasPrefix(buf.String(), test.startsWith) || !strings.HasSuffix(buf.String(), test.endsWith) {
|
|
t.Fatalf("Expected to start with: %s and end with: %s, but got: %s", test.startsWith, test.endsWith, buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
type stringTestList []struct {
|
|
name, got, exp string
|
|
}
|
|
|
|
func TestTranslateTimestamp(t *testing.T) {
|
|
tl := stringTestList{
|
|
{"a while from now", translateTimestamp(metav1.Time{Time: time.Now().Add(2.1e9)}), "<invalid>"},
|
|
{"almost now", translateTimestamp(metav1.Time{Time: time.Now().Add(1.9e9)}), "0s"},
|
|
{"now", translateTimestamp(metav1.Time{Time: time.Now()}), "0s"},
|
|
{"unknown", translateTimestamp(metav1.Time{}), "<unknown>"},
|
|
{"30 seconds ago", translateTimestamp(metav1.Time{Time: time.Now().Add(-3e10)}), "30s"},
|
|
{"5 minutes ago", translateTimestamp(metav1.Time{Time: time.Now().Add(-3e11)}), "5m"},
|
|
{"an hour ago", translateTimestamp(metav1.Time{Time: time.Now().Add(-6e12)}), "1h"},
|
|
{"2 days ago", translateTimestamp(metav1.Time{Time: time.Now().UTC().AddDate(0, 0, -2)}), "2d"},
|
|
{"months ago", translateTimestamp(metav1.Time{Time: time.Now().UTC().AddDate(0, 0, -90)}), "90d"},
|
|
{"10 years ago", translateTimestamp(metav1.Time{Time: time.Now().UTC().AddDate(-10, 0, 0)}), "10y"},
|
|
}
|
|
for _, test := range tl {
|
|
if test.got != test.exp {
|
|
t.Errorf("On %v, expected '%v', but got '%v'",
|
|
test.name, test.exp, test.got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPrintDeployment(t *testing.T) {
|
|
tests := []struct {
|
|
deployment extensions.Deployment
|
|
expect string
|
|
wideExpect string
|
|
}{
|
|
{
|
|
extensions.Deployment{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test1",
|
|
CreationTimestamp: metav1.Time{Time: time.Now().Add(1.9e9)},
|
|
},
|
|
Spec: extensions.DeploymentSpec{
|
|
Replicas: 5,
|
|
Template: api.PodTemplateSpec{
|
|
Spec: api.PodSpec{
|
|
Containers: []api.Container{
|
|
{
|
|
Name: "fake-container1",
|
|
Image: "fake-image1",
|
|
},
|
|
{
|
|
Name: "fake-container2",
|
|
Image: "fake-image2",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
Selector: &metav1.LabelSelector{MatchLabels: map[string]string{"foo": "bar"}},
|
|
},
|
|
Status: extensions.DeploymentStatus{
|
|
Replicas: 10,
|
|
UpdatedReplicas: 2,
|
|
AvailableReplicas: 1,
|
|
UnavailableReplicas: 4,
|
|
},
|
|
},
|
|
"test1\t5\t10\t2\t1\t0s\n",
|
|
"test1\t5\t10\t2\t1\t0s\tfake-container1,fake-container2\tfake-image1,fake-image2\tfoo=bar\n",
|
|
},
|
|
}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
for _, test := range tests {
|
|
printDeployment(&test.deployment, buf, PrintOptions{})
|
|
if buf.String() != test.expect {
|
|
t.Fatalf("Expected: %s, got: %s", test.expect, buf.String())
|
|
}
|
|
buf.Reset()
|
|
// print deployment with '-o wide' option
|
|
printDeployment(&test.deployment, buf, PrintOptions{Wide: true})
|
|
if buf.String() != test.wideExpect {
|
|
t.Fatalf("Expected: %s, got: %s", test.wideExpect, buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
func TestPrintDaemonSet(t *testing.T) {
|
|
tests := []struct {
|
|
ds extensions.DaemonSet
|
|
startsWith string
|
|
}{
|
|
{
|
|
extensions.DaemonSet{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test1",
|
|
CreationTimestamp: metav1.Time{Time: time.Now().Add(1.9e9)},
|
|
},
|
|
Spec: extensions.DaemonSetSpec{
|
|
Template: api.PodTemplateSpec{
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
},
|
|
},
|
|
Status: extensions.DaemonSetStatus{
|
|
CurrentNumberScheduled: 2,
|
|
DesiredNumberScheduled: 3,
|
|
NumberReady: 1,
|
|
},
|
|
},
|
|
"test1\t3\t2\t1\t<none>\t0s\n",
|
|
},
|
|
}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
for _, test := range tests {
|
|
printDaemonSet(&test.ds, buf, PrintOptions{})
|
|
if !strings.HasPrefix(buf.String(), test.startsWith) {
|
|
t.Fatalf("Expected to start with %s but got %s", test.startsWith, buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
func TestPrintJob(t *testing.T) {
|
|
completions := int32(2)
|
|
tests := []struct {
|
|
job batch.Job
|
|
expect string
|
|
}{
|
|
{
|
|
batch.Job{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "job1",
|
|
CreationTimestamp: metav1.Time{Time: time.Now().Add(1.9e9)},
|
|
},
|
|
Spec: batch.JobSpec{
|
|
Completions: &completions,
|
|
},
|
|
Status: batch.JobStatus{
|
|
Succeeded: 1,
|
|
},
|
|
},
|
|
"job1\t2\t1\t0s\n",
|
|
},
|
|
{
|
|
batch.Job{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "job2",
|
|
CreationTimestamp: metav1.Time{Time: time.Now().AddDate(-10, 0, 0)},
|
|
},
|
|
Spec: batch.JobSpec{
|
|
Completions: nil,
|
|
},
|
|
Status: batch.JobStatus{
|
|
Succeeded: 0,
|
|
},
|
|
},
|
|
"job2\t<none>\t0\t10y\n",
|
|
},
|
|
}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
for _, test := range tests {
|
|
printJob(&test.job, buf, PrintOptions{})
|
|
if buf.String() != test.expect {
|
|
t.Fatalf("Expected: %s, got: %s", test.expect, buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
func TestPrintPodShowLabels(t *testing.T) {
|
|
tests := []struct {
|
|
pod api.Pod
|
|
startsWith string
|
|
endsWith string
|
|
showLabels bool
|
|
}{
|
|
{
|
|
// Test name, num of containers, restarts, container ready status
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test1",
|
|
Labels: map[string]string{"col1": "asd", "COL2": "zxc"},
|
|
},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test1\t1/2\tpodPhase\t6\t",
|
|
"\tCOL2=zxc,col1=asd\n",
|
|
true,
|
|
},
|
|
{
|
|
// Test name, num of containers, restarts, container ready status
|
|
api.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test1",
|
|
Labels: map[string]string{"col3": "asd", "COL4": "zxc"},
|
|
},
|
|
Spec: api.PodSpec{Containers: make([]api.Container, 2)},
|
|
Status: api.PodStatus{
|
|
Phase: "podPhase",
|
|
ContainerStatuses: []api.ContainerStatus{
|
|
{Ready: true, RestartCount: 3, State: api.ContainerState{Running: &api.ContainerStateRunning{}}},
|
|
{RestartCount: 3},
|
|
},
|
|
},
|
|
},
|
|
"test1\t1/2\tpodPhase\t6\t",
|
|
"\n",
|
|
false,
|
|
},
|
|
}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
printer := HumanReadablePrinter{}
|
|
|
|
for _, test := range tests {
|
|
printer.printPod(&test.pod, buf, PrintOptions{ShowLabels: test.showLabels})
|
|
// We ignore time
|
|
if !strings.HasPrefix(buf.String(), test.startsWith) || !strings.HasSuffix(buf.String(), test.endsWith) {
|
|
t.Fatalf("Expected to start with: %s and end with: %s, but got: %s", test.startsWith, test.endsWith, buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
func TestPrintService(t *testing.T) {
|
|
tests := []struct {
|
|
service api.Service
|
|
expect string
|
|
}{
|
|
{
|
|
// Test name, cluster ip, port with protocol
|
|
api.Service{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test1"},
|
|
Spec: api.ServiceSpec{
|
|
Type: api.ServiceTypeClusterIP,
|
|
Ports: []api.ServicePort{
|
|
{Protocol: "tcp",
|
|
Port: 2233},
|
|
},
|
|
ClusterIP: "0.0.0.0",
|
|
},
|
|
},
|
|
"test1\t0.0.0.0\t<none>\t2233/tcp\t<unknown>\n",
|
|
},
|
|
{
|
|
// Test name, cluster ip, port:nodePort with protocol
|
|
api.Service{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "test2"},
|
|
Spec: api.ServiceSpec{
|
|
Type: api.ServiceTypeClusterIP,
|
|
Ports: []api.ServicePort{
|
|
{Protocol: "tcp",
|
|
Port: 8888,
|
|
NodePort: 9999,
|
|
},
|
|
},
|
|
ClusterIP: "10.9.8.7",
|
|
},
|
|
},
|
|
"test2\t10.9.8.7\t<none>\t8888:9999/tcp\t<unknown>\n",
|
|
},
|
|
}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
for _, test := range tests {
|
|
printService(&test.service, buf, PrintOptions{})
|
|
// We ignore time
|
|
if buf.String() != test.expect {
|
|
t.Fatalf("Expected: %s, got: %s %d", test.expect, buf.String(), strings.Compare(test.expect, buf.String()))
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
func TestPrintPodDisruptionBudget(t *testing.T) {
|
|
tests := []struct {
|
|
pdb policy.PodDisruptionBudget
|
|
expect string
|
|
}{
|
|
{
|
|
policy.PodDisruptionBudget{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: "ns1",
|
|
Name: "pdb1",
|
|
CreationTimestamp: metav1.Time{Time: time.Now().Add(1.9e9)},
|
|
},
|
|
Spec: policy.PodDisruptionBudgetSpec{
|
|
MinAvailable: intstr.FromInt(22),
|
|
},
|
|
Status: policy.PodDisruptionBudgetStatus{
|
|
PodDisruptionsAllowed: 5,
|
|
},
|
|
},
|
|
"pdb1\t22\t5\t0s\n",
|
|
}}
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
for _, test := range tests {
|
|
printPodDisruptionBudget(&test.pdb, buf, PrintOptions{})
|
|
if buf.String() != test.expect {
|
|
t.Fatalf("Expected: %s, got: %s", test.expect, buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
}
|
|
|
|
func TestAllowMissingKeys(t *testing.T) {
|
|
tests := []struct {
|
|
Name string
|
|
AllowMissingTemplateKeys bool
|
|
Format string
|
|
Template string
|
|
Input runtime.Object
|
|
Expect string
|
|
Error string
|
|
}{
|
|
{"test template, allow missing keys", true, "template", "{{.blarg}}", &api.Pod{}, "<no value>", ""},
|
|
{"test template, strict", false, "template", "{{.blarg}}", &api.Pod{}, "", `error executing template "{{.blarg}}": template: output:1:2: executing "output" at <.blarg>: map has no entry for key "blarg"`},
|
|
{"test jsonpath, allow missing keys", true, "jsonpath", "{.blarg}", &api.Pod{}, "", ""},
|
|
{"test jsonpath, strict", false, "jsonpath", "{.blarg}", &api.Pod{}, "", "error executing jsonpath \"{.blarg}\": blarg is not found\n"},
|
|
}
|
|
for _, test := range tests {
|
|
buf := bytes.NewBuffer([]byte{})
|
|
printer, _, err := GetPrinter(test.Format, test.Template, false, test.AllowMissingTemplateKeys)
|
|
if err != nil {
|
|
t.Errorf("in %s, unexpected error: %#v", test.Name, err)
|
|
}
|
|
err = printer.PrintObj(test.Input, buf)
|
|
if len(test.Error) == 0 && err != nil {
|
|
t.Errorf("in %s, unexpected error: %v", test.Name, err)
|
|
continue
|
|
}
|
|
if len(test.Error) > 0 {
|
|
if err == nil {
|
|
t.Errorf("in %s, expected to get error: %v", test.Name, test.Error)
|
|
} else if e, a := test.Error, err.Error(); e != a {
|
|
t.Errorf("in %s, expected error %q, got %q", test.Name, e, a)
|
|
}
|
|
continue
|
|
}
|
|
if buf.String() != test.Expect {
|
|
t.Errorf("in %s, expect %q, got %q", test.Name, test.Expect, buf.String())
|
|
}
|
|
}
|
|
}
|