Mirrors/flux2
1
0
Fork 0
mirror of synced 2026-02-13 13:06:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

add e2e test for check --pre with templating support

Browse Source 
Signed-off-by: Chanwit Kaewkasi <chanwit@gmail.com>
This commit is contained in:
Chanwit Kaewkasi
2021-08-17 20:46:20 +07:00
parent a3ba9817a3
commit cc32c1be07
3 changed files with 73 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
cmd/flux/main_test.go
View File

@@ -10,6 +10,7 @@ import (
"os"
"path/filepath"
"testing"
"text/template"
"time"
"github.com/fluxcd/flux2/internal/utils"
@@ -169,6 +170,8 @@ type cmdTestCase struct {
objectFile string
// TestClusterMode to bootstrap and testing, default to Fake
testClusterMode TestClusterMode
// TemplateValues enable template preprocessing for the golden file, or golden value
templateValues map[string]string
}
func (cmd *cmdTestCase) runTestCmd(t *testing.T) {
@@ -203,14 +206,28 @@ func (cmd *cmdTestCase) runTestCmd(t *testing.T) {
var expected string
if cmd.goldenValue != "" {
expected = cmd.goldenValue
if cmd.templateValues != nil {
expected, err = executeGoldenTemplate(cmd.goldenValue, cmd.templateValues)
if err != nil {
t.Fatalf("Error executing golden template: %s", err)
}
} else {
expected = cmd.goldenValue
}
}
if cmd.goldenFile != "" {
expectedOutput, err := os.ReadFile(cmd.goldenFile)
goldenFileContent, err := os.ReadFile(cmd.goldenFile)
if err != nil {
t.Fatalf("Error reading golden file: '%s'", err)
}
expected = string(expectedOutput)
if cmd.templateValues != nil {
expected, err = executeGoldenTemplate(string(goldenFileContent), cmd.templateValues)
if err != nil {
t.Fatalf("Error executing golden template file '%s': %s", cmd.goldenFile, err)
}
} else {
expected = string(goldenFileContent)
}
}
diff := cmp.Diff(expected, actual)
@@ -219,6 +236,15 @@ func (cmd *cmdTestCase) runTestCmd(t *testing.T) {
}
}
func executeGoldenTemplate(goldenValue string, templateValues map[string]string) (string, error) {
tmpl := template.Must(template.New("golden").Parse(goldenValue))
var out bytes.Buffer
if err := tmpl.Execute(&out, templateValues); err != nil {
return "", err
}
return out.String(), nil
}
// Run the command and return the captured output.
func executeCommand(cmd string) (string, error) {
args, err := shellwords.Parse(cmd)