cost-model/pkg/env/costmodel_test.go

package env

import (
	"os"
	"testing"
)

func TestGetExportCSVMaxDays(t *testing.T) {
	tests := []struct {
		name string
		want int
		pre  func()
	}{
		{
			name: "Ensure the default value is 90d",
			want: 90,
		},
		{
			name: "Ensure the value is 30 when EXPORT_CSV_MAX_DAYS is set to 30",
			want: 30,
			pre: func() {
				os.Setenv("EXPORT_CSV_MAX_DAYS", "30")
			},
		},
		{
			name: "Ensure the value is 90 when EXPORT_CSV_MAX_DAYS is set to empty string",
			want: 90,
			pre: func() {
				os.Setenv("EXPORT_CSV_MAX_DAYS", "")
			},
		},
		{
			name: "Ensure the value is 90 when EXPORT_CSV_MAX_DAYS is set to invalid value",
			want: 90,
			pre: func() {
				os.Setenv("EXPORT_CSV_MAX_DAYS", "foo")
			},
		},
	}
	for _, tt := range tests {
		if tt.pre != nil {
			tt.pre()
		}
		t.Run(tt.name, func(t *testing.T) {
			if got := GetExportCSVMaxDays(); got != tt.want {
				t.Errorf("GetExportCSVMaxDays() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestGetKubernetesEnabled(t *testing.T) {
	tests := []struct {
		name string
		want bool
		pre  func()
	}{
		{
			name: "Ensure the default value is false",
			want: false,
		},
		{
			name: "Ensure the value is true when KUBERNETES_PORT has a value",
			want: true,
			pre: func() {
				os.Setenv("KUBERNETES_PORT", "tcp://10.43.0.1:443")
			},
		},
	}
	for _, tt := range tests {
		if tt.pre != nil {
			tt.pre()
		}
		t.Run(tt.name, func(t *testing.T) {
			if got := IsKubernetesEnabled(); got != tt.want {
				t.Errorf("IsKubernetesEnabled() = %v, want %v", got, tt.want)
			}
		})
	}

}

func TestIsMCPServerEnabled_DefaultFalse(t *testing.T) {
	old, hadOld := os.LookupEnv("MCP_SERVER_ENABLED")
	os.Unsetenv("MCP_SERVER_ENABLED")
	t.Cleanup(func() {
		if hadOld {
			os.Setenv("MCP_SERVER_ENABLED", old)
		} else {
			os.Unsetenv("MCP_SERVER_ENABLED")
		}
	})
	if got := IsMCPServerEnabled(); got {
		t.Fatalf("expected false when MCP_SERVER_ENABLED is unset, got %v", got)
	}
}

func TestIsMCPServerEnabled_True(t *testing.T) {
	t.Setenv("MCP_SERVER_ENABLED", "true")
	if got := IsMCPServerEnabled(); !got {
		t.Fatalf("expected true when env var set to true, got %v", got)
	}
}