mirror of https://github.com/k3s-io/k3s
104 lines
2.6 KiB
Go
104 lines
2.6 KiB
Go
|
/*
|
||
|
|
Copyright 2014 Google Inc. All rights reserved.
|
||
|
|
|
||
|
|
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 kubecfg
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
||
|
|
"gopkg.in/v1/yaml"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestYAMLPrinterPrint(t *testing.T) {
|
||
|
|
type testStruct struct {
|
||
|
|
Key string `yaml:"Key" json:"Key"`
|
||
|
|
Map map[string]int `yaml:"Map" json:"Map"`
|
||
|
|
StringList []string `yaml:"StringList" json:"StringList"`
|
||
|
|
IntList []int `yaml:"IntList" json:"IntList"`
|
||
|
|
}
|
||
|
|
testData := testStruct{
|
||
|
|
"testValue",
|
||
|
|
map[string]int{"TestSubkey": 1},
|
||
|
|
[]string{"a", "b", "c"},
|
||
|
|
[]int{1, 2, 3},
|
||
|
|
}
|
||
|
|
printer := &YAMLPrinter{}
|
||
|
|
buf := bytes.NewBuffer([]byte{})
|
||
|
|
|
||
|
|
err := printer.Print([]byte("invalidJSON"), buf)
|
||
|
|
if err == nil {
|
||
|
|
t.Error("Error: didn't fail on invalid JSON data")
|
||
|
|
}
|
||
|
|
|
||
|
|
jTestData, err := json.Marshal(&testData)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Unexpected error: couldn't marshal test data")
|
||
|
|
}
|
||
|
|
err = printer.Print(jTestData, buf)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
var poutput testStruct
|
||
|
|
err = yaml.Unmarshal(buf.Bytes(), &poutput)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if !reflect.DeepEqual(testData, poutput) {
|
||
|
|
t.Errorf("Test data and unmarshaled data are not equal: %#v vs %#v", poutput, testData)
|
||
|
|
}
|
||
|
|
|
||
|
|
obj := api.Pod{
|
||
|
|
JSONBase: api.JSONBase{ID: "foo"},
|
||
|
|
}
|
||
|
|
buf.Reset()
|
||
|
|
printer.PrintObj(obj, buf)
|
||
|
|
var objOut api.Pod
|
||
|
|
err = yaml.Unmarshal([]byte(buf.String()), &objOut)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Unexpeted error: %#v", err)
|
||
|
|
}
|
||
|
|
if !reflect.DeepEqual(obj, objOut) {
|
||
|
|
t.Errorf("Unexpected inequality: %#v vs %#v", obj, objOut)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIdentityPrinter(t *testing.T) {
|
||
|
|
printer := &IdentityPrinter{}
|
||
|
|
buff := bytes.NewBuffer([]byte{})
|
||
|
|
str := "this is a test string"
|
||
|
|
printer.Print([]byte(str), buff)
|
||
|
|
if buff.String() != str {
|
||
|
|
t.Errorf("Bytes are not equal: %s vs %s", str, buff.String())
|
||
|
|
}
|
||
|
|
|
||
|
|
obj := api.Pod{
|
||
|
|
JSONBase: api.JSONBase{ID: "foo"},
|
||
|
|
}
|
||
|
|
buff.Reset()
|
||
|
|
printer.PrintObj(obj, buff)
|
||
|
|
objOut, err := api.Decode([]byte(buff.String()))
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Unexpeted error: %#v", err)
|
||
|
|
}
|
||
|
|
if !reflect.DeepEqual(&obj, objOut) {
|
||
|
|
t.Errorf("Unexpected inequality: %#v vs %#v", obj, objOut)
|
||
|
|
}
|
||
|
|
}
|