Files
opentf/internal/tofu/transform_root_test.go
Martin Atkins 75c7834ca6 tofu: GraphTransformer.Transform takes context.Context
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>
2025-05-08 07:16:09 -07:00

101 lines
2.2 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 TestRootTransformer(t *testing.T) {
t.Run("many nodes", func(t *testing.T) {
mod := testModule(t, "transform-root-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 := &MissingProviderTransformer{}
if err := transform.Transform(t.Context(), &g); err != nil {
t.Fatalf("err: %s", err)
}
}
{
transform := &ProviderTransformer{}
if err := transform.Transform(t.Context(), &g); err != nil {
t.Fatalf("err: %s", err)
}
}
{
transform := &RootTransformer{}
if err := transform.Transform(t.Context(), &g); err != nil {
t.Fatalf("err: %s", err)
}
}
actual := strings.TrimSpace(g.String())
expected := strings.TrimSpace(testTransformRootBasicStr)
if actual != expected {
t.Fatalf("wrong result\n\ngot:\n%s\n\nwant:\n%s", actual, expected)
}
root, err := g.Root()
if err != nil {
t.Fatalf("err: %s", err)
}
if _, ok := root.(graphNodeRoot); !ok {
t.Fatalf("bad: %#v", root)
}
})
t.Run("only one initial node", func(t *testing.T) {
g := Graph{Path: addrs.RootModuleInstance}
g.Add("foo")
addRootNodeToGraph(&g)
got := strings.TrimSpace(g.String())
want := strings.TrimSpace(`
foo
root
foo
`)
if got != want {
t.Errorf("wrong final graph\ngot:\n%s\nwant:\n%s", got, want)
}
})
t.Run("graph initially empty", func(t *testing.T) {
g := Graph{Path: addrs.RootModuleInstance}
addRootNodeToGraph(&g)
got := strings.TrimSpace(g.String())
want := `root`
if got != want {
t.Errorf("wrong final graph\ngot:\n%s\nwant:\n%s", got, want)
}
})
}
const testTransformRootBasicStr = `
aws_instance.foo
provider["registry.opentofu.org/hashicorp/aws"]
do_droplet.bar
provider["registry.opentofu.org/hashicorp/do"]
provider["registry.opentofu.org/hashicorp/aws"]
provider["registry.opentofu.org/hashicorp/do"]
root
aws_instance.foo
do_droplet.bar
`