@@ -19,12 +19,12 @@ import (
1919 "testing"
2020
2121 "github.com/aep-dev/api-linter/internal/desc"
22- "github.com/jhump/protoreflect/desc/builder"
22+
2323)
2424
2525func TestFileRule (t * testing.T ) {
2626 // Create a file descriptor with nothing in it.
27- fd , err := builder .NewFile ("test.proto" ).Build ()
27+ fd , err := testutils .NewFile (t , "test.proto" ).Build ()
2828 if err != nil {
2929 t .Fatalf ("Could not build file descriptor: %q" , err )
3030 }
@@ -50,10 +50,10 @@ func TestFileRule(t *testing.T) {
5050
5151func TestMessageRule (t * testing.T ) {
5252 // Create a file descriptor with two messages in it.
53- fd , err := builder .NewFile ("test.proto" ).AddMessage (
54- builder .NewMessage ("Book" ),
53+ fd , err := testutils .NewFile (t , "test.proto" ).AddMessage (
54+ testutils .NewMessage (t , "Book" ),
5555 ).AddMessage (
56- builder .NewMessage ("Author" ),
56+ testutils .NewMessage (t , "Author" ),
5757 ).Build ()
5858 if err != nil {
5959 t .Fatalf ("Failed to build file descriptor." )
@@ -82,8 +82,8 @@ func TestMessageRule(t *testing.T) {
8282// Establish that nested messages are tested.
8383func TestMessageRuleNested (t * testing.T ) {
8484 // Create a file descriptor with a message and nested message in it.
85- fd , err := builder .NewFile ("test.proto" ).AddMessage (
86- builder .NewMessage ("Book" ).AddNestedMessage (builder .NewMessage ("Author" )),
85+ fd , err := testutils .NewFile (t , "test.proto" ).AddMessage (
86+ testutils .NewMessage (t , "Book" ).AddNestedMessage (testutils .NewMessage (t , "Author" )),
8787 ).Build ()
8888 if err != nil {
8989 t .Fatalf ("Failed to build file descriptor." )
@@ -111,11 +111,11 @@ func TestMessageRuleNested(t *testing.T) {
111111
112112func TestFieldRule (t * testing.T ) {
113113 // Create a file descriptor with one message and two fields in that message.
114- fd , err := builder .NewFile ("test.proto" ).AddMessage (
115- builder .NewMessage ("Book" ).AddField (
116- builder . NewField ( "title" , builder .FieldTypeString () ),
114+ fd , err := testutils .NewFile (t , "test.proto" ).AddMessage (
115+ testutils .NewMessage (t , "Book" ).AddField (
116+ /* field: "title", testutils .FieldTypeString( */ ),
117117 ).AddField (
118- builder . NewField ( "edition_count" , builder .FieldTypeInt32 () ),
118+ /* field: "edition_count", testutils .FieldTypeInt32( */ ),
119119 ),
120120 ).Build ()
121121 if err != nil {
@@ -144,7 +144,7 @@ func TestFieldRule(t *testing.T) {
144144
145145func TestServiceRule (t * testing.T ) {
146146 // Create a file descriptor with a service.
147- fd , err := builder .NewFile ("test.proto" ).AddService (
147+ fd , err := testutils .NewFile (t , "test.proto" ).AddService (
148148 builder .NewService ("Library" ),
149149 ).Build ()
150150 if err != nil {
@@ -170,18 +170,18 @@ func TestServiceRule(t *testing.T) {
170170
171171func TestMethodRule (t * testing.T ) {
172172 // Create a file descriptor with a service.
173- book := builder .RpcTypeMessage (builder .NewMessage ("Book" ), false )
174- fd , err := builder .NewFile ("test.proto" ).AddService (
173+ book := builder .RpcTypeMessage (testutils .NewMessage (t , "Book" ), false )
174+ fd , err := testutils .NewFile (t , "test.proto" ).AddService (
175175 builder .NewService ("Library" ).AddMethod (
176176 builder .NewMethod (
177177 "GetBook" ,
178- builder .RpcTypeMessage (builder .NewMessage ("GetBookRequest" ), false ),
178+ builder .RpcTypeMessage (testutils .NewMessage (t , "GetBookRequest" ), false ),
179179 book ,
180180 ),
181181 ).AddMethod (
182182 builder .NewMethod (
183183 "CreateBook" ,
184- builder .RpcTypeMessage (builder .NewMessage ("CreateBookRequest" ), false ),
184+ builder .RpcTypeMessage (testutils .NewMessage (t , "CreateBookRequest" ), false ),
185185 book ,
186186 ),
187187 ),
@@ -212,10 +212,10 @@ func TestMethodRule(t *testing.T) {
212212
213213func TestEnumRule (t * testing.T ) {
214214 // Create a file descriptor with top-level enums.
215- fd , err := builder .NewFile ("test.proto" ).AddEnum (
216- builder . NewEnum ( "Format" ) .AddValue (builder . NewEnumValue ( "PDF" ) ),
215+ fd , err := testutils .NewFile (t , "test.proto" ).AddEnum (
216+ /* enum: "Format" */ .AddValue (/* enum_value: "PDF" */ ),
217217 ).AddEnum (
218- builder . NewEnum ( "Edition" ) .AddValue (builder . NewEnumValue ( "PUBLISHER_ONLY" ) ),
218+ /* enum: "Edition" */ .AddValue (/* enum_value: "PUBLISHER_ONLY" */ ),
219219 ).Build ()
220220 if err != nil {
221221 t .Fatalf ("Error building test proto:%s " , err )
@@ -242,8 +242,8 @@ func TestEnumRule(t *testing.T) {
242242
243243func TestEnumValueRule (t * testing.T ) {
244244 // Create a file descriptor with a top-level enum with values.
245- fd , err := builder .NewFile ("test.proto" ).AddEnum (
246- builder . NewEnum ( "Format" ) .AddValue (builder . NewEnumValue ( "YAML" )) .AddValue (builder . NewEnumValue ( "JSON" ) ),
245+ fd , err := testutils .NewFile (t , "test.proto" ).AddEnum (
246+ /* enum: "Format" */ .AddValue (/* enum_value: "YAML" */ ) .AddValue (/* enum_value: "JSON" */ ),
247247 ).Build ()
248248 if err != nil {
249249 t .Fatalf ("Error building test proto:%s " , err )
@@ -270,11 +270,11 @@ func TestEnumValueRule(t *testing.T) {
270270
271271func TestEnumRuleNested (t * testing.T ) {
272272 // Create a file descriptor with top-level enums.
273- fd , err := builder .NewFile ("test.proto" ).AddMessage (
274- builder .NewMessage ("Book" ).AddNestedEnum (
275- builder . NewEnum ( "Format" ) .AddValue (builder . NewEnumValue ( "PDF" ) ),
273+ fd , err := testutils .NewFile (t , "test.proto" ).AddMessage (
274+ testutils .NewMessage (t , "Book" ).AddNestedEnum (
275+ /* enum: "Format" */ .AddValue (/* enum_value: "PDF" */ ),
276276 ).AddNestedEnum (
277- builder . NewEnum ( "Edition" ) .AddValue (builder . NewEnumValue ( "PUBLISHER_ONLY" ) ),
277+ /* enum: "Edition" */ .AddValue (/* enum_value: "PUBLISHER_ONLY" */ ),
278278 ),
279279 ).Build ()
280280 if err != nil {
@@ -302,22 +302,22 @@ func TestEnumRuleNested(t *testing.T) {
302302
303303func TestDescriptorRule (t * testing.T ) {
304304 // Create a file with one of everything in it.
305- book := builder .NewMessage ("Book" ).AddNestedEnum (
306- builder . NewEnum ( "Format" ) .AddValue (
307- builder . NewEnumValue ( "FORMAT_UNSPECIFIED" ) ,
308- ).AddValue (builder . NewEnumValue ( "PAPERBACK" ) ),
309- ).AddField (builder . NewField ( "name" , builder .FieldTypeString () )).AddNestedMessage (
310- builder .NewMessage ("Author" ),
305+ book := testutils .NewMessage (t , "Book" ).AddNestedEnum (
306+ /* enum: "Format" */ .AddValue (
307+ /* enum_value: "FORMAT_UNSPECIFIED" */ ,
308+ ).AddValue (/* enum_value: "PAPERBACK" */ ),
309+ ).AddField (/* field: "name", testutils .FieldTypeString( */ )).AddNestedMessage (
310+ testutils .NewMessage (t , "Author" ),
311311 )
312- fd , err := builder .NewFile ("library.proto" ).AddMessage (book ).AddService (
312+ fd , err := testutils .NewFile (t , "library.proto" ).AddMessage (book ).AddService (
313313 builder .NewService ("Library" ).AddMethod (
314314 builder .NewMethod (
315315 "ConjureBook" ,
316316 builder .RpcTypeMessage (book , false ),
317317 builder .RpcTypeMessage (book , false ),
318318 ),
319319 ),
320- ).AddEnum (builder . NewEnum ( "State" ) .AddValue (builder . NewEnumValue ( "AVAILABLE" ) )).Build ()
320+ ).AddEnum (/* enum: "State" */ .AddValue (/* enum_value: "AVAILABLE" */ )).Build ()
321321 if err != nil {
322322 t .Fatalf ("%v" , err )
323323 }
0 commit comments