1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// Copyright 2025 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package protodesc
import (
"testing"
"google.golang.org/protobuf/internal/genid"
"google.golang.org/protobuf/proto"
"google.golang.org/protobuf/reflect/protoreflect"
"google.golang.org/protobuf/reflect/protoregistry"
"google.golang.org/protobuf/types/descriptorpb"
"google.golang.org/protobuf/types/dynamicpb"
"google.golang.org/protobuf/types/gofeaturespb"
)
func TestGoFeatures_NotExpectedType(t *testing.T) {
md := (*gofeaturespb.GoFeatures)(nil).ProtoReflect().Descriptor()
gf := dynamicpb.NewMessage(md)
opaque := protoreflect.ValueOfEnum(gofeaturespb.GoFeatures_API_OPAQUE.Number())
gf.Set(md.Fields().ByNumber(genid.GoFeatures_ApiLevel_field_number), opaque)
dynamicExt := dynamicpb.NewExtensionType(gofeaturespb.E_Go.TypeDescriptor().Descriptor())
extFile, err := NewFile(&descriptorpb.FileDescriptorProto{
Name: proto.String("test.proto"),
Dependency: []string{"google/protobuf/descriptor.proto"},
Extension: []*descriptorpb.FieldDescriptorProto{
{
Name: proto.String("ext1002"),
Number: proto.Int32(int32(gofeaturespb.E_Go.TypeDescriptor().Number())),
Type: descriptorpb.FieldDescriptorProto_TYPE_STRING.Enum(),
Label: descriptorpb.FieldDescriptorProto_LABEL_OPTIONAL.Enum(),
Extendee: proto.String(".google.protobuf.FeatureSet"),
},
},
}, protoregistry.GlobalFiles)
if err != nil {
t.Fatal(err)
}
nonMessageExt := dynamicpb.NewExtensionType(extFile.Extensions().Get(0))
extFdProto := ToFieldDescriptorProto(gofeaturespb.E_Go.TypeDescriptor().Descriptor())
extMsgProto := ToDescriptorProto(gofeaturespb.E_Go.TypeDescriptor().Descriptor().Message())
extMsgProto.Field = extMsgProto.Field[:1] // remove all but first field from the message
extFile, err = NewFile(&descriptorpb.FileDescriptorProto{
Name: proto.String("google/protobuf/go_features.proto"),
Package: proto.String("pb"),
Dependency: []string{"google/protobuf/descriptor.proto"},
Extension: []*descriptorpb.FieldDescriptorProto{extFdProto},
MessageType: []*descriptorpb.DescriptorProto{extMsgProto},
}, protoregistry.GlobalFiles)
if err != nil {
t.Fatal(err)
}
missingFieldsExt := dynamicpb.NewExtensionType(extFile.Extensions().Get(0))
gfMissingFields := dynamicpb.NewMessage(extFile.Messages().Get(0))
gfPresentField := gfMissingFields.Descriptor().Fields().Get(0)
gfMissingFields.Set(gfPresentField, gfMissingFields.NewField(gfPresentField))
testCases := []struct {
name string
ext protoreflect.ExtensionType
val any
}{
{
name: "dynamic_message",
ext: dynamicExt,
val: gf,
},
{
name: "not_a_message",
ext: nonMessageExt,
val: "abc",
},
{
name: "message_missing_fields",
ext: missingFieldsExt,
val: gfMissingFields,
},
}
for _, tc := range testCases {
tc := tc
t.Run(tc.name, func(t *testing.T) {
featureSet := &descriptorpb.FeatureSet{}
proto.SetExtension(featureSet, tc.ext, tc.val)
fd := &descriptorpb.FileDescriptorProto{
Name: proto.String("a.proto"),
Dependency: []string{
"google/protobuf/go_features.proto",
},
Edition: descriptorpb.Edition_EDITION_2023.Enum(),
Syntax: proto.String("editions"),
Options: &descriptorpb.FileOptions{
Features: featureSet,
},
}
fds := &descriptorpb.FileDescriptorSet{
File: []*descriptorpb.FileDescriptorProto{
ToFileDescriptorProto(descriptorpb.File_google_protobuf_descriptor_proto),
ToFileDescriptorProto(gofeaturespb.File_google_protobuf_go_features_proto),
fd,
},
}
if _, err := NewFiles(fds); err != nil {
t.Fatal(err)
}
})
}
}
|
