fix test

pulumi · Jul 23, 2024 · e316c27 · e316c27
1 parent 3e4d5d7
commit e316c27
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pf/internal/schemashim/custom_type_test.go b/pf/internal/schemashim/custom_type_test.go
@@ -24,7 +24,7 @@ import (
 	"github.com/stretchr/testify/assert"
 
 	"github.com/pulumi/pulumi-terraform-bridge/pf/internal/pfutils"
-	"github.com/pulumi/pulumi-terraform-bridge/v3/pkg/tfshim"
+	shim "github.com/pulumi/pulumi-terraform-bridge/v3/pkg/tfshim"
 )
 
 // This example is taken from aws:resourceexplorer/index:Index "timeouts" property.
@@ -109,10 +109,10 @@ func TestCustomListAttribute(t *testing.T) {
 	shimmed := &attrSchema{"key", pfutils.FromAttrLike(raw)}
 	assert.Equal(t, shim.TypeList, shimmed.Type())
 	assert.NotNil(t, shimmed.Elem())
-	_, isPseudoResource := shimmed.Elem().(shim.Schema)
+	_, isPseudoResource := shimmed.Elem().(shim.Resource)
 	assert.Truef(t, isPseudoResource, "expected shim.Elem() to be of type shim.Resource, encoding an object type")
 
-	create := shimmed.Elem().(shim.Schema).Elem().(shim.Resource).Schema().Get("filter_string")
+	create := shimmed.Elem().(shim.Resource).Schema().Get("filter_string")
 	assert.Equal(t, shim.TypeString, create.Type())
 }