76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
// +-------------------------------------------------------------------------
|
|
// | Copyright (C) 2016 Yunify, Inc.
|
|
// +-------------------------------------------------------------------------
|
|
// | Licensed under the Apache License, Version 2.0 (the "License");
|
|
// | you may not use this work except in compliance with the License.
|
|
// | You may obtain a copy of the License in the LICENSE file, or 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 utils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestYAMLDecode_Unknown(t *testing.T) {
|
|
yamlString := `
|
|
key1: "This is a string." # Single Line Comment
|
|
key2: 10.50
|
|
key3:
|
|
- null
|
|
- nestedKey1: Anothor string
|
|
`
|
|
|
|
anyData, err := YAMLDecode([]byte(yamlString))
|
|
assert.Nil(t, err)
|
|
data := anyData.(map[interface{}]interface{})
|
|
assert.Equal(t, 10.50, data["key2"])
|
|
}
|
|
|
|
func TestYAMLDecode_Known(t *testing.T) {
|
|
type SampleYAML struct {
|
|
Name string `yaml:"name"`
|
|
Description string `yaml:"description"`
|
|
}
|
|
sampleYAMLString := `name: "NAME"`
|
|
|
|
sample := SampleYAML{Name: "NaMe", Description: "DeScRiPtIoN"}
|
|
anyDataPointer, err := YAMLDecode([]byte(sampleYAMLString), &sample)
|
|
assert.Nil(t, err)
|
|
data := anyDataPointer.(*SampleYAML)
|
|
assert.Equal(t, "NAME", sample.Name)
|
|
assert.Equal(t, "DeScRiPtIoN", sample.Description)
|
|
assert.Equal(t, "NAME", (*data).Name)
|
|
assert.Equal(t, "DeScRiPtIoN", (*data).Description)
|
|
}
|
|
|
|
func TestYAMLDecode_Empty(t *testing.T) {
|
|
yamlString := ""
|
|
|
|
anyData, err := YAMLDecode([]byte(yamlString))
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, anyData)
|
|
}
|
|
|
|
func TestYAMLEncode(t *testing.T) {
|
|
type SampleYAML struct {
|
|
Name string `yaml:"name"`
|
|
Description string `yaml:"description"`
|
|
}
|
|
sample := SampleYAML{Name: "NaMe", Description: "DeScRiPtIoN"}
|
|
|
|
yamlBytes, err := YAMLEncode(sample)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "name: NaMe\ndescription: DeScRiPtIoN\n", string(yamlBytes))
|
|
}
|