mirror of https://github.com/mikefarah/yq.git
42 lines
670 B
Go
42 lines
670 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"gopkg.in/yaml.v2"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
var raw_data = `
|
|
a: Easy!
|
|
b:
|
|
c: 2
|
|
d: [3, 4]
|
|
`
|
|
|
|
var parsed_data map[interface{}]interface{}
|
|
|
|
func TestMain(m *testing.M) {
|
|
err := yaml.Unmarshal([]byte(raw_data), &parsed_data)
|
|
if err != nil {
|
|
fmt.Println("Error parsing yaml: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
func TestRead_map_simple(t *testing.T) {
|
|
result := read_map(parsed_data, "b", []string{"c"})
|
|
if result != 2 {
|
|
t.Error("Excpted 2 but got ", result)
|
|
}
|
|
}
|
|
|
|
func TestRead_map_array(t *testing.T) {
|
|
result := read_map(parsed_data, "b", []string{"d", "1"})
|
|
if result != 4 {
|
|
t.Error("Excpted 4 but got ", result)
|
|
}
|
|
}
|