improved test

databricks · Aug 16, 2023 · 534e480 · 534e480
1 parent 269e8a7
commit 534e480
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/bundle/tests/string_interpolation_test.go b/bundle/tests/string_interpolation_test.go
@@ -5,7 +5,8 @@ import (
 	"testing"
 
 	"github.com/databricks/cli/bundle"
-	"github.com/databricks/cli/bundle/phases"
+	"github.com/databricks/cli/bundle/config/interpolation"
+	"github.com/databricks/cli/bundle/config/variable"
 	"github.com/stretchr/testify/require"
 )
 
@@ -14,7 +15,11 @@ func TestBundleEnvironmentStringInterpolation(t *testing.T) {
 	b, err := bundle.Load(ctx, "./string_interpolation")
 	require.NoError(t, err)
 
-	init := phases.Initialize()
+	init := interpolation.Interpolate(
+		interpolation.IncludeLookupsInPath("bundle"),
+		interpolation.IncludeLookupsInPath("workspace"),
+		interpolation.IncludeLookupsInPath(variable.VariableReferencePrefix),
+	)
 	err = init.Apply(ctx, b)
 	require.NoError(t, err)