public_git/helmfile
1
0
Fork 0
mirror of https://github.com/helmfile/helmfile.git synced 2026-10-22 07:20:03 +02:00
Code Issues Projects Releases Wiki Activity

feat: include func support (#1187)

Browse Source 
* feat: include func support

Signed-off-by: yxxhero <aiopsclub@163.com>
This commit is contained in:
yxxhero
2023-12-04 21:51:01 +08:00
committed by GitHub
parent 06504477f6
commit 7d6ed97333
20 changed files with 380 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/tmpl/context_tmpl_test.go
+120 -27
View File
@@ -16,12 +16,16 @@ func TestRenderTemplate_Values(t *testing.T) {
bar: FOO_BAR
`
expectedFilename := "values.yaml"
ctx := &Context{fs: &ffs.FileSystem{ReadFile: func(filename string) ([]byte, error) {
if filename != expectedFilename {
return nil, fmt.Errorf("unexpected filename: expected=%v, actual=%s", expectedFilename, filename)
}
return []byte(valuesYamlContent), nil
}}}
ctx := &Context{fs: &ffs.FileSystem{
Glob: func(s string) ([]string, error) {
return nil, nil
},
ReadFile: func(filename string) ([]byte, error) {
if filename != expectedFilename {
return nil, fmt.Errorf("unexpected filename: expected=%v, actual=%s", expectedFilename, filename)
}
return []byte(valuesYamlContent), nil
}}}
buf, err := ctx.RenderTemplateToBuffer(`{{ readFile "values.yaml" | fromYaml | setValueAtPath "foo.bar" "FOO_BAR" | toYaml }}`)
if err != nil {
t.Errorf("unexpected error: %v", err)
@@ -45,12 +49,16 @@ func TestRenderTemplate_WithData(t *testing.T) {
"bar": "FOO_BAR",
},
}
ctx := &Context{fs: &ffs.FileSystem{ReadFile: func(filename string) ([]byte, error) {
if filename != expectedFilename {
return nil, fmt.Errorf("unexpected filename: expected=%v, actual=%s", expectedFilename, filename)
}
return []byte(valuesYamlContent), nil
}}}
ctx := &Context{fs: &ffs.FileSystem{
Glob: func(s string) ([]string, error) {
return nil, nil
},
ReadFile: func(filename string) ([]byte, error) {
if filename != expectedFilename {
return nil, fmt.Errorf("unexpected filename: expected=%v, actual=%s", expectedFilename, filename)
}
return []byte(valuesYamlContent), nil
}}}
buf, err := ctx.RenderTemplateToBuffer(valuesYamlContent, data)
if err != nil {
t.Errorf("unexpected error: %v", err)
@@ -70,12 +78,16 @@ func TestRenderTemplate_AccessingMissingKeyWithGetOrNil(t *testing.T) {
`
expectedFilename := "values.yaml"
data := map[string]any{}
ctx := &Context{fs: &ffs.FileSystem{ReadFile: func(filename string) ([]byte, error) {
if filename != expectedFilename {
return nil, fmt.Errorf("unexpected filename: expected=%v, actual=%s", expectedFilename, filename)
}
return []byte(valuesYamlContent), nil
}}}
ctx := &Context{fs: &ffs.FileSystem{
Glob: func(s string) ([]string, error) {
return nil, nil
},
ReadFile: func(filename string) ([]byte, error) {
if filename != expectedFilename {
return nil, fmt.Errorf("unexpected filename: expected=%v, actual=%s", expectedFilename, filename)
}
return []byte(valuesYamlContent), nil
}}}
buf, err := ctx.RenderTemplateToBuffer(valuesYamlContent, data)
if err != nil {
t.Errorf("unexpected error: %v", err)
@@ -95,12 +107,16 @@ func TestRenderTemplate_Defaulting(t *testing.T) {
`
expectedFilename := "values.yaml"
data := map[string]any{}
ctx := &Context{fs: &ffs.FileSystem{ReadFile: func(filename string) ([]byte, error) {
if filename != expectedFilename {
return nil, fmt.Errorf("unexpected filename: expected=%v, actual=%s", expectedFilename, filename)
}
return []byte(valuesYamlContent), nil
}}}
ctx := &Context{fs: &ffs.FileSystem{
Glob: func(s string) ([]string, error) {
return nil, nil
},
ReadFile: func(filename string) ([]byte, error) {
if filename != expectedFilename {
return nil, fmt.Errorf("unexpected filename: expected=%v, actual=%s", expectedFilename, filename)
}
return []byte(valuesYamlContent), nil
}}}
buf, err := ctx.RenderTemplateToBuffer(valuesYamlContent, data)
if err != nil {
t.Errorf("unexpected error: %v", err)
@@ -112,9 +128,13 @@ func TestRenderTemplate_Defaulting(t *testing.T) {
}
func renderTemplateToString(s string, data ...any) (string, error) {
ctx := &Context{fs: &ffs.FileSystem{ReadFile: func(filename string) ([]byte, error) {
return nil, fmt.Errorf("unexpected call to readFile: filename=%s", filename)
}}}
ctx := &Context{fs: &ffs.FileSystem{
Glob: func(s string) ([]string, error) {
return nil, nil
},
ReadFile: func(filename string) ([]byte, error) {
return nil, fmt.Errorf("unexpected call to readFile: filename=%s", filename)
}}}
tplString, err := ctx.RenderTemplateToBuffer(s, data...)
if err != nil {
return "", err
@@ -326,3 +346,76 @@ func TestRenderTemplate_Required(t *testing.T) {
}
}
}
func TestContext_helperTPLs(t *testing.T) {
c := &Context{
fs: &ffs.FileSystem{
Glob: func(s string) ([]string, error) {
return []string{
"/helmfiletmpl/_template1.tpl",
"/helmfiletmpl/_template2.tpl",
}, nil
},
ReadFile: func(filename string) ([]byte, error) {
switch filename {
case "/helmfiletmpl/_template1.tpl":
return []byte("Template 1 content"), nil
case "/helmfiletmpl/_template2.tpl":
return []byte("Template 2 content"), nil
default:
return nil, fmt.Errorf("unexpected filename: %s", filename)
}
},
},
}
want := []tplInfo{
{
name: "/helmfiletmpl/_template1.tpl",
content: "Template 1 content",
},
{
name: "/helmfiletmpl/_template2.tpl",
content: "Template 2 content",
},
}
got, err := c.helperTPLs()
if err != nil {
t.Errorf("unexpected error: %v", err)
}
if !reflect.DeepEqual(got, want) {
t.Errorf("unexpected result: got=%v, want=%v", got, want)
}
}
func TestContext_RenderTemplateToBuffer(t *testing.T) {
c := &Context{
basePath: "/helmfile",
fs: &ffs.FileSystem{
Glob: func(s string) ([]string, error) {
return []string{
"/helmfile/_template1.tpl",
}, nil
},
ReadFile: func(filename string) ([]byte, error) {
if filename == "/helmfile/_template1.tpl" {
return []byte("{{- define \"name\" -}}\n{{ .Name }}\n{{- end }}"), nil
}
return nil, fmt.Errorf("unexpected filename: %s", filename)
},
},
}
s := "Hello, {{ include \"name\" . }}!"
data := map[string]interface{}{
"Name": "Alice",
}
expected := "Hello, Alice!"
buf, err := c.RenderTemplateToBuffer(s, data)
if err != nil {
t.Errorf("unexpected error: %v", err)
}
actual := buf.String()
if actual != expected {
t.Errorf("unexpected result: expected=%s, actual=%s", expected, actual)
}
}