cozystack/internal/template/template_test.go

package template

import (
	"encoding/json"
	"testing"

	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func TestTemplate_PodTemplateSpec(t *testing.T) {
	original := corev1.PodTemplateSpec{
		ObjectMeta: metav1.ObjectMeta{
			Name: "my-pod",
			Labels: map[string]string{
				"app": "demo",
			},
			Annotations: map[string]string{
				"note": "hello",
			},
		},
		Spec: corev1.PodSpec{
			Containers: []corev1.Container{
				{
					Name:  "{{ .Release.Name }}",
					Image: "nginx:1.21",
					Args:  []string{"--flag={{ .Values.value }}"},
					Env: []corev1.EnvVar{
						{
							Name:  "FOO",
							Value: "{{ .Release.Namespace }}",
						},
					},
				},
			},
		},
	}
	templateContext := map[string]any{
		"Release": map[string]any{
			"Name":      "foo",
			"Namespace": "notdefault",
		},
		"Values": map[string]any{
			"value": 3,
		},
	}
	reference := *original.DeepCopy()
	reference.Spec.Containers[0].Name = "foo"
	reference.Spec.Containers[0].Args[0] = "--flag=3"
	reference.Spec.Containers[0].Env[0].Value = "notdefault"
	got, err := Template(&original, templateContext)
	if err != nil {
		t.Fatalf("Template returned error: %v", err)
	}
	b1, err := json.Marshal(reference)
	t.Logf("reference:\n%s", string(b1))
	if err != nil {
		t.Fatalf("failed to marshal reference value: %v", err)
	}
	b2, err := json.Marshal(got)
	t.Logf("got:\n%s", string(b2))
	if err != nil {
		t.Fatalf("failed to marshal transformed value: %v", err)
	}
	if string(b1) != string(b2) {
		t.Fatalf("transformed value not equal to reference value, expected: %s, got: %s", string(b1), string(b2))
	}
}

func TestTemplate_SprigFunctions(t *testing.T) {
	// Strategy templates lean on Helm-style helpers for nullable parameter
	// handling (e.g. picking a default Secret name when an override is not
	// supplied via BackupClass). Validating one such expression here pins the
	// contract: Sprig's text/template FuncMap is reachable through the engine
	// used by every strategy driver.
	original := corev1.PodTemplateSpec{
		Spec: corev1.PodSpec{
			Containers: []corev1.Container{{
				Name: "secret-name-test",
				Env: []corev1.EnvVar{{
					Name:  "SECRET_NAME",
					Value: `{{ default (printf "%s-backup-s3" .Release.Name) (index .Parameters "s3SecretName") }}`,
				}, {
					Name:  "SECRET_NAME_OVERRIDE",
					Value: `{{ default (printf "%s-backup-s3" .Release.Name) (index .ParametersOverride "s3SecretName") }}`,
				}},
			}},
		},
	}
	templateContext := map[string]any{
		"Release": map[string]any{
			"Name": "demo",
		},
		"Parameters":         map[string]string{},
		"ParametersOverride": map[string]string{"s3SecretName": "tenant-managed-creds"},
	}
	got, err := Template(&original, templateContext)
	if err != nil {
		t.Fatalf("Template returned error: %v", err)
	}
	if v := got.Spec.Containers[0].Env[0].Value; v != "demo-backup-s3" {
		t.Errorf("default fallback path: expected %q, got %q", "demo-backup-s3", v)
	}
	if v := got.Spec.Containers[0].Env[1].Value; v != "tenant-managed-creds" {
		t.Errorf("default override path: expected %q, got %q", "tenant-managed-creds", v)
	}
}