opentofu/internal/instances/set_test.go
2023-05-02 15:33:06 +00:00

215 lines
11 KiB
Go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0
package instances
import (
"testing"
"github.com/hashicorp/terraform/internal/addrs"
"github.com/zclconf/go-cty/cty"
)
func TestSet(t *testing.T) {
exp := NewExpander()
// The following constructs the following imaginary module/resource tree:
// - root module
// - test_thing.single: no repetition
// - test_thing.count: count = 1
// - test_thing.for_each: for_each = { c = "C" }
// - module.single: no repetition
// - test_thing.single: no repetition
// - module.nested_single: no repetition
// - module.zero_count: count = 0
// - module.count: count = 2
// - module.nested_for_each: [0] for_each = {}, [1] for_each = { e = "E" }
// - module.for_each: for_each = { a = "A", b = "B" }
// - test_thing.count: ["a"] count = 0, ["b"] count = 1
exp.SetModuleSingle(addrs.RootModuleInstance, addrs.ModuleCall{Name: "single"})
exp.SetModuleCount(addrs.RootModuleInstance, addrs.ModuleCall{Name: "count"}, 2)
exp.SetModuleForEach(addrs.RootModuleInstance, addrs.ModuleCall{Name: "for_each"}, map[string]cty.Value{
"a": cty.StringVal("A"),
"b": cty.StringVal("B"),
})
exp.SetModuleSingle(addrs.RootModuleInstance.Child("single", addrs.NoKey), addrs.ModuleCall{Name: "nested_single"})
exp.SetModuleForEach(addrs.RootModuleInstance.Child("count", addrs.IntKey(0)), addrs.ModuleCall{Name: "nested_for_each"}, nil)
exp.SetModuleForEach(addrs.RootModuleInstance.Child("count", addrs.IntKey(1)), addrs.ModuleCall{Name: "nested_for_each"}, map[string]cty.Value{
"e": cty.StringVal("E"),
})
exp.SetModuleCount(
addrs.RootModuleInstance.Child("single", addrs.NoKey).Child("nested_single", addrs.NoKey),
addrs.ModuleCall{Name: "zero_count"},
0,
)
rAddr := func(name string) addrs.Resource {
return addrs.Resource{
Mode: addrs.ManagedResourceMode,
Type: "test_thing",
Name: name,
}
}
exp.SetResourceSingle(addrs.RootModuleInstance, rAddr("single"))
exp.SetResourceCount(addrs.RootModuleInstance, rAddr("count"), 1)
exp.SetResourceForEach(addrs.RootModuleInstance, rAddr("for_each"), map[string]cty.Value{
"c": cty.StringVal("C"),
})
exp.SetResourceSingle(addrs.RootModuleInstance.Child("single", addrs.NoKey), rAddr("single"))
exp.SetResourceCount(addrs.RootModuleInstance.Child("for_each", addrs.StringKey("a")), rAddr("count"), 0)
exp.SetResourceCount(addrs.RootModuleInstance.Child("for_each", addrs.StringKey("b")), rAddr("count"), 1)
set := exp.AllInstances()
// HasModuleInstance tests
if input := addrs.RootModuleInstance; !set.HasModuleInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.NoKey); !set.HasModuleInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.NoKey).Child("nested_single", addrs.NoKey); !set.HasModuleInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.IntKey(0)); !set.HasModuleInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.IntKey(1)); !set.HasModuleInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.IntKey(1)).Child("nested_for_each", addrs.StringKey("e")); !set.HasModuleInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("for_each", addrs.StringKey("a")); !set.HasModuleInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("for_each", addrs.StringKey("b")); !set.HasModuleInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.IntKey(0)); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.StringKey("a")); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.NoKey).Child("nonexist", addrs.NoKey); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.NoKey); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.IntKey(2)); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.StringKey("a")); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.IntKey(0)).Child("nested_for_each", addrs.StringKey("e")); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.NoKey).Child("nested_single", addrs.NoKey).Child("zero_count", addrs.NoKey); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.NoKey).Child("nested_single", addrs.NoKey).Child("zero_count", addrs.IntKey(0)); set.HasModuleInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
// HasModuleCall tests
if input := addrs.RootModuleInstance.ChildCall("single"); !set.HasModuleCall(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.NoKey).ChildCall("nested_single"); !set.HasModuleCall(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.ChildCall("count"); !set.HasModuleCall(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.IntKey(0)).ChildCall("nested_for_each"); !set.HasModuleCall(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("count", addrs.IntKey(1)).ChildCall("nested_for_each"); !set.HasModuleCall(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.ChildCall("for_each"); !set.HasModuleCall(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.NoKey).Child("nested_single", addrs.NoKey).ChildCall("zero_count"); !set.HasModuleCall(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.ChildCall("nonexist"); set.HasModuleCall(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := addrs.RootModuleInstance.Child("single", addrs.NoKey).ChildCall("nonexist"); set.HasModuleCall(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
// HasResourceInstance tests
if input := rAddr("single").Instance(addrs.NoKey).Absolute(addrs.RootModuleInstance); !set.HasResourceInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("count").Instance(addrs.IntKey(0)).Absolute(addrs.RootModuleInstance); !set.HasResourceInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("for_each").Instance(addrs.StringKey("c")).Absolute(addrs.RootModuleInstance); !set.HasResourceInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("single").Instance(addrs.NoKey).Absolute(addrs.RootModuleInstance.Child("single", addrs.NoKey)); !set.HasResourceInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("count").Instance(addrs.IntKey(0)).Absolute(addrs.RootModuleInstance.Child("for_each", addrs.StringKey("b"))); !set.HasResourceInstance(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("single").Instance(addrs.IntKey(0)).Absolute(addrs.RootModuleInstance); set.HasResourceInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := rAddr("single").Instance(addrs.StringKey("")).Absolute(addrs.RootModuleInstance); set.HasResourceInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := rAddr("count").Instance(addrs.NoKey).Absolute(addrs.RootModuleInstance); set.HasResourceInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := rAddr("count").Instance(addrs.StringKey("")).Absolute(addrs.RootModuleInstance); set.HasResourceInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := rAddr("count").Instance(addrs.IntKey(1)).Absolute(addrs.RootModuleInstance); set.HasResourceInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := rAddr("single").Instance(addrs.NoKey).Absolute(addrs.RootModuleInstance.Child("single", addrs.IntKey(0))); set.HasResourceInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := rAddr("count").Instance(addrs.IntKey(0)).Absolute(addrs.RootModuleInstance.Child("for_each", addrs.StringKey("a"))); set.HasResourceInstance(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
// HasResource tests
if input := rAddr("single").Absolute(addrs.RootModuleInstance); !set.HasResource(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("count").Absolute(addrs.RootModuleInstance); !set.HasResource(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("for_each").Absolute(addrs.RootModuleInstance); !set.HasResource(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("single").Absolute(addrs.RootModuleInstance.Child("single", addrs.NoKey)); !set.HasResource(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("count").Absolute(addrs.RootModuleInstance.Child("for_each", addrs.StringKey("a"))); !set.HasResource(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("count").Absolute(addrs.RootModuleInstance.Child("for_each", addrs.StringKey("b"))); !set.HasResource(input) {
t.Errorf("missing %T %s", input, input.String())
}
if input := rAddr("nonexist").Absolute(addrs.RootModuleInstance); set.HasResource(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
if input := rAddr("count").Absolute(addrs.RootModuleInstance.Child("for_each", addrs.StringKey("nonexist"))); set.HasResource(input) {
t.Errorf("unexpected %T %s", input, input.String())
}
// ensure we can lookup non-existent addrs in a set without panic
if set.InstancesForModule(addrs.RootModule.Child("missing")) != nil {
t.Error("unexpected instances from missing module")
}
}