| /* |
| Copyright 2018 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 printers |
| |
| import ( |
| "bytes" |
| "strings" |
| "testing" |
| |
| "k8s.io/api/core/v1" |
| "k8s.io/apimachinery/pkg/runtime" |
| ) |
| |
| func TestTemplate(t *testing.T) { |
| testCase := []struct { |
| name string |
| template string |
| obj runtime.Object |
| expectOut string |
| expectErr func(error) (string, bool) |
| }{ |
| { |
| name: "support base64 decoding of secret data", |
| template: "{{ .data.username | base64decode }}", |
| obj: &v1.Secret{ |
| Data: map[string][]byte{ |
| "username": []byte("hunter"), |
| }, |
| }, |
| expectOut: "hunter", |
| }, |
| { |
| name: "test error path for base64 decoding", |
| template: "{{ .data.username | base64decode }}", |
| obj: &badlyMarshaledSecret{}, |
| expectErr: func(err error) (string, bool) { |
| matched := strings.Contains(err.Error(), "base64 decode") |
| return "a base64 decode error", matched |
| }, |
| }, |
| } |
| for _, test := range testCase { |
| t.Run(test.name, func(t *testing.T) { |
| buffer := &bytes.Buffer{} |
| |
| p, err := NewGoTemplatePrinter([]byte(test.template)) |
| if err != nil { |
| if test.expectErr == nil { |
| t.Errorf("[%s]expected success but got:\n %v\n", test.name, err) |
| return |
| } |
| if expected, ok := test.expectErr(err); !ok { |
| t.Errorf("[%s]expect:\n %v\n but got:\n %v\n", test.name, expected, err) |
| } |
| return |
| } |
| |
| err = p.PrintObj(test.obj, buffer) |
| if err != nil { |
| if test.expectErr == nil { |
| t.Errorf("[%s]expected success but got:\n %v\n", test.name, err) |
| return |
| } |
| if expected, ok := test.expectErr(err); !ok { |
| t.Errorf("[%s]expect:\n %v\n but got:\n %v\n", test.name, expected, err) |
| } |
| return |
| } |
| |
| if test.expectErr != nil { |
| t.Errorf("[%s]expect:\n error\n but got:\n no error\n", test.name) |
| return |
| } |
| |
| if test.expectOut != buffer.String() { |
| t.Errorf("[%s]expect:\n %v\n but got:\n %v\n", test.name, test.expectOut, buffer.String()) |
| } |
| }) |
| } |
| } |
| |
| type badlyMarshaledSecret struct { |
| v1.Secret |
| } |
| |
| func (a badlyMarshaledSecret) MarshalJSON() ([]byte, error) { |
| return []byte(`{"apiVersion":"v1","data":{"username":"--THIS IS NOT BASE64--"},"kind":"Secret"}`), nil |
| } |