2015-02-04 19:02:18 -06:00
|
|
|
package terraform
|
|
|
|
|
|
|
|
import (
|
|
|
|
"strings"
|
|
|
|
"testing"
|
2018-05-04 21:24:06 -05:00
|
|
|
|
|
|
|
"github.com/hashicorp/terraform/addrs"
|
2015-02-04 19:02:18 -06:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestRootTransformer(t *testing.T) {
|
|
|
|
mod := testModule(t, "transform-root-basic")
|
|
|
|
|
2018-05-04 21:24:06 -05:00
|
|
|
g := Graph{Path: addrs.RootModuleInstance}
|
2015-02-04 19:02:18 -06:00
|
|
|
{
|
2018-05-04 21:24:06 -05:00
|
|
|
tf := &ConfigTransformer{Config: mod}
|
2015-02-04 19:02:18 -06:00
|
|
|
if err := tf.Transform(&g); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-01-26 21:40:46 -06:00
|
|
|
{
|
|
|
|
transform := &MissingProviderTransformer{
|
|
|
|
Providers: []string{"aws", "do"},
|
|
|
|
}
|
|
|
|
if err := transform.Transform(&g); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-02-04 19:02:18 -06:00
|
|
|
{
|
|
|
|
transform := &ProviderTransformer{}
|
|
|
|
if err := transform.Transform(&g); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
{
|
|
|
|
transform := &RootTransformer{}
|
|
|
|
if err := transform.Transform(&g); err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
actual := strings.TrimSpace(g.String())
|
|
|
|
expected := strings.TrimSpace(testTransformRootBasicStr)
|
|
|
|
if actual != expected {
|
2018-05-07 19:21:30 -05:00
|
|
|
t.Fatalf("wrong result\n\ngot:\n%s\n\nwant:\n%s", actual, expected)
|
2015-02-04 19:02:18 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
root, err := g.Root()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
if _, ok := root.(graphNodeRoot); !ok {
|
|
|
|
t.Fatalf("bad: %#v", root)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
const testTransformRootBasicStr = `
|
|
|
|
aws_instance.foo
|
2020-04-01 12:13:40 -05:00
|
|
|
provider["registry.terraform.io/hashicorp/aws"]
|
2015-02-04 19:02:18 -06:00
|
|
|
do_droplet.bar
|
2020-04-01 12:13:40 -05:00
|
|
|
provider["registry.terraform.io/hashicorp/do"]
|
|
|
|
provider["registry.terraform.io/hashicorp/aws"]
|
|
|
|
provider["registry.terraform.io/hashicorp/do"]
|
2015-02-04 19:02:18 -06:00
|
|
|
root
|
|
|
|
aws_instance.foo
|
|
|
|
do_droplet.bar
|
|
|
|
`
|