go-variables/parser_yaml_test.go at main · d7z-team/go-variables · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package variables

import (
	"fmt"
	"strings"
	"testing"

	"github.com/goccy/go-yaml"

	"github.com/stretchr/testify/assert"
)

func TestFromYaml(t *testing.T) {
	variables := NewVariables()
	err := variables.FromYaml(`
metadata:
  name: dragon
data:
 - "{{.metadata.name}}"
 - bbb
`, "")

	assert.NoError(t, err)
	s, err := variables.Template("{{index .data 0 }}")
	assert.NoError(t, err)
	assert.Equal(t, "dragon", s)
}

func TestBigFromYaml(t *testing.T) {
	variables := NewVariables()
	err := variables.FromYaml(`
metadata:
  name: dragon
data:
 - "{{.metadata.name}}"
 - name: dragon
   value: test
 - bbb
`, "")
	marshal, err := yaml.Marshal(variables)
	fmt.Printf("%s\n", string(marshal))
	assert.NoError(t, err)
}

func TestVariables_FromYamlFilter(t *testing.T) {
	variables := NewVariables()
	err := variables.FromYamlFilter(`
metadata:
  name: dragon
  version: 1.0.0
`, "", func(key string) bool {
		return strings.HasPrefix(key, "metadata.")
	})
	assert.NoError(t, err)
	s, err := variables.Template("{{.metadata.name }}-{{.metadata.version}}")
	assert.NoError(t, err)
	assert.Equal(t, "dragon-1.0.0", s)
}

func TestVariables_Get(t *testing.T) {
	variables := NewVariables()
	err := variables.FromYaml(`
metadata:
  name: dragon
  age: false
data:
 - "{{.metadata.name}}"
 - bbb
 - ccc:
    name: "data"
`, "")
	assert.NoError(t, err)
	assert.Equal(t, variables.Get("metadata.name"), "dragon")
	assert.Equal(t, variables.Get("metadata.age"), false)
	assert.Equal(t, variables.Get("data.0"), "dragon")
	assert.Equal(t, variables.Get("data.2.ccc.name"), "data")
	assert.NotNil(t, variables.Get("data.-1"))
}