wenshu/k8s-orchestrator/internal/controller/starrocksvirtualcluster_controller_test.go

package controller

import "testing"

func TestBuildStarRocksComponentSpec_FeAddsChecksumEnvVars(t *testing.T) {
	spec := buildStarRocksComponentSpec(
		"starrocks-15",
		"fe",
		"starrocks/fe-ubuntu:3.5-latest",
		1,
		[]string{"node-a"},
		nil,
	)

	_, hasBeEnv := spec["beEnvVars"]
	if hasBeEnv {
		t.Fatalf("expected fe component to not carry beEnvVars")
	}
	feEnv, ok := spec["feEnvVars"].([]interface{})
	if !ok {
		t.Fatalf("expected feEnvVars to be []interface{}, got %T", spec["feEnvVars"])
	}
	assertChecksumEnvVars(t, feEnv)
}

func TestBuildStarRocksComponentSpec_BeAddsChecksumEnvVars(t *testing.T) {
	spec := buildStarRocksComponentSpec(
		"starrocks-15",
		"be",
		"starrocks/be-ubuntu:3.5-latest",
		1,
		[]string{"node-b"},
		nil,
	)

	_, hasFeEnv := spec["feEnvVars"]
	if hasFeEnv {
		t.Fatalf("expected be component to not carry feEnvVars")
	}
	beEnv, ok := spec["beEnvVars"].([]interface{})
	if !ok {
		t.Fatalf("expected beEnvVars to be []interface{}, got %T", spec["beEnvVars"])
	}
	assertChecksumEnvVars(t, beEnv)
}

func assertChecksumEnvVars(t *testing.T, envVars []interface{}) {
	t.Helper()
	if len(envVars) != 2 {
		t.Fatalf("expected 2 checksum env vars, got %d", len(envVars))
	}
	envMap := make(map[string]string, len(envVars))
	for _, item := range envVars {
		envItem, ok := item.(map[string]interface{})
		if !ok {
			t.Fatalf("expected env item map[string]interface{}, got %T", item)
		}
		name, _ := envItem["name"].(string)
		value, _ := envItem["value"].(string)
		envMap[name] = value
	}
	if envMap["AWS_REQUEST_CHECKSUM_CALCULATION"] != "WHEN_REQUIRED" {
		t.Fatalf("unexpected AWS_REQUEST_CHECKSUM_CALCULATION value: %q", envMap["AWS_REQUEST_CHECKSUM_CALCULATION"])
	}
	if envMap["AWS_RESPONSE_CHECKSUM_VALIDATION"] != "WHEN_REQUIRED" {
		t.Fatalf("unexpected AWS_RESPONSE_CHECKSUM_VALIDATION value: %q", envMap["AWS_RESPONSE_CHECKSUM_VALIDATION"])
	}
}