mantle/packages/engine/internal/cel/cel_values_test.go at e3b03fc5c487978c02bd884f59c0355759f31ea7 · dvflw/mantle · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package cel

import (
	"testing"
)

func TestSetValuesEnv_LayersBetweenConfigAndOS(t *testing.T) {
	eval, err := NewEvaluator()
	if err != nil {
		t.Fatalf("NewEvaluator() error: %v", err)
	}

	eval.SetConfigEnv(map[string]string{
		"LOG_LEVEL": "info",
		"REGION":    "us-west-2",
		"APP_NAME":  "config-app",
	})

	eval.SetValuesEnv(map[string]string{
		"LOG_LEVEL": "error",
		"DB_HOST":   "prod-db.example.com",
	})

	ctx := &Context{
		Steps:  make(map[string]map[string]any),
		Inputs: make(map[string]any),
	}

	// Values override config.
	result, err := eval.Eval(`env.LOG_LEVEL`, ctx)
	if err != nil {
		t.Fatalf("Eval(env.LOG_LEVEL) error: %v", err)
	}
	if result != "error" {
		t.Errorf("env.LOG_LEVEL = %v, want %q (values overrides config)", result, "error")
	}

	// Config value preserved when values doesn't override.
	result, err = eval.Eval(`env.REGION`, ctx)
	if err != nil {
		t.Fatalf("Eval(env.REGION) error: %v", err)
	}
	if result != "us-west-2" {
		t.Errorf("env.REGION = %v, want %q (config preserved)", result, "us-west-2")
	}

	// Values-only key is accessible.
	result, err = eval.Eval(`env.DB_HOST`, ctx)
	if err != nil {
		t.Fatalf("Eval(env.DB_HOST) error: %v", err)
	}
	if result != "prod-db.example.com" {
		t.Errorf("env.DB_HOST = %v, want %q", result, "prod-db.example.com")
	}
}

func TestSetValuesEnv_OSEnvTakesPrecedence(t *testing.T) {
	eval, err := NewEvaluator()
	if err != nil {
		t.Fatalf("NewEvaluator() error: %v", err)
	}

	t.Setenv("MANTLE_ENV_LOG_LEVEL", "debug")

	eval.SetConfigEnv(map[string]string{
		"LOG_LEVEL": "info",
	})
	eval.SetValuesEnv(map[string]string{
		"LOG_LEVEL": "error",
	})

	ctx := &Context{
		Steps:  make(map[string]map[string]any),
		Inputs: make(map[string]any),
	}

	result, err := eval.Eval(`env.LOG_LEVEL`, ctx)
	if err != nil {
		t.Fatalf("Eval(env.LOG_LEVEL) error: %v", err)
	}
	if result != "debug" {
		t.Errorf("env.LOG_LEVEL = %v, want %q (OS env wins)", result, "debug")
	}
}

func TestSetValuesEnv_NilClearsValuesLayer(t *testing.T) {
	eval, err := NewEvaluator()
	if err != nil {
		t.Fatalf("NewEvaluator() error: %v", err)
	}

	eval.SetConfigEnv(map[string]string{
		"LOG_LEVEL": "info",
	})
	eval.SetValuesEnv(map[string]string{
		"LOG_LEVEL": "error",
	})

	eval.SetValuesEnv(nil)

	ctx := &Context{
		Steps:  make(map[string]map[string]any),
		Inputs: make(map[string]any),
	}

	result, err := eval.Eval(`env.LOG_LEVEL`, ctx)
	if err != nil {
		t.Fatalf("Eval(env.LOG_LEVEL) error: %v", err)
	}
	if result != "info" {
		t.Errorf("env.LOG_LEVEL = %v, want %q (back to config after clear)", result, "info")
	}
}