mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-24 20:24:09 +00:00
Merge pull request #82054 from seans3/table-printer-tests
Rewrite tests for tableprinter.go
This commit is contained in:
Kubernetes Prow Robot
GitHub
commit
2b7ceb215a
@ -53,7 +53,7 @@ go_test(
|
|||||||
deps = [
|
deps = [
|
||||||
"//staging/src/k8s.io/api/core/v1:go_default_library",
|
"//staging/src/k8s.io/api/core/v1:go_default_library",
|
||||||
"//staging/src/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library",
|
"//staging/src/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library",
|
||||||
"//staging/src/k8s.io/apimachinery/pkg/apis/meta/v1beta1:go_default_library",
|
|
||||||
"//staging/src/k8s.io/apimachinery/pkg/runtime:go_default_library",
|
"//staging/src/k8s.io/apimachinery/pkg/runtime:go_default_library",
|
||||||
|
"//staging/src/k8s.io/apimachinery/pkg/runtime/schema:go_default_library",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|||||||
@ -18,128 +18,579 @@ package printers
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"fmt"
|
|
||||||
"reflect"
|
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
corev1 "k8s.io/api/core/v1"
|
corev1 "k8s.io/api/core/v1"
|
||||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||||
metav1beta1 "k8s.io/apimachinery/pkg/apis/meta/v1beta1"
|
|
||||||
"k8s.io/apimachinery/pkg/runtime"
|
"k8s.io/apimachinery/pkg/runtime"
|
||||||
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||||
)
|
)
|
||||||
|
|
||||||
var testNamespaceColumnDefinitions = []metav1beta1.TableColumnDefinition{
|
var testPodName = "test-pod-name"
|
||||||
{Name: "Name", Type: "string", Format: "name", Description: metav1.ObjectMeta{}.SwaggerDoc()["name"]},
|
var testPodNamespace = "test-namespace"
|
||||||
{Name: "Status", Type: "string", Description: "The status of the namespace"},
|
|
||||||
{Name: "Age", Type: "string", Description: metav1.ObjectMeta{}.SwaggerDoc()["creationTimestamp"]},
|
var testPod = &corev1.Pod{
|
||||||
|
TypeMeta: metav1.TypeMeta{APIVersion: "v1", Kind: "Pod"},
|
||||||
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
|
Name: testPodName,
|
||||||
|
Namespace: testPodNamespace,
|
||||||
|
Labels: map[string]string{
|
||||||
|
"first-label": "12",
|
||||||
|
"second-label": "label-value",
|
||||||
|
},
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
func testPrintNamespace(obj *corev1.Namespace, options PrintOptions) ([]metav1beta1.TableRow, error) {
|
var testStatus = &metav1.Status{
|
||||||
if options.WithNamespace {
|
TypeMeta: metav1.TypeMeta{APIVersion: "metav1", Kind: "Status"},
|
||||||
return nil, fmt.Errorf("namespace is not namespaced")
|
Status: "Failure",
|
||||||
}
|
Message: "test-status-message",
|
||||||
row := metav1beta1.TableRow{
|
Reason: "test-status-reason",
|
||||||
Object: runtime.RawExtension{Object: obj},
|
|
||||||
}
|
|
||||||
row.Cells = append(row.Cells, obj.Name, obj.Status.Phase, "<unknown>")
|
|
||||||
return []metav1beta1.TableRow{row}, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestPrintRowsForHandlerEntry(t *testing.T) {
|
func TestPrintTable_MissingColumnsRows(t *testing.T) {
|
||||||
printFunc := reflect.ValueOf(testPrintNamespace)
|
tests := []struct {
|
||||||
|
columns []metav1.TableColumnDefinition
|
||||||
testCase := []struct {
|
rows []metav1.TableRow
|
||||||
name string
|
options PrintOptions
|
||||||
h *printHandler
|
expected string
|
||||||
opt PrintOptions
|
|
||||||
eventType string
|
|
||||||
obj runtime.Object
|
|
||||||
includeHeader bool
|
|
||||||
expectOut string
|
|
||||||
expectErr string
|
|
||||||
}{
|
}{
|
||||||
|
// No columns and no rows means table string is empty.
|
||||||
{
|
{
|
||||||
name: "no tablecolumndefinition and includeheader flase",
|
columns: []metav1.TableColumnDefinition{},
|
||||||
h: &printHandler{
|
rows: []metav1.TableRow{},
|
||||||
columnDefinitions: []metav1beta1.TableColumnDefinition{},
|
options: PrintOptions{},
|
||||||
printFunc: printFunc,
|
expected: "",
|
||||||
},
|
|
||||||
opt: PrintOptions{},
|
|
||||||
obj: &corev1.Namespace{
|
|
||||||
ObjectMeta: metav1.ObjectMeta{Name: "test"},
|
|
||||||
},
|
|
||||||
includeHeader: false,
|
|
||||||
expectOut: "test\t\t<unknown>\n",
|
|
||||||
},
|
},
|
||||||
|
// No rows, means table string is empty (columns aren't printed).
|
||||||
{
|
{
|
||||||
name: "no tablecolumndefinition and includeheader true",
|
columns: []metav1.TableColumnDefinition{
|
||||||
h: &printHandler{
|
{Name: "Name", Type: "string"},
|
||||||
columnDefinitions: []metav1beta1.TableColumnDefinition{},
|
{Name: "Ready", Type: "string"},
|
||||||
printFunc: printFunc,
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
},
|
},
|
||||||
opt: PrintOptions{},
|
rows: []metav1.TableRow{},
|
||||||
obj: &corev1.Namespace{
|
options: PrintOptions{},
|
||||||
ObjectMeta: metav1.ObjectMeta{Name: "test"},
|
expected: "",
|
||||||
},
|
|
||||||
includeHeader: true,
|
|
||||||
expectOut: "\ntest\t\t<unknown>\n",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "have tablecolumndefinition and includeheader true",
|
|
||||||
h: &printHandler{
|
|
||||||
columnDefinitions: testNamespaceColumnDefinitions,
|
|
||||||
printFunc: printFunc,
|
|
||||||
},
|
|
||||||
opt: PrintOptions{},
|
|
||||||
obj: &corev1.Namespace{
|
|
||||||
ObjectMeta: metav1.ObjectMeta{Name: "test"},
|
|
||||||
},
|
|
||||||
includeHeader: true,
|
|
||||||
expectOut: "NAME\tSTATUS\tAGE\ntest\t\t<unknown>\n",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "with event type",
|
|
||||||
h: &printHandler{
|
|
||||||
columnDefinitions: testNamespaceColumnDefinitions,
|
|
||||||
printFunc: printFunc,
|
|
||||||
},
|
|
||||||
opt: PrintOptions{},
|
|
||||||
eventType: "ADDED",
|
|
||||||
obj: &corev1.Namespace{
|
|
||||||
ObjectMeta: metav1.ObjectMeta{Name: "test"},
|
|
||||||
},
|
|
||||||
includeHeader: true,
|
|
||||||
expectOut: "EVENT\tNAME\tSTATUS\tAGE\nADDED \ttest\t\t<unknown>\n",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "print namespace and withnamespace true, should not print header",
|
|
||||||
h: &printHandler{
|
|
||||||
columnDefinitions: testNamespaceColumnDefinitions,
|
|
||||||
printFunc: printFunc,
|
|
||||||
},
|
|
||||||
opt: PrintOptions{
|
|
||||||
WithNamespace: true,
|
|
||||||
},
|
|
||||||
obj: &corev1.Namespace{
|
|
||||||
ObjectMeta: metav1.ObjectMeta{Name: "test"},
|
|
||||||
},
|
|
||||||
includeHeader: true,
|
|
||||||
expectOut: "",
|
|
||||||
expectErr: "namespace is not namespaced",
|
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
for _, test := range testCase {
|
|
||||||
t.Run(test.name, func(t *testing.T) {
|
for _, test := range tests {
|
||||||
buffer := &bytes.Buffer{}
|
// Create the table from the columns and rows.
|
||||||
err := printRowsForHandlerEntry(buffer, test.h, test.eventType, test.obj, test.opt, test.includeHeader)
|
table := &metav1.Table{
|
||||||
if err != nil {
|
ColumnDefinitions: test.columns,
|
||||||
if err.Error() != test.expectErr {
|
Rows: test.rows,
|
||||||
t.Errorf("[%s]expect:\n %v\n but got:\n %v\n", test.name, test.expectErr, err)
|
}
|
||||||
}
|
// Print the table
|
||||||
}
|
out := bytes.NewBuffer([]byte{})
|
||||||
if test.expectOut != buffer.String() {
|
printer := NewTablePrinter(test.options)
|
||||||
t.Errorf("[%s]expect:\n %v\n but got:\n %v\n", test.name, test.expectOut, buffer.String())
|
printer.PrintObj(table, out)
|
||||||
}
|
|
||||||
})
|
// Validate the printed table is empty.
|
||||||
|
if len(out.String()) > 0 {
|
||||||
|
t.Errorf("Error Printing Table. Should be empty; got (%s)", out.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPrintTable_ColumnPriority(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
columns []metav1.TableColumnDefinition
|
||||||
|
rows []metav1.TableRow
|
||||||
|
options PrintOptions
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
// Test a basic single row table. Columns with priority > 0 are not printed.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer", Priority: 1}, // Priority > 0
|
||||||
|
{Name: "Age", Type: "string", Priority: 1}, // Priority > 0
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"}},
|
||||||
|
},
|
||||||
|
options: PrintOptions{},
|
||||||
|
expected: "NAME READY STATUS\ntest1 1/1 podPhase\n",
|
||||||
|
},
|
||||||
|
// Test a basic multi-row table row table. Columns with priority > 0 are not printed.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer", Priority: 1}, // Priority > 0
|
||||||
|
{Name: "Age", Type: "string", Priority: 1}, // Priority > 0
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"}},
|
||||||
|
{Cells: []interface{}{"test2", "1/2", "podPhase", int64(30), "21h"}},
|
||||||
|
{Cells: []interface{}{"test3", "4/4", "podPhase", int64(1), "22h"}},
|
||||||
|
},
|
||||||
|
options: PrintOptions{},
|
||||||
|
expected: `NAME READY STATUS
|
||||||
|
test1 1/1 podPhase
|
||||||
|
test2 1/2 podPhase
|
||||||
|
test3 4/4 podPhase
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
// Test a single row table with "wide" printing option. Columns with priority > 0 are printed.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer", Priority: 1},
|
||||||
|
{Name: "Age", Type: "string", Priority: 1},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"}},
|
||||||
|
},
|
||||||
|
// Print with no headers.
|
||||||
|
options: PrintOptions{Wide: true},
|
||||||
|
expected: "NAME READY STATUS RETRIES AGE\ntest1 1/1 podPhase 5 20h\n",
|
||||||
|
},
|
||||||
|
// Test a multi-row table row table with "wide" printing option. Columns with priority > 0 are printed.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer", Priority: 1},
|
||||||
|
{Name: "Age", Type: "string", Priority: 1},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"}},
|
||||||
|
{Cells: []interface{}{"test2", "1/2", "podPhase", int64(30), "21h"}},
|
||||||
|
{Cells: []interface{}{"test3", "4/4", "podPhase", int64(1), "22h"}},
|
||||||
|
},
|
||||||
|
options: PrintOptions{Wide: true},
|
||||||
|
expected: `NAME READY STATUS RETRIES AGE
|
||||||
|
test1 1/1 podPhase 5 20h
|
||||||
|
test2 1/2 podPhase 30 21h
|
||||||
|
test3 4/4 podPhase 1 22h
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
// Create the table from the columns and rows.
|
||||||
|
table := &metav1.Table{
|
||||||
|
ColumnDefinitions: test.columns,
|
||||||
|
Rows: test.rows,
|
||||||
|
}
|
||||||
|
// Print the table
|
||||||
|
out := bytes.NewBuffer([]byte{})
|
||||||
|
printer := NewTablePrinter(test.options)
|
||||||
|
printer.PrintObj(table, out)
|
||||||
|
// Validate the expected output matches the printed table.
|
||||||
|
if test.expected != out.String() {
|
||||||
|
t.Errorf("Table printing error: expected (%s), got (%s)", test.expected, out.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPrintTable_ColumnHeaders(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
columns []metav1.TableColumnDefinition
|
||||||
|
rows []metav1.TableRow
|
||||||
|
options PrintOptions
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
// Test a basic single row table, shows columns.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"}},
|
||||||
|
},
|
||||||
|
options: PrintOptions{},
|
||||||
|
expected: "NAME READY STATUS RETRIES AGE\ntest1 1/1 podPhase 5 20h\n",
|
||||||
|
},
|
||||||
|
// Test a basic multi-row table row table, shows columns.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"}},
|
||||||
|
{Cells: []interface{}{"test2", "1/2", "podPhase", int64(30), "21h"}},
|
||||||
|
{Cells: []interface{}{"test3", "4/4", "podPhase", int64(1), "22h"}},
|
||||||
|
},
|
||||||
|
options: PrintOptions{},
|
||||||
|
expected: `NAME READY STATUS RETRIES AGE
|
||||||
|
test1 1/1 podPhase 5 20h
|
||||||
|
test2 1/2 podPhase 30 21h
|
||||||
|
test3 4/4 podPhase 1 22h
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
// Test a single row table with "NoHeaders" option, doesn't print columns.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"}},
|
||||||
|
},
|
||||||
|
// Print with no headers.
|
||||||
|
options: PrintOptions{NoHeaders: true},
|
||||||
|
expected: "test1 1/1 podPhase 5 20h\n",
|
||||||
|
},
|
||||||
|
// Test a multi-row table row table with "NoHeaders" option, doesn't print columns.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"}},
|
||||||
|
{Cells: []interface{}{"test2", "1/2", "podPhase", int64(30), "21h"}},
|
||||||
|
{Cells: []interface{}{"test3", "4/4", "podPhase", int64(1), "22h"}},
|
||||||
|
},
|
||||||
|
options: PrintOptions{NoHeaders: true},
|
||||||
|
expected: `test1 1/1 podPhase 5 20h
|
||||||
|
test2 1/2 podPhase 30 21h
|
||||||
|
test3 4/4 podPhase 1 22h
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
// Create the table from the columns and rows.
|
||||||
|
table := &metav1.Table{
|
||||||
|
ColumnDefinitions: test.columns,
|
||||||
|
Rows: test.rows,
|
||||||
|
}
|
||||||
|
// Print the table
|
||||||
|
out := bytes.NewBuffer([]byte{})
|
||||||
|
printer := NewTablePrinter(test.options)
|
||||||
|
printer.PrintObj(table, out)
|
||||||
|
// Validate the expected output matches the printed table.
|
||||||
|
if test.expected != out.String() {
|
||||||
|
t.Errorf("Table printing error: expected (%s), got (%s)", test.expected, out.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPrintTable_WithNamespace(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
columns []metav1.TableColumnDefinition
|
||||||
|
rows []metav1.TableRow
|
||||||
|
options PrintOptions
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
// Test a single row table "WithNamespace" option, prepends NAMESPACE column.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{
|
||||||
|
Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"},
|
||||||
|
Object: runtime.RawExtension{Object: testPod},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
// Print with namespace column prepended.
|
||||||
|
options: PrintOptions{WithNamespace: true},
|
||||||
|
expected: `NAMESPACE NAME READY STATUS RETRIES AGE
|
||||||
|
test-namespace test1 1/1 podPhase 5 20h
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
// Create the table from the columns and rows.
|
||||||
|
table := &metav1.Table{
|
||||||
|
ColumnDefinitions: test.columns,
|
||||||
|
Rows: test.rows,
|
||||||
|
}
|
||||||
|
// Print the table
|
||||||
|
out := bytes.NewBuffer([]byte{})
|
||||||
|
printer := NewTablePrinter(test.options)
|
||||||
|
printer.PrintObj(table, out)
|
||||||
|
// Validate the expected output matches the printed table.
|
||||||
|
if test.expected != out.String() {
|
||||||
|
t.Errorf("Table printing error: expected (%s), got (%s)", test.expected, out.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPrintTable_WithKind(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
columns []metav1.TableColumnDefinition
|
||||||
|
rows []metav1.TableRow
|
||||||
|
options PrintOptions
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
// Test a single row table "WithKind" option, prepends "pod" to name.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string", Format: "name"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{
|
||||||
|
Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
// Print with Kind "pod" prepended to name.
|
||||||
|
options: PrintOptions{WithKind: true, Kind: schema.GroupKind{Kind: "Pod"}},
|
||||||
|
expected: `NAME READY STATUS RETRIES AGE
|
||||||
|
pod/test1 1/1 podPhase 5 20h
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
// Create the table from the columns and rows.
|
||||||
|
table := &metav1.Table{
|
||||||
|
ColumnDefinitions: test.columns,
|
||||||
|
Rows: test.rows,
|
||||||
|
}
|
||||||
|
// Print the table
|
||||||
|
out := bytes.NewBuffer([]byte{})
|
||||||
|
printer := NewTablePrinter(test.options)
|
||||||
|
printer.PrintObj(table, out)
|
||||||
|
// Validate the expected output matches the printed table.
|
||||||
|
if test.expected != out.String() {
|
||||||
|
t.Errorf("Table printing error: expected (%s), got (%s)", test.expected, out.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPrintTable_WithLabels(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
columns []metav1.TableColumnDefinition
|
||||||
|
rows []metav1.TableRow
|
||||||
|
options PrintOptions
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
// Test a table "ShowLabels" option, appends labels as columns.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{
|
||||||
|
Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"},
|
||||||
|
Object: runtime.RawExtension{Object: testPod},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
options: PrintOptions{ShowLabels: true},
|
||||||
|
expected: `NAME READY STATUS RETRIES AGE LABELS
|
||||||
|
test1 1/1 podPhase 5 20h first-label=12,second-label=label-value
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
// Test a table "ColumnLabels" option, appends labels as columns.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{
|
||||||
|
Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"},
|
||||||
|
Object: runtime.RawExtension{Object: testPod},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
// Print "second-label" as column name, with label value.
|
||||||
|
options: PrintOptions{ColumnLabels: []string{"second-label"}},
|
||||||
|
expected: `NAME READY STATUS RETRIES AGE SECOND-LABEL
|
||||||
|
test1 1/1 podPhase 5 20h label-value
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
// Test a table "ColumnLabels" option, appends labels as columns.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{
|
||||||
|
Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"},
|
||||||
|
Object: runtime.RawExtension{Object: testPod},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
// Print multiple-labels as columns, with their values.
|
||||||
|
options: PrintOptions{ColumnLabels: []string{"first-label", "second-label"}},
|
||||||
|
expected: `NAME READY STATUS RETRIES AGE FIRST-LABEL SECOND-LABEL
|
||||||
|
test1 1/1 podPhase 5 20h 12 label-value
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
// Create the table from the columns and rows.
|
||||||
|
table := &metav1.Table{
|
||||||
|
ColumnDefinitions: test.columns,
|
||||||
|
Rows: test.rows,
|
||||||
|
}
|
||||||
|
// Print the table
|
||||||
|
out := bytes.NewBuffer([]byte{})
|
||||||
|
printer := NewTablePrinter(test.options)
|
||||||
|
printer.PrintObj(table, out)
|
||||||
|
// Validate the expected output matches the printed table.
|
||||||
|
if test.expected != out.String() {
|
||||||
|
t.Errorf("Table printing error: expected (%s), got (%s)", test.expected, out.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPrintTable_NonTable(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
object runtime.Object
|
||||||
|
options PrintOptions
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
// Test non-table default printing for a pod.
|
||||||
|
{
|
||||||
|
object: testPod,
|
||||||
|
options: PrintOptions{},
|
||||||
|
expected: "NAME AGE\ntest-pod-name <unknown>\n",
|
||||||
|
},
|
||||||
|
// Test non-table default printing for a pod with "NoHeaders" option.
|
||||||
|
{
|
||||||
|
object: testPod,
|
||||||
|
options: PrintOptions{NoHeaders: true},
|
||||||
|
expected: "test-pod-name <unknown>\n",
|
||||||
|
},
|
||||||
|
// Test non-table default printing for a pod with "WithNamespace" option.
|
||||||
|
{
|
||||||
|
object: testPod,
|
||||||
|
options: PrintOptions{WithNamespace: true},
|
||||||
|
expected: "NAMESPACE NAME AGE\ntest-namespace test-pod-name <unknown>\n",
|
||||||
|
},
|
||||||
|
// Test non-table default printing for a pod with "ShowLabels" option.
|
||||||
|
{
|
||||||
|
object: testPod,
|
||||||
|
options: PrintOptions{ShowLabels: true},
|
||||||
|
expected: `NAME AGE LABELS
|
||||||
|
test-pod-name <unknown> first-label=12,second-label=label-value
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
// Test non-table default printing for a Status resource.
|
||||||
|
{
|
||||||
|
object: testStatus,
|
||||||
|
options: PrintOptions{},
|
||||||
|
expected: `STATUS REASON MESSAGE
|
||||||
|
Failure test-status-reason test-status-message
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
// Test non-table default printing for a Status resource with NoHeaders options.
|
||||||
|
{
|
||||||
|
object: testStatus,
|
||||||
|
options: PrintOptions{NoHeaders: true},
|
||||||
|
expected: "Failure test-status-reason test-status-message\n",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
// Print the table
|
||||||
|
out := bytes.NewBuffer([]byte{})
|
||||||
|
printer := NewTablePrinter(test.options)
|
||||||
|
printer.PrintObj(test.object, out)
|
||||||
|
// Validate the expected output matches the printed table.
|
||||||
|
if test.expected != out.String() {
|
||||||
|
t.Errorf("Table printing error: expected (%s), got (%s)", test.expected, out.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPrintTable_WatchEvents(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
columns []metav1.TableColumnDefinition
|
||||||
|
rows []metav1.TableRow
|
||||||
|
options PrintOptions
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
// Print WatchEvent with Table resource.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{
|
||||||
|
Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"},
|
||||||
|
Object: runtime.RawExtension{Object: testPod},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
options: PrintOptions{},
|
||||||
|
expected: `EVENT NAME READY STATUS RETRIES AGE
|
||||||
|
Added test1 1/1 podPhase 5 20h
|
||||||
|
`,
|
||||||
|
},
|
||||||
|
// Print WatchEvent with Table, and "NoHeaders", "WithNamespace", and "ShowLabels" options.
|
||||||
|
{
|
||||||
|
columns: []metav1.TableColumnDefinition{
|
||||||
|
{Name: "Name", Type: "string"},
|
||||||
|
{Name: "Ready", Type: "string"},
|
||||||
|
{Name: "Status", Type: "string"},
|
||||||
|
{Name: "Retries", Type: "integer"},
|
||||||
|
{Name: "Age", Type: "string"},
|
||||||
|
},
|
||||||
|
rows: []metav1.TableRow{
|
||||||
|
{
|
||||||
|
Cells: []interface{}{"test1", "1/1", "podPhase", int64(5), "20h"},
|
||||||
|
Object: runtime.RawExtension{Object: testPod},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
options: PrintOptions{NoHeaders: true, WithNamespace: true, ShowLabels: true},
|
||||||
|
expected: "Added test-namespace test1 1/1 podPhase 5 20h first-label=12,second-label=label-value\n",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
// Create the table from the columns and rows.
|
||||||
|
table := &metav1.Table{
|
||||||
|
ColumnDefinitions: test.columns,
|
||||||
|
Rows: test.rows,
|
||||||
|
}
|
||||||
|
// Add the table to the WatchEvent.
|
||||||
|
event := &metav1.WatchEvent{
|
||||||
|
Type: "Added",
|
||||||
|
Object: runtime.RawExtension{Object: table},
|
||||||
|
}
|
||||||
|
// Print the event
|
||||||
|
out := bytes.NewBuffer([]byte{})
|
||||||
|
printer := NewTablePrinter(test.options)
|
||||||
|
printer.PrintObj(event, out)
|
||||||
|
// Validate the expected output matches the printed table.
|
||||||
|
if test.expected != out.String() {
|
||||||
|
t.Errorf("Event/Table printing error: expected (%s), got (%s)", test.expected, out.String())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user