Function TestYAMLHelper

policy/yaml_test.go:23–174

(t *testing.T)

)

func TestYAMLHelper(t *testing.T) {
	helper := YAMLHelper{}
	tests := []struct {
		name string
		yaml string
		want func(*testing.T, *yaml.Node)
	}{
		{
			name: "list",
			yaml: `
- first
- second
- third`,
			want: func(t *testing.T, node *yaml.Node) {
				if !helper.IsList(node) {
					t.Fatalf("IsList(node) returned false, want true")
				}

				helper.RangeList(node, func(i int, val *yaml.Node) bool {
					switch i {
					case 0:
						if !helper.IsString(val) || val.Value != "first" {
							t.Errorf("val[%d] = %q, want %q", i, val.Value, "first")
						}
					case 1:
						if !helper.IsString(val) || val.Value != "second" {
							t.Errorf("val[%d] = %q, want %q", i, val.Value, "second")
						}
					case 2:
						if !helper.IsString(val) || val.Value != "third" {
							t.Errorf("val[%d] = %q, want %q", i, val.Value, "third")
						}
					}
					return true
				})
			},
		},
		{
			name: "map",
			yaml: `
first: 1
second: 2
third: 3`,
			want: func(t *testing.T, node *yaml.Node) {
				if !helper.IsMap(node) {
					t.Fatalf("IsMap(node) returned false, want true")
				}
				helper.RangeMap(node, func(key *yaml.Node, val *yaml.Node) bool {
					if key.Value == "first" && helper.IsInteger(val) && val.Value != "1" {
						t.Errorf("val[%q] = %q, want %q", key.Value, val.Value, "1")
					}
					if key.Value == "second" && helper.IsInteger(val) && val.Value != "2" {
						t.Errorf("val[%q] = %q, want %q", key.Value, val.Value, "2")
					}
					if key.Value == "third" && helper.IsInteger(val) && val.Value != "3" {
						t.Errorf("val[%q] = %q, want %q", key.Value, val.Value, "3")
					}
					return true