mirror of
https://github.com/opentffoundation/opentf.git
synced 2025-12-19 17:59:05 -05:00
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>
101 lines
2.2 KiB
Go
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
|
|
`
|