BufferedLFISTest.java

package org.araqne.logstorage.file;

import static org.junit.Assert.assertArrayEquals;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Arrays;
import java.util.Random;

import org.araqne.storage.localfile.LocalFilePath;
import org.junit.Test;

public class BufferedLFISTest {
	@Test
	public void readSizeTest() throws IOException {
		File file = new File("braf_test.dat");
		try {
			BufferedOutputStream s = new BufferedOutputStream(new FileOutputStream(file));
			Random rand = new Random(1);
			final int BUFSIZE = 16384;
			byte[] buf = new byte[BUFSIZE];
			for (int i = 0; i < 100; ++i) {
				rand.nextBytes(buf);
				s.write(buf);
			}
			s.close();

			byte[] diffsrc = new byte[BUFSIZE * 100];
			new Random(1).nextBytes(diffsrc);

			BufferedStorageInputStream stream = new BufferedStorageInputStream(new LocalFilePath(file));
			assertArrayEquals(subarray(diffsrc, 0, 1024), readLFIS(stream, 0, 1024));
			assertArrayEquals(subarray(diffsrc, 0, 8192 * 4), readLFIS(stream, 0, 8192 * 4));
			assertArrayEquals(subarray(diffsrc, 8192 * 15, 8192 * 4), readLFIS(stream, 8192 * 15, 8192 * 4));

		} finally {
			file.delete();
		}

	}

	private byte[] subarray(byte[] diffsrc, int offset, int length) {
		return Arrays.copyOfRange(diffsrc, offset, offset + length);
	}

	private byte[] readLFIS(BufferedStorageInputStream stream, int seekpos, int length) throws IOException {
		byte[] ret = new byte[length];
		stream.seek(seekpos);
		stream.readFully(ret);
		return ret;
	}
}