113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package include
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/davecgh/go-spew/spew"
|
|
"github.com/hashicorp/hcl2/gozcl"
|
|
"github.com/hashicorp/hcl2/zcl"
|
|
"github.com/hashicorp/hcl2/zcltest"
|
|
"github.com/zclconf/go-cty/cty"
|
|
)
|
|
|
|
func TestTransformer(t *testing.T) {
|
|
caller := zcltest.MockBody(&zcl.BodyContent{
|
|
Blocks: zcl.Blocks{
|
|
{
|
|
Type: "include",
|
|
Body: zcltest.MockBody(&zcl.BodyContent{
|
|
Attributes: zcltest.MockAttrs(map[string]zcl.Expression{
|
|
"path": zcltest.MockExprVariable("var_path"),
|
|
}),
|
|
}),
|
|
},
|
|
{
|
|
Type: "include",
|
|
Body: zcltest.MockBody(&zcl.BodyContent{
|
|
Attributes: zcltest.MockAttrs(map[string]zcl.Expression{
|
|
"path": zcltest.MockExprLiteral(cty.StringVal("include2")),
|
|
}),
|
|
}),
|
|
},
|
|
{
|
|
Type: "foo",
|
|
Body: zcltest.MockBody(&zcl.BodyContent{
|
|
Attributes: zcltest.MockAttrs(map[string]zcl.Expression{
|
|
"from": zcltest.MockExprLiteral(cty.StringVal("caller")),
|
|
}),
|
|
}),
|
|
},
|
|
},
|
|
})
|
|
|
|
resolver := MapResolver(map[string]zcl.Body{
|
|
"include1": zcltest.MockBody(&zcl.BodyContent{
|
|
Blocks: zcl.Blocks{
|
|
{
|
|
Type: "foo",
|
|
Body: zcltest.MockBody(&zcl.BodyContent{
|
|
Attributes: zcltest.MockAttrs(map[string]zcl.Expression{
|
|
"from": zcltest.MockExprLiteral(cty.StringVal("include1")),
|
|
}),
|
|
}),
|
|
},
|
|
},
|
|
}),
|
|
"include2": zcltest.MockBody(&zcl.BodyContent{
|
|
Blocks: zcl.Blocks{
|
|
{
|
|
Type: "foo",
|
|
Body: zcltest.MockBody(&zcl.BodyContent{
|
|
Attributes: zcltest.MockAttrs(map[string]zcl.Expression{
|
|
"from": zcltest.MockExprLiteral(cty.StringVal("include2")),
|
|
}),
|
|
}),
|
|
},
|
|
},
|
|
}),
|
|
})
|
|
|
|
ctx := &zcl.EvalContext{
|
|
Variables: map[string]cty.Value{
|
|
"var_path": cty.StringVal("include1"),
|
|
},
|
|
}
|
|
|
|
transformer := Transformer("include", ctx, resolver)
|
|
merged := transformer.TransformBody(caller)
|
|
|
|
type foo struct {
|
|
From string `zcl:"from,attr"`
|
|
}
|
|
type result struct {
|
|
Foos []foo `zcl:"foo,block"`
|
|
}
|
|
var got result
|
|
diags := gozcl.DecodeBody(merged, nil, &got)
|
|
if len(diags) != 0 {
|
|
t.Errorf("unexpected diags")
|
|
for _, diag := range diags {
|
|
t.Logf("- %s", diag)
|
|
}
|
|
}
|
|
|
|
want := result{
|
|
Foos: []foo{
|
|
{
|
|
From: "caller",
|
|
},
|
|
{
|
|
From: "include1",
|
|
},
|
|
{
|
|
From: "include2",
|
|
},
|
|
},
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("wrong result\ngot: %swant: %s", spew.Sdump(got), spew.Sdump(want))
|
|
}
|
|
}
|