137 lines
2.4 KiB
Go
137 lines
2.4 KiB
Go
package lang
|
|
|
|
import (
|
|
"reflect"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/config/lang/ast"
|
|
)
|
|
|
|
func TestEngineExecute(t *testing.T) {
|
|
cases := []struct {
|
|
Input string
|
|
Scope *Scope
|
|
Error bool
|
|
Result interface{}
|
|
ResultType ast.Type
|
|
}{
|
|
{
|
|
"foo",
|
|
nil,
|
|
false,
|
|
"foo",
|
|
ast.TypeString,
|
|
},
|
|
|
|
{
|
|
"foo ${bar}",
|
|
&Scope{
|
|
VarMap: map[string]Variable{
|
|
"bar": Variable{
|
|
Value: "baz",
|
|
Type: ast.TypeString,
|
|
},
|
|
},
|
|
},
|
|
false,
|
|
"foo baz",
|
|
ast.TypeString,
|
|
},
|
|
|
|
{
|
|
"foo ${rand()}",
|
|
&Scope{
|
|
FuncMap: map[string]Function{
|
|
"rand": Function{
|
|
ReturnType: ast.TypeString,
|
|
Callback: func([]interface{}) (interface{}, error) {
|
|
return "42", nil
|
|
},
|
|
},
|
|
},
|
|
},
|
|
false,
|
|
"foo 42",
|
|
ast.TypeString,
|
|
},
|
|
|
|
{
|
|
`foo ${rand("foo", "bar")}`,
|
|
&Scope{
|
|
FuncMap: map[string]Function{
|
|
"rand": Function{
|
|
ReturnType: ast.TypeString,
|
|
Variadic: true,
|
|
VariadicType: ast.TypeString,
|
|
Callback: func(args []interface{}) (interface{}, error) {
|
|
var result string
|
|
for _, a := range args {
|
|
result += a.(string)
|
|
}
|
|
return result, nil
|
|
},
|
|
},
|
|
},
|
|
},
|
|
false,
|
|
"foo foobar",
|
|
ast.TypeString,
|
|
},
|
|
|
|
// Testing implicit type conversions
|
|
|
|
{
|
|
"foo ${bar}",
|
|
&Scope{
|
|
VarMap: map[string]Variable{
|
|
"bar": Variable{
|
|
Value: 42,
|
|
Type: ast.TypeInt,
|
|
},
|
|
},
|
|
},
|
|
false,
|
|
"foo 42",
|
|
ast.TypeString,
|
|
},
|
|
|
|
{
|
|
`foo ${foo("42")}`,
|
|
&Scope{
|
|
FuncMap: map[string]Function{
|
|
"foo": Function{
|
|
ArgTypes: []ast.Type{ast.TypeInt},
|
|
ReturnType: ast.TypeString,
|
|
Callback: func(args []interface{}) (interface{}, error) {
|
|
return strconv.FormatInt(int64(args[0].(int)), 10), nil
|
|
},
|
|
},
|
|
},
|
|
},
|
|
false,
|
|
"foo 42",
|
|
ast.TypeString,
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
node, err := Parse(tc.Input)
|
|
if err != nil {
|
|
t.Fatalf("Error: %s\n\nInput: %s", err, tc.Input)
|
|
}
|
|
|
|
engine := &Engine{GlobalScope: tc.Scope}
|
|
out, outType, err := engine.Execute(node)
|
|
if (err != nil) != tc.Error {
|
|
t.Fatalf("Error: %s\n\nInput: %s", err, tc.Input)
|
|
}
|
|
if outType != tc.ResultType {
|
|
t.Fatalf("Bad: %s\n\nInput: %s", outType, tc.Input)
|
|
}
|
|
if !reflect.DeepEqual(out, tc.Result) {
|
|
t.Fatalf("Bad: %#v\n\nInput: %s", out, tc.Input)
|
|
}
|
|
}
|
|
}
|