0ct0pu5/kafka-ui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update complex schema in test

Browse source 
Make schemas more consistent
This commit is contained in:
kostasdizas 2023-08-05 21:08:45 +01:00
parent 504962447b
commit 4b7d89a198
No known key found for this signature in database
GPG key ID: 97876C44A0C60912
1 changed files with 21 additions and 50 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

71
kafka-ui-api/src/test/java/com/provectus/kafka/ui/serdes/builtin/ProtobufRawSerdeTest.java
View file

@ -25,20 +25,20 @@ class ProtobufRawSerdeTest {
}
@SneakyThrows
Descriptors.Descriptor getDescriptor() {
ProtobufSchema getSampleSchema() {
return new ProtobufSchema(
"""
syntax = "proto3";
message MyMessage {
message Message1 {
int32 my_field = 1;
}
"""
).toDescriptor("MyMessage");
);
}
@SneakyThrows
private byte[] getProtobufMessage() {
DynamicMessage.Builder builder = DynamicMessage.newBuilder(getDescriptor());
DynamicMessage.Builder builder = DynamicMessage.newBuilder(getSampleSchema().toDescriptor("Message1"));
builder.setField(builder.getDescriptorForType().findFieldByName("my_field"), 5);
return builder.build().toByteArray();
}
@ -73,55 +73,26 @@ class ProtobufRawSerdeTest {
.hasMessageContaining("Cannot read the array length");
}
ProtobufSchema getSampleNestedSchema() {
return new ProtobufSchema(
"""
syntax = "proto3";
message Message2 {
int32 my_nested_field = 1;
}
message Message1 {
int32 my_field = 1;
Message2 my_nested_message = 2;
}
"""
);
}
@SneakyThrows
private byte[] getComplexProtobufMessage() {
DescriptorProtos.FileDescriptorProto fileDescriptorProto = DescriptorProtos.FileDescriptorProto
.newBuilder()
.setSyntax("proto3")
.setName("test.proto")
.addMessageType(
DescriptorProtos.DescriptorProto.newBuilder()
.setName("MyMessage")
.addField(
DescriptorProtos.FieldDescriptorProto.newBuilder()
.setName("my_field")
.setNumber(1)
.setType(DescriptorProtos.FieldDescriptorProto.Type.TYPE_INT32)
.build()
)
.addField(
DescriptorProtos.FieldDescriptorProto.newBuilder()
.setName("my_nested_message")
.setNumber(2)
.setType(DescriptorProtos.FieldDescriptorProto.Type.TYPE_MESSAGE)
.setTypeName("MyNestedMessage")
.build()
)
.build()
)
.addMessageType(
DescriptorProtos.DescriptorProto.newBuilder()
.setName("MyNestedMessage")
.addField(
DescriptorProtos.FieldDescriptorProto.newBuilder()
.setName("my_nested_field")
.setNumber(1)
.setType(DescriptorProtos.FieldDescriptorProto.Type.TYPE_INT32)
.build()
)
.build()
)
.build();
Descriptors.FileDescriptor fileDescriptor = Descriptors.FileDescriptor.buildFrom(
fileDescriptorProto, new Descriptors.FileDescriptor[0]);
Descriptors.Descriptor descriptor = fileDescriptor.findMessageTypeByName("MyMessage");
Descriptors.Descriptor nestedDescriptor = fileDescriptor.findMessageTypeByName("MyNestedMessage");
DynamicMessage.Builder builder = DynamicMessage.newBuilder(descriptor);
DynamicMessage.Builder builder = DynamicMessage.newBuilder(getSampleNestedSchema().toDescriptor("Message1"));
builder.setField(builder.getDescriptorForType().findFieldByName("my_field"), 5);
DynamicMessage.Builder nestedBuilder = DynamicMessage.newBuilder(nestedDescriptor);
DynamicMessage.Builder nestedBuilder = DynamicMessage.newBuilder(getSampleNestedSchema().toDescriptor("Message2"));
nestedBuilder.setField(nestedBuilder.getDescriptorForType().findFieldByName("my_nested_field"), 10);
builder.setField(builder.getDescriptorForType().findFieldByName("my_nested_message"), nestedBuilder.build());