| Index: grpc/cmd/rpc/printer_test.go
|
| diff --git a/grpc/cmd/rpc/printer_test.go b/grpc/cmd/rpc/printer_test.go
|
| index 892106884956eba9d5dd850c034cfb4988b9da35..3c0b9ba28aa25d8b54fd941978c6e28cccf6ce0a 100644
|
| --- a/grpc/cmd/rpc/printer_test.go
|
| +++ b/grpc/cmd/rpc/printer_test.go
|
| @@ -10,9 +10,11 @@ import (
|
| "strings"
|
| "testing"
|
|
|
| + "github.com/luci/luci-go/common/proto/google/descutil"
|
| +
|
| "github.com/golang/protobuf/proto"
|
|
|
| - "github.com/luci/luci-go/common/proto/google/descriptor"
|
| + "google.golang.org/genproto/protobuf"
|
|
|
| . "github.com/smartystreets/goconvey/convey"
|
| )
|
| @@ -36,7 +38,7 @@ func TestPrinter(t *testing.T) {
|
| file := desc.File[0]
|
|
|
| getExpectedDef := func(path ...int) string {
|
| - loc := file.SourceCodeInfo.FindLocation(path)
|
| + loc := descutil.FindLocation(file.SourceCodeInfo, path)
|
| So(loc, ShouldNotBeNil)
|
|
|
| startLine := loc.Span[0]
|
| @@ -68,14 +70,14 @@ func TestPrinter(t *testing.T) {
|
|
|
| Convey("package", func() {
|
| printer.Package(file.GetPackage())
|
| - checkOutput(descriptor.NumberFileDescriptorProto_Package)
|
| + checkOutput(descutil.FileDescriptorProtoPackageTag)
|
| })
|
|
|
| Convey("service", func() {
|
| for i, s := range file.Service {
|
| Convey(s.GetName(), func() {
|
| printer.Service(s, i, -1)
|
| - checkOutput(descriptor.NumberFileDescriptorProto_Service, i)
|
| + checkOutput(descutil.FileDescriptorProtoServiceTag, i)
|
| })
|
| }
|
| })
|
| @@ -89,7 +91,7 @@ func TestPrinter(t *testing.T) {
|
|
|
| Convey("enum", func() {
|
| for i, e := range file.EnumType {
|
| - testEnum(e, []int{descriptor.NumberFileDescriptorProto_Enum, i})
|
| + testEnum(e, []int{descutil.FileDescriptorProtoEnumTag, i})
|
| }
|
| })
|
|
|
| @@ -102,16 +104,16 @@ func TestPrinter(t *testing.T) {
|
| checkOutput(path...)
|
| } else {
|
| for i, m := range m.NestedType {
|
| - testMsg(m, append(path, descriptor.NumberDescriptorProto_NestedType, i))
|
| + testMsg(m, append(path, descutil.DescriptorProtoNestedTypeTag, i))
|
| }
|
| for i, e := range m.EnumType {
|
| - testEnum(e, append(path, descriptor.NumberDescriptorProto_EnumType, i))
|
| + testEnum(e, append(path, descutil.DescriptorProtoEnumTypeTag, i))
|
| }
|
| }
|
| })
|
| }
|
| for i, m := range file.MessageType {
|
| - testMsg(m, []int{descriptor.NumberFileDescriptorProto_Message, i})
|
| + testMsg(m, []int{descutil.FileDescriptorProtoMessageTag, i})
|
| }
|
| })
|
| })
|
|
|