From 9d70b6ade6bb22416733e2c9d7016a4cb5027ddf Mon Sep 17 00:00:00 2001 From: Gonzalo Ortiz Date: Fri, 29 Sep 2023 17:44:12 +0200 Subject: [PATCH] Changes to make test use proto3 --- .../protobuf/ProtoBufRecordExtractorSimpleTest.java | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pinot-plugins/pinot-input-format/pinot-protobuf/src/test/java/org/apache/pinot/plugin/inputformat/protobuf/ProtoBufRecordExtractorSimpleTest.java b/pinot-plugins/pinot-input-format/pinot-protobuf/src/test/java/org/apache/pinot/plugin/inputformat/protobuf/ProtoBufRecordExtractorSimpleTest.java index 470192f4b73..10966793ea3 100644 --- a/pinot-plugins/pinot-input-format/pinot-protobuf/src/test/java/org/apache/pinot/plugin/inputformat/protobuf/ProtoBufRecordExtractorSimpleTest.java +++ b/pinot-plugins/pinot-input-format/pinot-protobuf/src/test/java/org/apache/pinot/plugin/inputformat/protobuf/ProtoBufRecordExtractorSimpleTest.java @@ -4,6 +4,7 @@ import com.google.protobuf.Descriptors; import com.google.protobuf.DynamicMessage; import com.google.protobuf.Message; +import java.util.Locale; import org.apache.pinot.spi.data.readers.GenericRow; import org.apache.pinot.spi.data.readers.RecordExtractorConfig; import org.testng.Assert; @@ -38,6 +39,7 @@ public void init() .build(); DescriptorProtos.FileDescriptorProto build = DescriptorProtos.FileDescriptorProto.newBuilder() .setName("withOptional") + .setSyntax(Descriptors.FileDescriptor.Syntax.PROTO3.name().toLowerCase(Locale.US)) .addMessageType(DescriptorProtos.DescriptorProto.newBuilder() .setName("myMessage") .addField(myOptionalInt) @@ -45,6 +47,7 @@ public void init() ).build(); _fileDescriptor = Descriptors.FileDescriptor.buildFrom(build, new Descriptors.FileDescriptor[] {}); + _fileDescriptor.getSyntax(); _messageDescriptor = _fileDescriptor.findMessageTypeByName("myMessage"); _optionalDescriptor = _messageDescriptor.findFieldByName("myOptionalInt"); _nonOptionalDescriptor = _messageDescriptor.findFieldByName("myNonOptionalInt");