FilesExpand file tree

master

/

kubectl_kustomize_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

169 lines (145 loc) · 5.42 KB

master

/

kubectl_kustomize_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

169 lines (145 loc) · 5.42 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

package chartify

import (

"os"

"path/filepath"

"testing"

"github.com/stretchr/testify/require"

)

// stubKubectlScript is a minimal sh script that acts as a kubectl stub for tests.

// It handles `kubectl kustomize <dir> [-o|-o FILE|--output FILE]` by writing

// a minimal valid Kubernetes Deployment YAML to the specified output file.

const stubKubectlScript = `#!/bin/sh

if [ "$1" = "kustomize" ]; then

shift

OUTPUT=""

while [ $# -gt 0 ]; do

case "$1" in

-o) OUTPUT="$2"; shift 2;;

--output) OUTPUT="$2"; shift 2;;

*) shift;;

esac

done

if [ -n "$OUTPUT" ]; then

printf 'apiVersion: apps/v1\nkind: Deployment\nmetadata:\n name: stub\n' > "$OUTPUT"

fi

fi

`

// writeStubKubectl creates a stub kubectl script in dir.

func writeStubKubectl(t *testing.T, dir string) {

t.Helper()

p := filepath.Join(dir, "kubectl")

require.NoError(t, os.WriteFile(p, []byte(stubKubectlScript), 0755))

}

// TestKubectlKustomize tests behavior when kubectl kustomize is explicitly configured

// via KustomizeBin("kubectl kustomize"). The automatic fallback selection is tested

// in TestKustomizeBin.

func TestKubectlKustomize(t *testing.T) {

t.Run("KustomizeBuild succeeds with kubectl kustomize option", func(t *testing.T) {

// Create a stub kubectl so the test is self-contained and always runs in CI.

stubDir := t.TempDir()

writeStubKubectl(t, stubDir)

origPath := os.Getenv("PATH")

defer os.Setenv("PATH", origPath)

os.Setenv("PATH", stubDir+string(os.PathListSeparator)+origPath)

tmpDir := t.TempDir()

srcDir := t.TempDir()

kustomizationContent := `apiVersion: kustomize.config.k8s.io/v1beta1

kind: Kustomization

resources:

- deployment.yaml

`

deploymentContent := `apiVersion: apps/v1

kind: Deployment

metadata:

name: test

spec:

replicas: 1

selector:

matchLabels:

app: test

template:

metadata:

labels:

app: test

spec:

containers:

- name: test

image: test:latest

`

templatesDir := filepath.Join(tmpDir, "templates")

require.NoError(t, os.MkdirAll(templatesDir, 0755))

require.NoError(t, os.WriteFile(filepath.Join(srcDir, "kustomization.yaml"), []byte(kustomizationContent), 0644))

require.NoError(t, os.WriteFile(filepath.Join(srcDir, "deployment.yaml"), []byte(deploymentContent), 0644))

r := New(KustomizeBin("kubectl kustomize"))

outputFile, err := r.KustomizeBuild(srcDir, tmpDir)

require.NoError(t, err)

require.FileExists(t, outputFile)

})

t.Run("Patch succeeds with kubectl kustomize option", func(t *testing.T) {

// Create a stub kubectl so the test is self-contained and always runs in CI.

stubDir := t.TempDir()

writeStubKubectl(t, stubDir)

origPath := os.Getenv("PATH")

defer os.Setenv("PATH", origPath)

os.Setenv("PATH", stubDir+string(os.PathListSeparator)+origPath)

tempDir := t.TempDir()

// Write a minimal manifest file that Patch() will reference.

manifestPath := filepath.Join(tempDir, "templates", "deploy.yaml")

require.NoError(t, os.MkdirAll(filepath.Dir(manifestPath), 0755))

require.NoError(t, os.WriteFile(manifestPath, []byte(`apiVersion: apps/v1

kind: Deployment

metadata:

name: test

`), 0644))

r := New(KustomizeBin("kubectl kustomize"))

err := r.Patch(tempDir, []string{manifestPath}, &PatchOpts{})

require.NoError(t, err)

})

t.Run("edit commands not supported with kubectl kustomize", func(t *testing.T) {

srcDir := t.TempDir()

require.NoError(t, os.WriteFile(filepath.Join(srcDir, "kustomization.yaml"), []byte("apiVersion: kustomize.config.k8s.io/v1beta1\nkind: Kustomization\n"), 0644))

require.NoError(t, os.WriteFile(filepath.Join(srcDir, "deployment.yaml"), []byte("apiVersion: apps/v1\nkind: Deployment\nmetadata:\n name: test\n"), 0644))

cases := []struct {

name string

valuesContent string

wantErrSubstr string

}{

{

name: "images",

valuesContent: "images:\n- name: test\n newName: newtest\n newTag: v2\n",

wantErrSubstr: "setting images via Chartify values files or kustomize build options is not supported when using 'kubectl kustomize'",

},

{

name: "namePrefix",

valuesContent: "namePrefix: prod-\n",

wantErrSubstr: "setting namePrefix via Chartify values files or kustomize build options is not supported when using 'kubectl kustomize'",

},

{

name: "nameSuffix",

valuesContent: "nameSuffix: -v2\n",

wantErrSubstr: "setting nameSuffix via Chartify values files or kustomize build options is not supported when using 'kubectl kustomize'",

},

{

name: "namespace",

valuesContent: "namespace: prod\n",

wantErrSubstr: "setting namespace via Chartify values files or kustomize build options is not supported when using 'kubectl kustomize'",

},

}

for _, tc := range cases {

t.Run(tc.name, func(t *testing.T) {

tmpDir := t.TempDir()

valuesDir := t.TempDir()

valuesFile := filepath.Join(valuesDir, "values.yaml")

require.NoError(t, os.WriteFile(valuesFile, []byte(tc.valuesContent), 0644))

r := New(KustomizeBin("kubectl kustomize"))

_, err := r.KustomizeBuild(srcDir, tmpDir, &KustomizeBuildOpts{ValuesFiles: []string{valuesFile}})

require.Error(t, err)

require.Contains(t, err.Error(), tc.wantErrSubstr)

// Validation happens before file I/O, so tempDir must be left clean.

entries, err := os.ReadDir(tmpDir)

require.NoError(t, err)

require.Empty(t, entries, "tempDir should be empty when validation fails before I/O")

})

}

})

}