From 6cb801cf173e2ffae1a3503368ed114d86a2b53f Mon Sep 17 00:00:00 2001
From: Anton Tayanovskyy <anton@pulumi.com>
Date: Tue, 23 Nov 2021 16:52:02 -0500
Subject: [PATCH] Add a unit test to check Array/Index (#8486)

---
 .../go-extras/tests/go_test.go                | 52 +++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 pkg/codegen/internal/test/testdata/resource-property-overlap/go-extras/tests/go_test.go

diff --git a/pkg/codegen/internal/test/testdata/resource-property-overlap/go-extras/tests/go_test.go b/pkg/codegen/internal/test/testdata/resource-property-overlap/go-extras/tests/go_test.go
new file mode 100644
index 000000000..3dce232da
--- /dev/null
+++ b/pkg/codegen/internal/test/testdata/resource-property-overlap/go-extras/tests/go_test.go
@@ -0,0 +1,52 @@
+package tests
+
+import (
+	"sync"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/pulumi/pulumi/sdk/v3/go/common/resource"
+	"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
+
+	"resource-property-overlap/example"
+)
+
+// Tests that XArray{x}.ToXArrayOutput().Index(pulumi.Int(0)) == x.
+func TestArrayOutputIndex(t *testing.T) {
+	err := pulumi.RunErr(func(ctx *pulumi.Context) error {
+
+		r1, err := example.NewRec(ctx, "rec1", &example.RecArgs{})
+		if err != nil {
+			return err
+		}
+
+		r1o := r1.ToRecOutput()
+
+		r2o := example.RecArray{r1o}.ToRecArrayOutput().
+			Index(pulumi.Int(0))
+
+		wg := &sync.WaitGroup{}
+		wg.Add(1)
+
+		pulumi.All(r1o, r2o).ApplyT(func(xs []interface{}) int {
+			assert.Equal(t, xs[0], xs[1])
+			wg.Done()
+			return 0
+		})
+
+		wg.Wait()
+		return nil
+	}, pulumi.WithMocks("project", "stack", mocks(0)))
+	assert.NoError(t, err)
+}
+
+type mocks int
+
+func (mocks) NewResource(args pulumi.MockResourceArgs) (string, resource.PropertyMap, error) {
+	return args.Name + "_id", args.Inputs, nil
+}
+
+func (mocks) Call(args pulumi.MockCallArgs) (resource.PropertyMap, error) {
+	return args.Args, nil
+}