databroker: preserve data type when deleting changeset (#5540)

* databroker: preserve data type when deleting changeset

* use cs.now

pkg/grpc/databroker/changeset_test.go

@@ -0,0 +1,52 @@
+package databroker_test
+
+import (
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	"github.com/stretchr/testify/assert"
+	"google.golang.org/protobuf/testing/protocmp"
+	"google.golang.org/protobuf/types/known/structpb"
+
+	"github.com/pomerium/datasource/pkg/directory"
+	"github.com/pomerium/pomerium/pkg/grpc/databroker"
+	"github.com/pomerium/pomerium/pkg/protoutil"
+)
+
+func TestGetChangeset(t *testing.T) {
+	t.Parallel()
+
+	rsb1 := databroker.RecordSetBundle{}
+	rsb2 := databroker.RecordSetBundle{}
+	updates := databroker.GetChangeSet(rsb1, rsb2, func(record1, record2 *databroker.Record) bool {
+		return cmp.Equal(record1, record2, protocmp.Transform())
+	})
+	assert.Len(t, updates, 0)
+
+	rsb1 = databroker.RecordSetBundle{}
+	rsb1.Add(&databroker.Record{
+		Type: directory.UserRecordType,
+		Id:   "user-1",
+		Data: protoutil.NewAny(mustNewStruct(map[string]any{
+			"email": "user-1@example.com",
+		})),
+	})
+	rsb2 = databroker.RecordSetBundle{}
+	updates = databroker.GetChangeSet(rsb1, rsb2, func(record1, record2 *databroker.Record) bool {
+		return cmp.Equal(record1, record2, protocmp.Transform())
+	})
+	if assert.Len(t, updates, 1) {
+		assert.Equal(t, directory.UserRecordType, updates[0].GetType())
+		assert.Equal(t, "type.googleapis.com/google.protobuf.Struct", updates[0].GetData().GetTypeUrl(),
+			"should preserve data type")
+		assert.NotNil(t, updates[0].GetDeletedAt())
+	}
+}
+
+func mustNewStruct(m map[string]any) *structpb.Struct {
+	s, err := structpb.NewStruct(m)
+	if err != nil {
+		panic(err)
+	}
+	return s
+}