Search in sources :

Example 1 with StructuredRecordWritable

use of co.cask.cdap.etl.batch.StructuredRecordWritable in project cdap by caskdata.

the class StructuredRecordWritableTest method testNonAsciiString.

@Test
public void testNonAsciiString() throws IOException {
    Schema schema = Schema.recordOf("rec", Schema.Field.of("x", Schema.of(Schema.Type.STRING)));
    StructuredRecord record = StructuredRecord.builder(schema).set("x", "идыло").build();
    StructuredRecordWritable writableOut = new StructuredRecordWritable(record);
    ByteArrayOutputStream os = new ByteArrayOutputStream();
    DataOutput output = new DataOutputStream(os);
    writableOut.write(output);
    os.flush();
    StructuredRecordWritable writableIn = new StructuredRecordWritable();
    ByteArrayInputStream is = new ByteArrayInputStream(os.toByteArray());
    DataInput input = new DataInputStream(is);
    writableIn.readFields(input);
    Assert.assertEquals(writableIn.get(), record);
}
Also used : DataInput(java.io.DataInput) DataOutput(java.io.DataOutput) ByteArrayInputStream(java.io.ByteArrayInputStream) DataOutputStream(java.io.DataOutputStream) Schema(co.cask.cdap.api.data.schema.Schema) ByteArrayOutputStream(java.io.ByteArrayOutputStream) DataInputStream(java.io.DataInputStream) StructuredRecordWritable(co.cask.cdap.etl.batch.StructuredRecordWritable) StructuredRecord(co.cask.cdap.api.data.format.StructuredRecord) Test(org.junit.Test)

Aggregations

StructuredRecord (co.cask.cdap.api.data.format.StructuredRecord)1 Schema (co.cask.cdap.api.data.schema.Schema)1 StructuredRecordWritable (co.cask.cdap.etl.batch.StructuredRecordWritable)1 ByteArrayInputStream (java.io.ByteArrayInputStream)1 ByteArrayOutputStream (java.io.ByteArrayOutputStream)1 DataInput (java.io.DataInput)1 DataInputStream (java.io.DataInputStream)1 DataOutput (java.io.DataOutput)1 DataOutputStream (java.io.DataOutputStream)1 Test (org.junit.Test)1