mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2024-12-27 06:21:25 +00:00
c2b76eee36
Change-Id: I2db557669ada6e031140a09b3a92bd901220f8f3 Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/580975 LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com> Reviewed-by: Lasse Folger <lassefolger@google.com> Reviewed-by: Michael Stapelberg <stapelberg@google.com> Auto-Submit: Michael Stapelberg <stapelberg@google.com>
101 lines
3.6 KiB
Go
101 lines
3.6 KiB
Go
// Copyright 2018 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 main
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"google.golang.org/protobuf/encoding/prototext"
|
|
"google.golang.org/protobuf/internal/genid"
|
|
"google.golang.org/protobuf/proto"
|
|
"google.golang.org/protobuf/testing/protocmp"
|
|
|
|
"google.golang.org/protobuf/types/descriptorpb"
|
|
)
|
|
|
|
func TestAnnotations(t *testing.T) {
|
|
sourceFile, err := os.ReadFile("testdata/annotations/annotations.pb.go")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
metaFile, err := os.ReadFile("testdata/annotations/annotations.pb.go.meta")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
gotInfo := &descriptorpb.GeneratedCodeInfo{}
|
|
if err := prototext.Unmarshal(metaFile, gotInfo); err != nil {
|
|
t.Fatalf("can't parse meta file: %v", err)
|
|
}
|
|
|
|
wantInfo := &descriptorpb.GeneratedCodeInfo{}
|
|
for _, want := range []struct {
|
|
prefix, text, suffix string
|
|
annotation *descriptorpb.GeneratedCodeInfo_Annotation
|
|
}{{
|
|
"type ", "AnnotationsTestEnum", " int32",
|
|
&descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Path: []int32{int32(genid.FileDescriptorProto_EnumType_field_number), 0},
|
|
},
|
|
}, {
|
|
"\t", "AnnotationsTestEnum_ANNOTATIONS_TEST_ENUM_VALUE", " AnnotationsTestEnum = 0",
|
|
&descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Path: []int32{int32(genid.FileDescriptorProto_EnumType_field_number), 0, int32(genid.EnumDescriptorProto_Value_field_number), 0},
|
|
},
|
|
}, {
|
|
"type ", "AnnotationsTestMessage", " struct {",
|
|
&descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Path: []int32{int32(genid.FileDescriptorProto_MessageType_field_number), 0},
|
|
},
|
|
}, {
|
|
"\t", "AnnotationsTestField", " ",
|
|
&descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Path: []int32{int32(genid.FileDescriptorProto_MessageType_field_number), 0, int32(genid.DescriptorProto_Field_field_number), 0},
|
|
},
|
|
}, {
|
|
"\t", "XXX_weak_M", " ",
|
|
&descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Path: []int32{int32(genid.FileDescriptorProto_MessageType_field_number), 0, int32(genid.DescriptorProto_Field_field_number), 1},
|
|
},
|
|
}, {
|
|
"func (x *AnnotationsTestMessage) ", "GetAnnotationsTestField", "() string {",
|
|
&descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Path: []int32{int32(genid.FileDescriptorProto_MessageType_field_number), 0, int32(genid.DescriptorProto_Field_field_number), 0},
|
|
},
|
|
}, {
|
|
"func (x *AnnotationsTestMessage) ", "GetM", "() proto.Message {",
|
|
&descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Path: []int32{int32(genid.FileDescriptorProto_MessageType_field_number), 0, int32(genid.DescriptorProto_Field_field_number), 1},
|
|
},
|
|
}, {
|
|
"func (x *AnnotationsTestMessage) ", "SetM", "(v proto.Message) {",
|
|
&descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Path: []int32{int32(genid.FileDescriptorProto_MessageType_field_number), 0, int32(genid.DescriptorProto_Field_field_number), 1},
|
|
Semantic: descriptorpb.GeneratedCodeInfo_Annotation_SET.Enum(),
|
|
},
|
|
}} {
|
|
s := want.prefix + want.text + want.suffix
|
|
pos := bytes.Index(sourceFile, []byte(s))
|
|
if pos < 0 {
|
|
t.Errorf("source file does not contain: %v", s)
|
|
continue
|
|
}
|
|
begin := pos + len(want.prefix)
|
|
end := begin + len(want.text)
|
|
a := &descriptorpb.GeneratedCodeInfo_Annotation{
|
|
Begin: proto.Int32(int32(begin)),
|
|
End: proto.Int32(int32(end)),
|
|
SourceFile: proto.String("cmd/protoc-gen-go/testdata/annotations/annotations.proto"),
|
|
}
|
|
proto.Merge(a, want.annotation)
|
|
wantInfo.Annotation = append(wantInfo.Annotation, a)
|
|
}
|
|
if diff := cmp.Diff(wantInfo, gotInfo, protocmp.Transform()); diff != "" {
|
|
t.Fatalf("unexpected annotations for annotations.proto (-want +got):\n%s", diff)
|
|
}
|
|
}
|