Search in sources :

Example 1 with Nested

use of org.apache.beam.sdk.extensions.protobuf.Proto3SchemaMessages.Nested in project beam by apache.

the class ProtoDynamicMessageSchemaTest method testNestedRowToProto.

@Test
public void testNestedRowToProto() throws InvalidProtocolBufferException {
    ProtoDynamicMessageSchema schemaProvider = schemaFromDescriptor(Nested.getDescriptor());
    SerializableFunction<Row, DynamicMessage> fromRow = schemaProvider.getFromRowFunction();
    Nested proto = parseFrom(fromRow.apply(NESTED_ROW).toString(), Nested.newBuilder()).build();
    assertEquals(NESTED_PROTO, proto);
}
Also used : Nested(org.apache.beam.sdk.extensions.protobuf.Proto3SchemaMessages.Nested) Row(org.apache.beam.sdk.values.Row) DynamicMessage(com.google.protobuf.DynamicMessage) Test(org.junit.Test)

Aggregations

DynamicMessage (com.google.protobuf.DynamicMessage)1 Nested (org.apache.beam.sdk.extensions.protobuf.Proto3SchemaMessages.Nested)1 Row (org.apache.beam.sdk.values.Row)1 Test (org.junit.Test)1