mirror of
https://github.com/opentffoundation/opentf.git
synced 2025-12-19 17:59:05 -05:00
75c7834ca6
Most of our transformers are pure compute and so don't really have a strong need to generate trace spans under our current focus of only exposing user-facing concepts and external requests in our traces, but unfortunately some of them indirectly depend on provider schema, which in turn means that they can potentially be unlucky enough to be the trigger for making all of the provider requests needed to fill the schema cache and therefore would end up with provider request spans being reported beneath them. As usual with these interface updates, this initial change focuses only on changing the interface and updating its direct callers and implementers to match, without any further refactoring or attempts to plumb contexts to or from other functions that don't have them yet. That means there are a few new context.TODO() calls here that we'll tidy up in a later commit that hopefully won't involve all of the noise that is caused by changing an interface API. Signed-off-by: Martin Atkins <mart@degeneration.co.uk>
336 lines
8.6 KiB
Go
336 lines
8.6 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 tofu
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/opentofu/opentofu/internal/addrs"
|
|
)
|
|
|
|
func TestTargetsTransformer(t *testing.T) {
|
|
mod := testModule(t, "transform-targets-basic")
|
|
|
|
g := Graph{Path: addrs.RootModuleInstance}
|
|
{
|
|
tf := &ConfigTransformer{Config: mod}
|
|
if err := tf.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &AttachResourceConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &ReferenceTransformer{}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &TargetingTransformer{
|
|
Targets: []addrs.Targetable{
|
|
addrs.RootModuleInstance.Resource(
|
|
addrs.ManagedResourceMode, "aws_instance", "me",
|
|
),
|
|
},
|
|
}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
actual := strings.TrimSpace(g.String())
|
|
expected := strings.TrimSpace(`
|
|
aws_instance.me
|
|
aws_subnet.me
|
|
aws_subnet.me
|
|
aws_vpc.me
|
|
aws_vpc.me
|
|
`)
|
|
if actual != expected {
|
|
t.Fatalf("bad:\n\nexpected:\n%s\n\ngot:\n%s\n", expected, actual)
|
|
}
|
|
}
|
|
|
|
func TestTargetsTransformerExclude(t *testing.T) {
|
|
mod := testModule(t, "transform-targets-basic")
|
|
|
|
g := Graph{Path: addrs.RootModuleInstance}
|
|
{
|
|
tf := &ConfigTransformer{Config: mod}
|
|
if err := tf.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &AttachResourceConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &ReferenceTransformer{}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &TargetingTransformer{
|
|
Excludes: []addrs.Targetable{
|
|
addrs.RootModuleInstance.Resource(
|
|
addrs.ManagedResourceMode, "aws_instance", "me",
|
|
),
|
|
addrs.RootModuleInstance.Resource(
|
|
addrs.ManagedResourceMode, "aws_vpc", "notme",
|
|
),
|
|
addrs.RootModuleInstance.Resource(
|
|
addrs.ManagedResourceMode, "aws_subnet", "notme",
|
|
),
|
|
addrs.RootModuleInstance.Resource(
|
|
addrs.ManagedResourceMode, "aws_instance", "notme",
|
|
),
|
|
},
|
|
}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
actual := strings.TrimSpace(g.String())
|
|
expected := strings.TrimSpace(`
|
|
aws_subnet.me
|
|
aws_vpc.me
|
|
aws_vpc.me
|
|
`)
|
|
if actual != expected {
|
|
t.Fatalf("bad:\n\nexpected:\n%s\n\ngot:\n%s\n", expected, actual)
|
|
}
|
|
}
|
|
|
|
func TestTargetsTransformer_downstream(t *testing.T) {
|
|
mod := testModule(t, "transform-targets-downstream")
|
|
|
|
g := Graph{Path: addrs.RootModuleInstance}
|
|
{
|
|
transform := &ConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &AttachResourceConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &AttachResourceConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &OutputTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &ReferenceTransformer{}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &TargetingTransformer{
|
|
Targets: []addrs.Targetable{
|
|
addrs.RootModuleInstance.
|
|
Child("child", addrs.NoKey).
|
|
Child("grandchild", addrs.NoKey).
|
|
Resource(
|
|
addrs.ManagedResourceMode, "aws_instance", "foo",
|
|
),
|
|
},
|
|
}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
actual := strings.TrimSpace(g.String())
|
|
// Even though we only asked to target the grandchild resource, all of the
|
|
// outputs that descend from it are also targeted.
|
|
expected := strings.TrimSpace(`
|
|
module.child.module.grandchild.aws_instance.foo
|
|
module.child.module.grandchild.output.id (expand)
|
|
module.child.module.grandchild.aws_instance.foo
|
|
module.child.output.grandchild_id (expand)
|
|
module.child.module.grandchild.output.id (expand)
|
|
output.grandchild_id (expand)
|
|
module.child.output.grandchild_id (expand)
|
|
`)
|
|
if actual != expected {
|
|
t.Fatalf("bad:\n\nexpected:\n%s\n\ngot:\n%s\n", expected, actual)
|
|
}
|
|
}
|
|
|
|
func TestTargetsTransformer_downstreamExclude(t *testing.T) {
|
|
mod := testModule(t, "transform-targets-downstream")
|
|
|
|
g := Graph{Path: addrs.RootModuleInstance}
|
|
{
|
|
transform := &ConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &AttachResourceConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &AttachResourceConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &OutputTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &ReferenceTransformer{}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &TargetingTransformer{
|
|
Excludes: []addrs.Targetable{
|
|
addrs.RootModuleInstance.Resource(addrs.ManagedResourceMode, "aws_instance", "foo"),
|
|
addrs.RootModuleInstance.
|
|
Child("child", addrs.NoKey).
|
|
Resource(addrs.ManagedResourceMode, "aws_instance", "foo"),
|
|
},
|
|
}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
actual := strings.TrimSpace(g.String())
|
|
// Even though we only asked to exclude all resources in root and child, only including the grandchild resource
|
|
// all of the outputs that descend from it are also targeted.
|
|
expected := strings.TrimSpace(`
|
|
module.child.module.grandchild.aws_instance.foo
|
|
module.child.module.grandchild.output.id (expand)
|
|
module.child.module.grandchild.aws_instance.foo
|
|
module.child.output.grandchild_id (expand)
|
|
module.child.module.grandchild.output.id (expand)
|
|
output.grandchild_id (expand)
|
|
module.child.output.grandchild_id (expand)
|
|
`)
|
|
if actual != expected {
|
|
t.Fatalf("bad:\n\nexpected:\n%s\n\ngot:\n%s\n", expected, actual)
|
|
}
|
|
}
|
|
|
|
// This tests the TargetingTransformer targeting a whole module,
|
|
// rather than a resource within a module instance.
|
|
func TestTargetsTransformer_wholeModule(t *testing.T) {
|
|
mod := testModule(t, "transform-targets-downstream")
|
|
|
|
g := Graph{Path: addrs.RootModuleInstance}
|
|
{
|
|
transform := &ConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &AttachResourceConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &AttachResourceConfigTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &OutputTransformer{Config: mod}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &ReferenceTransformer{}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
{
|
|
transform := &TargetingTransformer{
|
|
Targets: []addrs.Targetable{
|
|
addrs.RootModule.
|
|
Child("child").
|
|
Child("grandchild"),
|
|
},
|
|
}
|
|
if err := transform.Transform(t.Context(), &g); err != nil {
|
|
t.Fatalf("%T failed: %s", transform, err)
|
|
}
|
|
}
|
|
|
|
actual := strings.TrimSpace(g.String())
|
|
// Even though we only asked to target the grandchild module, all of the
|
|
// outputs that descend from it are also targeted.
|
|
expected := strings.TrimSpace(`
|
|
module.child.module.grandchild.aws_instance.foo
|
|
module.child.module.grandchild.output.id (expand)
|
|
module.child.module.grandchild.aws_instance.foo
|
|
module.child.output.grandchild_id (expand)
|
|
module.child.module.grandchild.output.id (expand)
|
|
output.grandchild_id (expand)
|
|
module.child.output.grandchild_id (expand)
|
|
`)
|
|
if actual != expected {
|
|
t.Fatalf("bad:\n\nexpected:\n%s\n\ngot:\n%s\n", expected, actual)
|
|
}
|
|
}
|