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