mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2024-12-28 00:19:55 +00:00
009e067ed8
Add the scalar package to reduce dependencies on the v1 proto runtime package. It may very well be the case that these functions should be exposed in the public API of v2, but that is not a decision we need to make now. Change-Id: Ifbc6d15311ba5837909ac72af47c630a80a142ef Reviewed-on: https://go-review.googlesource.com/c/151402 Reviewed-by: Herbie Ong <herbie@google.com>
95 lines
2.7 KiB
Go
95 lines
2.7 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.
|
|
|
|
// +build golden
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"flag"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
descpb "github.com/golang/protobuf/protoc-gen-go/descriptor"
|
|
"github.com/golang/protobuf/v2/internal/protogen/goldentest"
|
|
"github.com/golang/protobuf/v2/internal/scalar"
|
|
)
|
|
|
|
// Set --regenerate to regenerate the golden files.
|
|
var regenerate = flag.Bool("regenerate", false, "regenerate golden files")
|
|
|
|
func init() {
|
|
goldentest.Plugin(main)
|
|
}
|
|
|
|
func TestGolden(t *testing.T) {
|
|
goldentest.Run(t, *regenerate)
|
|
}
|
|
|
|
func TestAnnotations(t *testing.T) {
|
|
workdir, err := ioutil.TempDir("", "proto-test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(workdir)
|
|
|
|
goldentest.Protoc(t, []string{"--go_out=paths=source_relative,annotate_code:" + workdir, "-Itestdata/annotations", "testdata/annotations/annotations.proto"})
|
|
sourceFile, err := ioutil.ReadFile(filepath.Join(workdir, "annotations.pb.go"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
metaFile, err := ioutil.ReadFile(filepath.Join(workdir, "annotations.pb.go.meta"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
gotInfo := &descpb.GeneratedCodeInfo{}
|
|
if err := proto.UnmarshalText(string(metaFile), gotInfo); err != nil {
|
|
t.Fatalf("can't parse meta file: %v", err)
|
|
}
|
|
|
|
wantInfo := &descpb.GeneratedCodeInfo{}
|
|
for _, want := range []struct {
|
|
prefix, text, suffix string
|
|
path []int32
|
|
}{{
|
|
"type ", "AnnotationsTestEnum", " int32",
|
|
[]int32{5 /* enum_type */, 0},
|
|
}, {
|
|
"\t", "AnnotationsTestEnum_ANNOTATIONS_TEST_ENUM_VALUE", " AnnotationsTestEnum = 0",
|
|
[]int32{5 /* enum_type */, 0, 2 /* value */, 0},
|
|
}, {
|
|
"type ", "AnnotationsTestMessage", " struct {",
|
|
[]int32{4 /* message_type */, 0},
|
|
}, {
|
|
"\t", "AnnotationsTestField", " ",
|
|
[]int32{4 /* message_type */, 0, 2 /* field */, 0},
|
|
}, {
|
|
"func (m *AnnotationsTestMessage) ", "GetAnnotationsTestField", "() string {",
|
|
[]int32{4 /* message_type */, 0, 2 /* field */, 0},
|
|
}} {
|
|
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)
|
|
wantInfo.Annotation = append(wantInfo.Annotation, &descpb.GeneratedCodeInfo_Annotation{
|
|
Path: want.path,
|
|
Begin: scalar.Int32(int32(begin)),
|
|
End: scalar.Int32(int32(end)),
|
|
SourceFile: scalar.String("annotations.proto"),
|
|
})
|
|
}
|
|
if !proto.Equal(gotInfo, wantInfo) {
|
|
t.Errorf("unexpected annotations for annotations.proto; got:\n%v\nwant:\n%v",
|
|
proto.MarshalTextString(gotInfo), proto.MarshalTextString(wantInfo))
|
|
}
|
|
}
|