DataInputBufferTest.java

package com.facebook.infrastructure.io;

import org.testng.annotations.Test;

import java.util.Random;
import java.io.IOException;

public class DataInputBufferTest {
    @Test
    public void testRandom() throws IOException {
        Random random = new Random();
        byte[] bytes = new byte[1024*1024];
        random.nextBytes(bytes);

        DataInputBuffer.FastByteArrayInputStream bis = new DataInputBuffer.FastByteArrayInputStream(bytes);
        int read = 0;
        int n = 0;
        while ( true )
        {
            read = bis.read();
            if ( read == -1 )
                break;
            assert read == ((int)bytes[n++]&0xFF);
        }
        assert n == bytes.length;
    }

    @Test
    public void testSmall() throws IOException {
        DataOutputBuffer bufOut = new DataOutputBuffer();
        bufOut.writeUTF("Avinash");
        bufOut.writeInt(41*1024*1024);
        DataInputBuffer bufIn = new DataInputBuffer();
        bufIn.reset(bufOut.getData(), bufOut.getLength());
        assert bufIn.readUTF().equals("Avinash");
        assert bufIn.readInt() == 41 * 1024 * 1024;
    }

}