Create more tests

generator/service_test.go

@@ -0,0 +1,49 @@
+package generator
+
+import (
+	"os"
+	"testing"
+
+	v1 "k8s.io/api/core/v1"
+	"sigs.k8s.io/yaml"
+)
+
+func TestBasicService(t *testing.T) {
+	composeFile := `
+services:
+    web:
+        image: nginx:1.29
+        ports:
+        - 80:80
+        - 443:443
+    `
+	tmpDir := setup(composeFile)
+	defer teardown(tmpDir)
+
+	currentDir, _ := os.Getwd()
+	os.Chdir(tmpDir)
+	defer os.Chdir(currentDir)
+
+	output := _compile_test(t, "-s", "templates/web/service.yaml")
+	service := v1.Service{}
+	if err := yaml.Unmarshal([]byte(output), &service); err != nil {
+		t.Errorf("Failed to unmarshal the output: %s", err)
+	}
+
+	if len(service.Spec.Ports) != 2 {
+		t.Errorf("Expected 2 ports, got %d", len(service.Spec.Ports))
+	}
+
+	foundPort := 0
+	for _, port := range service.Spec.Ports {
+		if port.Port == 80 && port.TargetPort.StrVal == "http" {
+			foundPort++
+		}
+		if port.Port == 443 && port.TargetPort.StrVal == "https" {
+			foundPort++
+		}
+	}
+	if foundPort != 2 {
+		t.Errorf("Expected 2 ports, got %d", foundPort)
+	}
+}