Function TestProtoFiles

grpc/codegen/proto_test.go:19–56

(t *testing.T)

)

func TestProtoFiles(t *testing.T) {
	cases := []struct {
		Name string
		DSL  func()
	}{
		{"protofiles-unary-rpcs", testdata.UnaryRPCsDSL},
		{"protofiles-unary-rpc-no-payload", testdata.UnaryRPCNoPayloadDSL},
		{"protofiles-unary-rpc-no-result", testdata.UnaryRPCNoResultDSL},
		{"protofiles-server-streaming-rpc", testdata.ServerStreamingRPCDSL},
		{"protofiles-client-streaming-rpc", testdata.ClientStreamingRPCDSL},
		{"protofiles-bidirectional-streaming-rpc", testdata.BidirectionalStreamingRPCDSL},
		{"protofiles-same-service-and-message-name", testdata.MessageWithServiceNameDSL},
		{"protofiles-method-with-reserved-proto-name", testdata.MethodWithReservedNameDSL},
		{"protofiles-multiple-methods-same-return-type", testdata.MultipleMethodsSameResultCollectionDSL},
		{"protofiles-method-with-acronym", testdata.MethodWithAcronymDSL},
		{"protofiles-custom-package-name", testdata.ServiceWithPackageDSL},
		{"protofiles-struct-meta-type", testdata.StructMetaTypeDSL},
		{"protofiles-default-fields", testdata.DefaultFieldsDSL},
		{"protofiles-custom-message-name", testdata.CustomMessageNameDSL},
	}
	for _, c := range cases {
		t.Run(c.Name, func(t *testing.T) {
			root := RunGRPCDSL(t, c.DSL)
			services := CreateGRPCServices(root)
			fs := ProtoFiles("", services)
			if len(fs) != 1 {
				t.Fatalf("got %d files, expected one", len(fs))
			}
			sections := fs[0].SectionTemplates
			require.GreaterOrEqual(t, len(sections), 3)
			code := sectionCode(t, sections[1:]...)
			// testutil.AssertString handles line ending normalization internally
			testutil.AssertString(t, "testdata/golden/proto_"+c.Name+".proto.golden", code)
			fpath := codegen.CreateTempFile(t, code)
			assert.NoError(t, protoc(defaultProtocCmd, fpath, nil), "error occurred when compiling proto file %q", fpath)
		})
	}
}

func TestMessageDefSection(t *testing.T) {
	cases := []struct {