VanillaBytesMarshallerTest.java

/*
 * Copyright 2016 higherfrequencytrading.com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package net.openhft.lang.io.serialization;

import net.openhft.lang.io.DirectBytes;
import net.openhft.lang.io.DirectStore;
import net.openhft.lang.io.NativeBytes;
import net.openhft.lang.io.serialization.impl.VanillaBytesMarshallerFactory;
import org.junit.Test;
import sun.nio.ch.DirectBuffer;

import java.io.EOFException;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.concurrent.atomic.AtomicInteger;

import static org.junit.Assert.*;

/**
 * User: peter.lawrey Date: 20/09/13 Time: 09:28
 */
public class VanillaBytesMarshallerTest {

    @Test
    public void testObjects() {
        DirectBytes bytes = new DirectStore(1024).bytes();
        Object[] objects = {1, 1L, 1.0, "Hello"};
        for (Object o : objects) {
            long pos = bytes.position();
            bytes.writeObject(o);
            System.out.printf("%s used %,d bytes%n", o.getClass(), bytes.position() - pos);
        }
        bytes.clear();
        for (Object o : objects) {
            Object o2 = bytes.readObject();
            assertEquals(o, o2);
        }
    }

    @Test
    public void testMarshallable() {
        int capacity = 2 * 1024;
        ByteBuffer byteBuffer = ByteBuffer.allocateDirect(capacity);
        long addr = ((DirectBuffer) byteBuffer).address();
        NativeBytes nativeBytes = new NativeBytes(addr, addr + capacity);

        nativeBytes.writeObject(BuySell.BUY);
        nativeBytes.writeObject(BuySell.SELL);
        nativeBytes.finish();
        nativeBytes.clear();
        assertEquals(BuySell.BUY, nativeBytes.readObject());
        assertEquals(BuySell.SELL, nativeBytes.readObject());
    }

    @Test
    public void testExceptionWithoutCause() {
        final int capacity = 2 * 1024;
        final ByteBuffer byteBuffer = ByteBuffer.allocateDirect(capacity);
        final long addr = ((DirectBuffer) byteBuffer).address();

        final NativeBytes nativeBytes = new NativeBytes(
                new VanillaBytesMarshallerFactory(), addr, addr + capacity, new AtomicInteger(1));

        Throwable expected = new IOException("io-exception");

        nativeBytes.writeObject(expected);
        nativeBytes.finish();
        nativeBytes.clear();

        Throwable actual = nativeBytes.readObject(Throwable.class);
        assertNotNull(actual);
        assertNull(actual.getCause());

        assertEquals(expected.getMessage(), actual.getMessage());
        assertArrayEquals(expected.getStackTrace(), actual.getStackTrace());
    }

    @Test
    public void testExceptionWithCause() {
        final int capacity = 2 * 1024;
        final ByteBuffer byteBuffer = ByteBuffer.allocateDirect(capacity);
        final long addr = ((DirectBuffer) byteBuffer).address();

        final NativeBytes nativeBytes = new NativeBytes(
                new VanillaBytesMarshallerFactory(), addr, addr + capacity, new AtomicInteger(1));

        Throwable expected = new IOException(
                "io-exception", new EOFException("eof-exception"));

        nativeBytes.writeObject(expected);
        nativeBytes.finish();
        nativeBytes.clear();

        Throwable actual = nativeBytes.readObject(Throwable.class);
        assertNotNull(actual);
        assertNotNull(actual.getCause());

        assertEquals(expected.getMessage(), actual.getMessage());
        assertEquals(expected.getCause().getMessage(), actual.getCause().getMessage());
        assertArrayEquals(expected.getStackTrace(), actual.getStackTrace());
        assertArrayEquals(expected.getCause().getStackTrace(), actual.getCause().getStackTrace());
    }

    enum BuySell {
        BUY, SELL
    }
}