add tests for kustomize config name handling

Signed-off-by: crabique <[email protected]>
fluxcd · Mar 10, 2025 · 9a8e210 · 9a8e210
1 parent 3ae8936
commit 9a8e210
Show file tree

Hide file tree

Showing 6 changed files with 50 additions and 9 deletions.
diff --git a/pkg/update/filereader_test.go b/pkg/update/filereader_test.go
@@ -33,21 +33,23 @@ func TestScreeningLocalReader(t *testing.T) {
 	}
 	nodes, err := r.Read()
 	g.Expect(err).ToNot(HaveOccurred())
-	// the test fixture has three files that contain the marker:
-	// - otherns.yaml
+	// the test fixture has four files that contain the marker:
 	// - marked.yaml
-	// - kustomization.yaml
-	g.Expect(len(nodes)).To(Equal(3))
+	// - otherns.yaml
+	// - kustomization.yml
+	// - Kustomization
+	g.Expect(len(nodes)).To(Equal(4))
 	filesSeen := map[string]struct{}{}
 	for i := range nodes {
 		path, _, err := kioutil.GetFileAnnotations(nodes[i])
 		g.Expect(err).ToNot(HaveOccurred())
 		filesSeen[path] = struct{}{}
 	}
 	g.Expect(filesSeen).To(Equal(map[string]struct{}{
-		"marked.yaml":        {},
-		"kustomization.yaml": {},
-		"otherns.yaml":       {},
+		"marked.yaml":       {},
+		"otherns.yaml":      {},
+		"kustomization.yml": {},
+		"Kustomization":     {},
 	}))
 
 }
diff --git a/pkg/update/testdata/setters/expected/Kustomization b/pkg/update/testdata/setters/expected/Kustomization
@@ -0,0 +1,9 @@
+# This is not intended to be a working kustomization
+apiVersion: kustomize.config.k8s.io/v1beta1
+kind: Kustomization
+resources:
+  - irrelevant.yaml
+images:
+  - name: container
+    newName: index.repo.fake/updated # {"$imagepolicy": "automation-ns:policy:name"}
+    newTag: v1.0.1 # {"$imagepolicy": "automation-ns:policy:tag"}
diff --git a/...tdata/setters/expected/kustomization.yaml → ...stdata/setters/expected/kustomization.yml b/...tdata/setters/expected/kustomization.yaml → ...stdata/setters/expected/kustomization.yml
diff --git a/pkg/update/testdata/setters/original/Kustomization b/pkg/update/testdata/setters/original/Kustomization
@@ -0,0 +1,9 @@
+# This is not intended to be a working kustomization
+apiVersion: kustomize.config.k8s.io/v1beta1
+kind: Kustomization
+resources:
+  - irrelevant.yaml
+images:
+  - name: container
+    newName: replaced # {"$imagepolicy": "automation-ns:policy:name"}
+    newTag: v1 # {"$imagepolicy": "automation-ns:policy:tag"}
diff --git a/...tdata/setters/original/kustomization.yaml → ...stdata/setters/original/kustomization.yml b/...tdata/setters/original/kustomization.yaml → ...stdata/setters/original/kustomization.yml
diff --git a/pkg/update/update_test.go b/pkg/update/update_test.go
@@ -83,7 +83,14 @@ func TestUpdateWithSetters(t *testing.T) {
 
 	expectedResult := Result{
 		Files: map[string]FileResult{
-			"kustomization.yaml": {
+			"kustomization.yml": {
+				Objects: map[ObjectIdentifier][]ImageRef{
+					kustomizeResourceID: {
+						expectedImageRef,
+					},
+				},
+			},
+			"Kustomization": {
 				Objects: map[ObjectIdentifier][]ImageRef{
 					kustomizeResourceID: {
 						expectedImageRef,
@@ -111,7 +118,21 @@ func TestUpdateWithSetters(t *testing.T) {
 	expectedResultV2 := ResultV2{
 		ImageResult: expectedResult,
 		FileChanges: map[string]ObjectChanges{
-			"kustomization.yaml": {
+			"kustomization.yml": {
+				kustomizeResourceID: []Change{
+					{
+						OldValue: "replaced",
+						NewValue: "index.repo.fake/updated",
+						Setter:   "automation-ns:policy:name",
+					},
+					{
+						OldValue: "v1",
+						NewValue: "v1.0.1",
+						Setter:   "automation-ns:policy:tag",
+					},
+				},
+			},
+			"Kustomization": {
 				kustomizeResourceID: []Change{
 					{
 						OldValue: "replaced",