1package grpcurl 2 3import ( 4 "bytes" 5 "io/ioutil" 6 "testing" 7 8 "github.com/golang/protobuf/proto" //lint:ignore SA1019 we have to import this because it appears in exported API 9 "google.golang.org/protobuf/types/descriptorpb" 10) 11 12func TestWriteProtoset(t *testing.T) { 13 exampleProtoset, err := loadProtoset("./internal/testing/example.protoset") 14 if err != nil { 15 t.Fatalf("failed to load example.protoset: %v", err) 16 } 17 testProtoset, err := loadProtoset("./internal/testing/test.protoset") 18 if err != nil { 19 t.Fatalf("failed to load test.protoset: %v", err) 20 } 21 22 mergedProtoset := &descriptorpb.FileDescriptorSet{ 23 File: append(exampleProtoset.File, testProtoset.File...), 24 } 25 26 descSrc, err := DescriptorSourceFromFileDescriptorSet(mergedProtoset) 27 if err != nil { 28 t.Fatalf("failed to create descriptor source: %v", err) 29 } 30 31 checkWriteProtoset(t, descSrc, exampleProtoset, "TestService") 32 checkWriteProtoset(t, descSrc, testProtoset, "testing.TestService") 33 checkWriteProtoset(t, descSrc, mergedProtoset, "TestService", "testing.TestService") 34} 35 36func loadProtoset(path string) (*descriptorpb.FileDescriptorSet, error) { 37 b, err := ioutil.ReadFile(path) 38 if err != nil { 39 return nil, err 40 } 41 var protoset descriptorpb.FileDescriptorSet 42 if err := proto.Unmarshal(b, &protoset); err != nil { 43 return nil, err 44 } 45 return &protoset, nil 46} 47 48func checkWriteProtoset(t *testing.T, descSrc DescriptorSource, protoset *descriptorpb.FileDescriptorSet, symbols ...string) { 49 var buf bytes.Buffer 50 if err := WriteProtoset(&buf, descSrc, symbols...); err != nil { 51 t.Fatalf("failed to write protoset: %v", err) 52 } 53 54 var result descriptorpb.FileDescriptorSet 55 if err := proto.Unmarshal(buf.Bytes(), &result); err != nil { 56 t.Fatalf("failed to unmarshal written protoset: %v", err) 57 } 58 59 if !proto.Equal(protoset, &result) { 60 t.Fatalf("written protoset not equal to input:\nExpecting: %s\nActual: %s", protoset, &result) 61 } 62} 63