Add test for static volumes

And moved a test from deployment_test that was not the right place to be created.
metal3d · Apr 23, 2024 · 5568c38 · 5568c38
1 parent c01cdf5
commit 5568c38
Show file tree

Hide file tree

Showing 2 changed files with 100 additions and 29 deletions.
diff --git a/generator/deployment_test.go b/generator/deployment_test.go
@@ -38,32 +38,3 @@ services:
 		t.Errorf("Expected image to be nginx:1.29, got %s", dt.Spec.Template.Spec.Containers[0].Image)
 	}
 }
-
-func TestGenerateWithBoundVolume(t *testing.T) {
-	_compose_file := `
-services:
-    web:
-        image: nginx:1.29
-        volumes:
-        - data:/var/www
-volumes:
-    data:
-`
-	tmpDir := setup(_compose_file)
-	defer teardown(tmpDir)
-
-	currentDir, _ := os.Getwd()
-	os.Chdir(tmpDir)
-	defer os.Chdir(currentDir)
-
-	output := _compile_test(t, "-s", "templates/web/deployment.yaml")
-
-	dt := v1.Deployment{}
-	if err := yaml.Unmarshal([]byte(output), &dt); err != nil {
-		t.Errorf("Failed to unmarshal the output: %s", err)
-	}
-
-	if dt.Spec.Template.Spec.Containers[0].VolumeMounts[0].Name != "data" {
-		t.Errorf("Expected volume name to be data: %v", dt)
-	}
-}
diff --git a/generator/volume_test.go b/generator/volume_test.go
@@ -0,0 +1,100 @@
+package generator
+
+import (
+	"fmt"
+	"os"
+	"testing"
+
+	v1 "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
+	"sigs.k8s.io/yaml"
+)
+
+func TestGenerateWithBoundVolume(t *testing.T) {
+	_compose_file := `
+services:
+    web:
+        image: nginx:1.29
+        volumes:
+        - data:/var/www
+volumes:
+    data:
+`
+	tmpDir := setup(_compose_file)
+	defer teardown(tmpDir)
+
+	currentDir, _ := os.Getwd()
+	os.Chdir(tmpDir)
+	defer os.Chdir(currentDir)
+
+	output := _compile_test(t, "-s", "templates/web/deployment.yaml")
+
+	dt := v1.Deployment{}
+	if err := yaml.Unmarshal([]byte(output), &dt); err != nil {
+		t.Errorf("Failed to unmarshal the output: %s", err)
+	}
+
+	if dt.Spec.Template.Spec.Containers[0].VolumeMounts[0].Name != "data" {
+		t.Errorf("Expected volume name to be data: %v", dt)
+	}
+}
+
+func TestWithStaticFiles(t *testing.T) {
+	_compose_file := `
+services:
+    web:
+        image: nginx:1.29
+        volumes:
+        - ./static:/var/www
+        labels:
+            %sconfigmap-files: |-
+                - ./static
+`
+	_compose_file = fmt.Sprintf(_compose_file, KATENARY_PREFIX)
+	tmpDir := setup(_compose_file)
+	defer teardown(tmpDir)
+
+	// create a static directory with an index.html file
+	staticDir := tmpDir + "/static"
+	os.Mkdir(staticDir, 0o755)
+	indexFile, err := os.Create(staticDir + "/index.html")
+	if err != nil {
+		t.Errorf("Failed to create index.html: %s", err)
+	}
+	indexFile.WriteString("<html><body><h1>Hello, World!</h1></body></html>")
+	indexFile.Close()
+
+	currentDir, _ := os.Getwd()
+	os.Chdir(tmpDir)
+	defer os.Chdir(currentDir)
+
+	output := _compile_test(t, "-s", "templates/web/deployment.yaml")
+	dt := v1.Deployment{}
+	if err := yaml.Unmarshal([]byte(output), &dt); err != nil {
+		t.Errorf("Failed to unmarshal the output: %s", err)
+	}
+	// get the volume mount path
+	volumeMountPath := dt.Spec.Template.Spec.Containers[0].VolumeMounts[0].MountPath
+	if volumeMountPath != "/var/www" {
+		t.Errorf("Expected volume mount path to be /var/www, got %s", volumeMountPath)
+	}
+
+	// read the configMap
+	output, err = helmTemplate(ConvertOptions{
+		OutputDir: tmpDir + "/chart",
+	}, "-s", "templates/web/statics/static/configmap.yaml")
+	if err != nil {
+		t.Errorf("Failed to run helm template: %s", err)
+	}
+	configMap := corev1.ConfigMap{}
+	if err := yaml.Unmarshal([]byte(output), &configMap); err != nil {
+		t.Errorf("Failed to unmarshal the output: %s", err)
+	}
+	data := configMap.Data
+	if len(data) != 1 {
+		t.Errorf("Expected 1 data, got %d", len(data))
+	}
+	if data["index.html"] != "<html><body><h1>Hello, World!</h1></body></html>" {
+		t.Errorf("Expected index.html to be <html><body><h1>Hello, World!</h1></body></html>, got %s", data["index.html"])
+	}
+}