SpecificJsonSerializerUnitTest.java

package com.ctriposs.baiji.specific;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class SpecificJsonSerializerUnitTest {

    private SpecificJsonWriter<TestSerializerSample> writer;
    private SpecificJsonReader<TestSerializerSample> reader;

    @Before
    public void setUp() throws Exception {
        writer = new SpecificJsonWriter<>();
        reader = new SpecificJsonReader<>(TestSerializerSample.SCHEMA);
    }

    @Test
    public void testSerializeBoolean() throws Exception {
        singleFieldTest("boolean1", true);
    }

    @Test
    public void testSerializeInt() throws Exception {
        singleFieldTest("int1", 42);
    }

    @Test
    public void testSerializeLong() throws Exception {
        singleFieldTest("bigint1", 1024 * 1024 * 16L);
    }

    @Test
    public void testSerializeDouble() throws Exception {
        singleFieldTest("double1", 24.00000001);
    }

    @Test
    public void testSerializeString() throws Exception {
        singleFieldTest("string1", "好好学习");
    }

    @Test
    public void testSerializeBytes() throws Exception {
        TestSerializerSample sample = serializeAndDeserialize("bytes1", "天天向上".getBytes());
        Assert.assertArrayEquals(sample.getBytes1(), "天天向上".getBytes());
    }

    @Test
    public void testSerializeEnum() throws Exception {
        singleFieldTest("enum1", Enum1Values.RED);
    }

    @Test
    public void testSerializeArray() throws Exception {
        singleFieldTest("list1", Arrays.asList("a", "b", "c"));
    }

    @Test
    public void testSerializeMap() throws Exception {
        Map<String, Integer> map = new HashMap<String, Integer>();
        map.put("1a", 1);
        map.put("2b", 2);
        map.put("3c", 3);
        singleFieldTest("map1", map);
    }

    @Test
    public void testSerializeNullable() throws Exception {
        singleFieldTest("nullableint", null);
        singleFieldTest("nullableint", 1);
    }

    private void singleFieldTest(String fieldName, Object fieldValue) throws IOException {
        TestSerializerSample sample = serializeAndDeserialize(fieldName, fieldValue);
        Assert.assertEquals(fieldValue, sample.get(fieldName));
    }

    private TestSerializerSample serializeAndDeserialize(String fieldName, Object fieldValue) throws IOException {
        TestSerializerSample record = new TestSerializerSample();
        record.put(fieldName, fieldValue);

        // First serialize
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        writer.write(record.getSchema(), record, os);

        // Convert the output-stream to input-stream
        ByteArrayInputStream is = new ByteArrayInputStream(os.toByteArray());

        // Then deserialize
        return reader.read(null, is);
        //return record;
    }
}