Skip to content

Commit b1c9c9e

Browse files
committed
refactor
Signed-off-by: zirain <[email protected]>
1 parent a45a328 commit b1c9c9e

5 files changed

+8
-11
lines changed

internal/utils/merge_test.go

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
package utils
77

88
import (
9-
"fmt"
109
"os"
1110
"path/filepath"
1211
"strings"
@@ -21,16 +20,15 @@ import (
2120
)
2221

2322
func TestMergeBackendTrafficPolicy(t *testing.T) {
24-
baseDir := "testdata/merge/backendtrafficpolicy"
25-
caseFiles, err := filepath.Glob(filepath.Join(baseDir, "*.in.yaml"))
23+
baseDir := "testdata"
24+
caseFiles, err := filepath.Glob(filepath.Join(baseDir, "backendtrafficpolicy_*.in.yaml"))
2625
require.NoError(t, err)
2726

2827
for _, caseFile := range caseFiles {
2928
// get case name from path
30-
caseName := strings.TrimPrefix(strings.TrimSuffix(caseFile, ".in.yaml"), baseDir+"/")
31-
32-
for _, mergeType := range []egv1a1.MergeType{egv1a1.StrategicMerge, egv1a1.JSONMerge} {
33-
t.Run(fmt.Sprintf("%s/%s", mergeType, caseName), func(t *testing.T) {
29+
caseName := strings.TrimPrefix(strings.TrimSuffix(caseFile, ".in.yaml"), baseDir+"/backendtrafficpolicy_")
30+
t.Run(caseName, func(t *testing.T) {
31+
for _, mergeType := range []egv1a1.MergeType{egv1a1.StrategicMerge, egv1a1.JSONMerge} {
3432
patchedInput := strings.Replace(caseFile, ".in.yaml", ".patch.yaml", 1)
3533
var output string
3634
if mergeType == egv1a1.StrategicMerge {
@@ -49,14 +47,13 @@ func TestMergeBackendTrafficPolicy(t *testing.T) {
4947
b, err := yaml.Marshal(got)
5048
require.NoError(t, err)
5149
require.NoError(t, os.WriteFile(output, b, 0o600))
52-
return
50+
continue
5351
}
5452

5553
expected := readObject[*egv1a1.BackendTrafficPolicy](t, output)
5654
require.Equal(t, expected, got)
57-
})
58-
}
59-
55+
}
56+
})
6057
}
6158
}
6259

0 commit comments

Comments
 (0)