Github user fhueske commented on a diff in the pull request:

    https://github.com/apache/flink/pull/2612#discussion_r83290431
  
    --- Diff: 
flink-batch-connectors/flink-avro/src/test/java/org/apache/flink/api/java/io/AvroOutputFormatTest.java
 ---
    @@ -0,0 +1,126 @@
    +package org.apache.flink.api.java.io;
    +
    +import static org.apache.flink.api.java.io.AvroOutputFormat.Codec;
    +import static org.junit.Assert.assertTrue;
    +import static org.junit.Assert.fail;
    +
    +import java.io.File;
    +import java.io.FileInputStream;
    +import java.io.FileOutputStream;
    +import java.io.IOException;
    +import java.io.ObjectInputStream;
    +import java.io.ObjectOutputStream;
    +import java.nio.file.Files;
    +import java.nio.file.Paths;
    +
    +import org.apache.flink.api.io.avro.example.User;
    +import org.apache.flink.configuration.Configuration;
    +import org.apache.flink.core.fs.FileSystem;
    +import org.apache.flink.core.fs.Path;
    +import org.junit.Test;
    +
    +/**
    + * Tests for {@link AvroOutputFormat}
    + */
    +public class AvroOutputFormatTest {
    +
    +    @Test
    +    public void testSetCodec() throws Exception {
    +        // given
    +        final AvroOutputFormat<User> outputFormat = new 
AvroOutputFormat<>(User.class);
    +
    +        // when
    +        try {
    +            outputFormat.setCodec(Codec.SNAPPY);
    +        } catch (Exception ex) {
    +            // then
    +            fail("unexpected exception");
    +        }
    +    }
    +
    +    @Test
    +    public void testSetCodecError() throws Exception {
    +        // given
    +        boolean error = false;
    +        final AvroOutputFormat<User> outputFormat = new 
AvroOutputFormat<>(User.class);
    +
    +        // when
    +        try {
    +            outputFormat.setCodec(null);
    +        } catch (Exception ex) {
    +            error = true;
    +        }
    +
    +        // then
    +        assertTrue(error);
    +    }
    +
    +    @Test
    +    public void testSerialization() throws Exception {
    +
    +        serializeAndDeserialize(null);
    +        for (final Codec codec : Codec.values()) {
    +            serializeAndDeserialize(codec);
    +        }
    +    }
    +
    +    private void serializeAndDeserialize(final Codec codec) throws 
IOException, ClassNotFoundException {
    +        // given
    +        final AvroOutputFormat<User> outputFormat = new 
AvroOutputFormat<>(User.class);
    +        if (codec != null) {
    +            outputFormat.setCodec(codec);
    +        }
    +        outputFormat.setSchema(User.SCHEMA$);
    +
    +        final File serialized = File.createTempFile("avro-output-format", 
".serialized");
    +
    +        // when
    +        try (final ObjectOutputStream oos = new ObjectOutputStream(new 
FileOutputStream(serialized))) {
    +            oos.writeObject(outputFormat);
    +        }
    +        try (final ObjectInputStream ois = new ObjectInputStream(new 
FileInputStream(serialized))) {
    +            // then
    +            assertTrue(ois.readObject() instanceof AvroOutputFormat);
    --- End diff --
    
    Please check that the relevant properties (schema + codec) of the 
deserialized object are correctly restored.


---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastruct...@apache.org or file a JIRA ticket
with INFRA.
---

Reply via email to