From 827b5bc830bcb0ad0c48115397831202fc83b591 Mon Sep 17 00:00:00 2001
From: Patrice Ferlet <metal3d@gmail.com>
Date: Fri, 22 Nov 2024 16:23:00 +0100
Subject: [PATCH] test(values): add map-env and exchange volumes tests

---
 generator/configMap_test.go | 33 +++++++++++++++++++++
 generator/volume_test.go    | 57 +++++++++++++++++++++++++++++++++++++
 2 files changed, 90 insertions(+)

diff --git a/generator/configMap_test.go b/generator/configMap_test.go
index 5b0c192..a0b5f80 100644
--- a/generator/configMap_test.go
+++ b/generator/configMap_test.go
@@ -1,6 +1,8 @@
 package generator
 
 import (
+	"fmt"
+	"katenary/generator/labels"
 	"os"
 	"testing"
 
@@ -40,3 +42,34 @@ services:
 		t.Errorf("Expected BAR to be baz, got %s", data["BAR"])
 	}
 }
+
+func TestMapEnv(t *testing.T) {
+	composeFile := `
+services:
+  web:
+    image: nginx:1.29
+    environment:
+      FOO: bar
+    labels:
+      %[1]s/map-env: |-
+        FOO: 'baz'
+`
+
+	composeFile = fmt.Sprintf(composeFile, labels.KatenaryLabelPrefix)
+	tmpDir := setup(composeFile)
+	defer teardown(tmpDir)
+
+	currentDir, _ := os.Getwd()
+	os.Chdir(tmpDir)
+	defer os.Chdir(currentDir)
+
+	output := internalCompileTest(t, "-s", "templates/web/configmap.yaml")
+	configMap := v1.ConfigMap{}
+	if err := yaml.Unmarshal([]byte(output), &configMap); err != nil {
+		t.Errorf(unmarshalError, err)
+	}
+	data := configMap.Data
+	if v, ok := data["FOO"]; !ok || v != "baz" {
+		t.Errorf("Expected FOO to be baz, got %s", v)
+	}
+}
diff --git a/generator/volume_test.go b/generator/volume_test.go
index 1923140..ef365be 100644
--- a/generator/volume_test.go
+++ b/generator/volume_test.go
@@ -191,3 +191,60 @@ volumes:
 		t.Errorf("Expected volume name to be data: %v", dt)
 	}
 }
+
+func TestExchangeVolume(t *testing.T) {
+	composeFile := `
+services:
+  app1:
+    image: nginx:1.29
+    labels:
+      %[1]s/exchange-volumes: |-
+        - name: data
+          mountPath: /var/www
+  app2:
+    image: foo:bar
+    labels:
+      %[1]s/same-pod: app1
+      %[1]s/exchange-volumes: |-
+        - name: data
+          mountPath: /opt
+          init: cp -r /var/www /opt
+`
+	composeFile = fmt.Sprintf(composeFile, labels.KatenaryLabelPrefix)
+	tmpDir := setup(composeFile)
+	defer teardown(tmpDir)
+
+	currentDir, _ := os.Getwd()
+	os.Chdir(tmpDir)
+	defer os.Chdir(currentDir)
+	output := internalCompileTest(t, "-s", "templates/app1/deployment.yaml")
+	dt := v1.Deployment{}
+	if err := yaml.Unmarshal([]byte(output), &dt); err != nil {
+		t.Errorf(unmarshalError, err)
+	}
+	// the deployment should have a volume named "data"
+	volumes := dt.Spec.Template.Spec.Volumes
+	found := false
+	for v := range volumes {
+		if volumes[v].Name == "exchange-data" {
+			found = true
+			break
+		}
+	}
+	if !found {
+		t.Errorf("Expected volume name to be data: %v", volumes)
+	}
+	mounted := 0
+	// we should have a volume mount for both containers
+	containers := dt.Spec.Template.Spec.Containers
+	for c := range containers {
+		for _, vm := range containers[c].VolumeMounts {
+			if vm.Name == "exchange-data" {
+				mounted++
+			}
+		}
+	}
+	if mounted != 2 {
+		t.Errorf("Expected 2 mounted volumes, got %d", mounted)
+	}
+}