92 lines
2.0 KiB
Go
92 lines
2.0 KiB
Go
package command
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/terraform"
|
|
"github.com/mitchellh/colorstring"
|
|
)
|
|
|
|
// Test that a root level data source gets a special plan output on create
|
|
func TestFormatPlan_rootDataSource(t *testing.T) {
|
|
plan := &terraform.Plan{
|
|
Diff: &terraform.Diff{
|
|
Modules: []*terraform.ModuleDiff{
|
|
&terraform.ModuleDiff{
|
|
Path: []string{"root"},
|
|
Resources: map[string]*terraform.InstanceDiff{
|
|
"data.type.name": &terraform.InstanceDiff{
|
|
Attributes: map[string]*terraform.ResourceAttrDiff{
|
|
"A": &terraform.ResourceAttrDiff{
|
|
New: "B",
|
|
RequiresNew: true,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
opts := &FormatPlanOpts{
|
|
Plan: plan,
|
|
Color: &colorstring.Colorize{
|
|
Colors: colorstring.DefaultColors,
|
|
Disable: true,
|
|
},
|
|
ModuleDepth: 1,
|
|
}
|
|
|
|
actual := FormatPlan(opts)
|
|
|
|
expected := strings.TrimSpace(`
|
|
<= data.type.name
|
|
A: "B"
|
|
`)
|
|
if actual != expected {
|
|
t.Fatalf("expected:\n\n%s\n\ngot:\n\n%s", expected, actual)
|
|
}
|
|
}
|
|
|
|
// Test that data sources nested in modules get the same plan output
|
|
func TestFormatPlan_nestedDataSource(t *testing.T) {
|
|
plan := &terraform.Plan{
|
|
Diff: &terraform.Diff{
|
|
Modules: []*terraform.ModuleDiff{
|
|
&terraform.ModuleDiff{
|
|
Path: []string{"root", "nested"},
|
|
Resources: map[string]*terraform.InstanceDiff{
|
|
"data.type.name": &terraform.InstanceDiff{
|
|
Attributes: map[string]*terraform.ResourceAttrDiff{
|
|
"A": &terraform.ResourceAttrDiff{
|
|
New: "B",
|
|
RequiresNew: true,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
opts := &FormatPlanOpts{
|
|
Plan: plan,
|
|
Color: &colorstring.Colorize{
|
|
Colors: colorstring.DefaultColors,
|
|
Disable: true,
|
|
},
|
|
ModuleDepth: 2,
|
|
}
|
|
|
|
actual := FormatPlan(opts)
|
|
|
|
expected := strings.TrimSpace(`
|
|
<= module.nested.data.type.name
|
|
A: "B"
|
|
`)
|
|
if actual != expected {
|
|
t.Fatalf("expected:\n\n%s\n\ngot:\n\n%s", expected, actual)
|
|
}
|
|
}
|