CompactTestRunner.java

package de.jpaw.bonaparte.testrunner;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;

import org.testng.Assert;

import de.jpaw.bonaparte.core.BonaCustom;
import de.jpaw.bonaparte.core.BonaPortable;
import de.jpaw.bonaparte.core.CompactComposer;
import de.jpaw.bonaparte.core.CompactParser;

/** Compact composer, using the alternate output channel (Stream instead of buffer). */
public class CompactTestRunner extends AbstractTestrunner<byte[]> {

    @Override
    public byte[] serializationTest(BonaCustom src, byte[] expectedResult) throws Exception {
        ByteArrayOutputStream baos = new ByteArrayOutputStream(4000);
        DataOutputStream dataOut = new DataOutputStream(baos);
        CompactComposer bac = new CompactComposer(dataOut, false);
        bac.writeRecord(src);
        dataOut.flush();
        byte [] result = baos.toByteArray();
        if (expectedResult != null)
            Assert.assertEquals(result, expectedResult);
        return result;
    }

    @Override
    public BonaPortable deserializationTest(byte[] src, BonaPortable expectedResult) throws Exception {
        ByteArrayInputStream bais = new ByteArrayInputStream(src);
        DataInputStream dis = new DataInputStream(bais);
        CompactParser cp = new CompactParser(dis);
        BonaPortable result = cp.readRecord();
        if (expectedResult != null)
            Assert.assertEquals(result, expectedResult);
        return result;
    }
}