mirror of
https://github.com/opentofu/opentofu.git
synced 2024-12-28 01:41:48 -06:00
217 lines
11 KiB
Go
217 lines
11 KiB
Go
// Copyright (c) The OpenTofu Authors
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
// Copyright (c) 2023 HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package instances
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/opentofu/opentofu/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")
|
|
}
|
|
}
|