-
Notifications
You must be signed in to change notification settings - Fork 0
/
buildin_map_test.go
104 lines (98 loc) · 2.41 KB
/
buildin_map_test.go
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
package template
import (
"errors"
"testing"
)
func TestExtractFilter(t *testing.T) {
cases := []struct {
name string
template string
context map[string]interface{}
expected string
}{
{
name: "ExtractFromMap",
template: "{{ data | extract:'user.address.city' }}",
context: map[string]interface{}{
"data": map[string]interface{}{
"user": map[string]interface{}{
"address": map[string]interface{}{
"city": "New York",
},
},
},
},
expected: "New York",
},
{
name: "ExtractFromArray",
template: "{{ data | extract:'0' }}",
context: map[string]interface{}{
"data": []interface{}{"First Element", "Second Element"},
},
expected: "First Element",
},
{
name: "ExtractFromNestedArray",
template: "{{ data | extract:'1.0' }}",
context: map[string]interface{}{
"data": []interface{}{
[]interface{}{"Nested First Element"},
[]interface{}{"Nested Second Element"},
},
},
expected: "Nested Second Element",
},
{
name: "KeyNotFound",
template: "{{ data | extract:'nonexistent.key' }}",
context: map[string]interface{}{
"data": map[string]interface{}{
"exists": "This exists",
},
},
expected: "ErrContextKeyNotFound",
},
{
name: "IndexOutOfRange",
template: "{{ data | extract:'2' }}",
context: map[string]interface{}{
"data": []interface{}{"First", "Second"},
},
expected: "ErrContextIndexOutOfRange",
},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
// Parse the template
tpl, err := Parse(tc.template)
if err != nil {
t.Fatalf("Failed to parse template: %v", err)
}
// Create a context and add variables
context := NewContext()
for k, v := range tc.context {
context.Set(k, v)
}
// Execute the template
output, err := Execute(tpl, context)
var finalOutput string
if err != nil {
switch {
case errors.Is(err, ErrContextKeyNotFound):
finalOutput = "ErrContextKeyNotFound"
case errors.Is(err, ErrContextIndexOutOfRange):
finalOutput = "ErrContextIndexOutOfRange"
default:
t.Fatalf("Unexpected error during execution: %v", err)
}
} else {
finalOutput = output
}
// Verify the output matches the expected result
if finalOutput != tc.expected {
t.Errorf("Expected '%s', got '%s' for test case '%s'", tc.expected, finalOutput, tc.name)
}
})
}
}