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
| package testing
|
| import (
| "testing"
|
| "github.com/stretchr/testify/assert"
| "google.golang.org/protobuf/reflect/protoreflect"
|
| "github.com/v2fly/v2ray-core/v5/common/protoext"
| )
|
| func TestMessageOpt(t *testing.T) {
| msg := TestingMessage{}
| opt, err := protoext.GetMessageOptions(msg.ProtoReflect().Descriptor())
| assert.Nil(t, err)
| assert.EqualValues(t, []string{"demo", "demo2"}, opt.Type)
| }
|
| func TestFieldOpt(t *testing.T) {
| msg := TestingMessage{
| TestField: "Test",
| }
| msgreflect := msg.ProtoReflect()
| msgreflect.Range(func(descriptor protoreflect.FieldDescriptor, value protoreflect.Value) bool {
| opt, err := protoext.GetFieldOptions(descriptor)
| assert.Nil(t, err)
| assert.EqualValues(t, []string{"test", "test2"}, opt.AllowedValues)
| return true
| })
| }
|
|