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
79
80
81
82
83
84
85
86
87
88
89
90
|
// Copyright 2019 The Kubernetes Authors.
// SPDX-License-Identifier: Apache-2.0
package e2e
import (
"bytes"
"os"
"os/exec"
"path/filepath"
"strings"
"testing"
"github.com/stretchr/testify/assert"
"sigs.k8s.io/kustomize/kyaml/testutil"
)
func TestMain(m *testing.M) {
var code int
func() {
d := build()
defer os.RemoveAll(d)
code = m.Run()
}()
os.Exit(code)
}
type test struct {
name string
args []string
files map[string]string
expectedFiles map[string]string
expectedErr string
expectedStdOut string
}
func runTests(t *testing.T, tests []test) {
t.Helper()
dir := build()
bin := filepath.Join(dir, kyamlBin)
for i := range tests {
tt := tests[i]
t.Run(tt.name, func(t *testing.T) {
dataDir, err := os.MkdirTemp("", "kustomize-test-data-")
if !assert.NoError(t, err) {
t.FailNow()
}
defer os.RemoveAll(dataDir)
os.Chdir(dataDir)
// write the input
for path, data := range tt.files {
err := os.WriteFile(path, []byte(data), 0600)
testutil.AssertNoError(t, err)
}
cmd := exec.Command(bin, tt.args...)
cmd.Dir = dataDir
var stdErr, stdOut bytes.Buffer
cmd.Stdout = &stdOut
cmd.Stderr = &stdErr
cmd.Env = os.Environ()
err = cmd.Run()
if tt.expectedErr != "" {
if !assert.Contains(t, stdErr.String(), tt.expectedErr, stdErr.String()) {
t.FailNow()
}
return
}
testutil.AssertNoError(t, err, stdErr.String(), stdOut.String())
if tt.expectedStdOut != "" {
if !assert.Equal(t, strings.TrimSpace(stdOut.String()), strings.TrimSpace(tt.expectedStdOut)) {
t.FailNow()
}
}
for path, data := range tt.expectedFiles {
b, err := os.ReadFile(path)
testutil.AssertNoError(t, err, stdErr.String())
if !assert.Equal(t, strings.TrimSpace(data), strings.TrimSpace(string(b)), stdErr.String()) {
t.FailNow()
}
}
})
}
}
|
