70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package terraform
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestResourceCountTransformer(t *testing.T) {
|
|
cfg := testModule(t, "transform-resource-count-basic").Config()
|
|
resource := cfg.Resources[0]
|
|
|
|
g := Graph{Path: RootModulePath}
|
|
{
|
|
tf := &ResourceCountTransformer{Resource: resource}
|
|
if err := tf.Transform(&g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
actual := strings.TrimSpace(g.String())
|
|
expected := strings.TrimSpace(testResourceCountTransformStr)
|
|
if actual != expected {
|
|
t.Fatalf("bad:\n\n%s", actual)
|
|
}
|
|
}
|
|
|
|
func TestResourceCountTransformer_countNegative(t *testing.T) {
|
|
cfg := testModule(t, "transform-resource-count-negative").Config()
|
|
resource := cfg.Resources[0]
|
|
|
|
g := Graph{Path: RootModulePath}
|
|
{
|
|
tf := &ResourceCountTransformer{Resource: resource}
|
|
if err := tf.Transform(&g); err == nil {
|
|
t.Fatal("should error")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestResourceCountTransformer_deps(t *testing.T) {
|
|
cfg := testModule(t, "transform-resource-count-deps").Config()
|
|
resource := cfg.Resources[0]
|
|
|
|
g := Graph{Path: RootModulePath}
|
|
{
|
|
tf := &ResourceCountTransformer{Resource: resource}
|
|
if err := tf.Transform(&g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
actual := strings.TrimSpace(g.String())
|
|
expected := strings.TrimSpace(testResourceCountTransformDepsStr)
|
|
if actual != expected {
|
|
t.Fatalf("bad:\n\n%s", actual)
|
|
}
|
|
}
|
|
|
|
const testResourceCountTransformStr = `
|
|
aws_instance.foo #0
|
|
aws_instance.foo #1
|
|
aws_instance.foo #2
|
|
`
|
|
|
|
const testResourceCountTransformDepsStr = `
|
|
aws_instance.foo #0
|
|
aws_instance.foo #1
|
|
aws_instance.foo #0
|
|
`
|