Hunter0x7c7
2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece
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
    })
}