org.fcrepo.utilities.io.TestByteRangeInputStream.java Source code

Introduction

Here is the source code for org.fcrepo.utilities.io.TestByteRangeInputStream.java

Source

/* The contents of this file are subject to the license and copyright terms
 * detailed in the license directory at the root of the source tree (also
 * available online at http://fedora-commons.org/license/).
 */
package org.fcrepo.utilities.io;

import static org.junit.Assert.assertEquals;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.IOException;
import java.io.SequenceInputStream;
import java.nio.charset.Charset;

import junit.framework.JUnit4TestAdapter;

import org.junit.Test;
import org.apache.commons.io.IOUtils;
import org.fcrepo.utilities.io.NullInputStream;

/**
 * @author armintor@gmail.com
 */
public class TestByteRangeInputStream {

    @Test
    public void testGoodRangeHeaders() throws IOException {
        // these are semantically equivalent for a 10 byte stream
        String[] inputs = new String[] { "bytes=0-9", "bytes=0", "bytes=-10", "bytes=-12", "bytes=0-" };
        for (String input : inputs) {
            ByteRangeInputStream test = new ByteRangeInputStream(NullInputStream.NULL_STREAM, 10, input);
            try {
                assertEquals("bad offset of " + test.offset + " for " + input, 0, test.offset);
                assertEquals("bad length of " + test.length + " for " + input, 10, test.length);
                assertEquals("bytes 0-9/10", test.contentRange);
            } finally {
                test.close();
            }
        }
        inputs = new String[] { "bytes=1-9", "bytes=1", "bytes=-9", "bytes=1-12", "bytes=1-" };
        for (String input : inputs) {
            ByteRangeInputStream test = new ByteRangeInputStream(NullInputStream.NULL_STREAM, 10, input);
            try {
                assertEquals("bad offset of " + test.offset + " for " + input, 1, test.offset);
                assertEquals("bad length of " + test.length + " for " + input, 9, test.length);
                assertEquals("bytes 1-9/10", test.contentRange);
            } finally {
                test.close();
            }
        }
        inputs = new String[] { "bytes= 1-9 ", "bytes= 1", "bytes= - 9", "bytes = 1-12", "bytes= 1 - " };
        for (String input : inputs) {
            ByteRangeInputStream test = new ByteRangeInputStream(NullInputStream.NULL_STREAM, 10, input);
            try {
                assertEquals("bad offset of " + test.offset + " for " + input, 1, test.offset);
                assertEquals("bad length of " + test.length + " for " + input, 9, test.length);
                assertEquals("bytes 1-9/10", test.contentRange);
            } finally {
                test.close();
            }
        }
        String input = "bytes=2-6";
        ByteRangeInputStream test = new ByteRangeInputStream(NullInputStream.NULL_STREAM, 10, input);
        try {
            assertEquals("bad offset of " + test.offset + " for " + input, 2, test.offset);
            assertEquals("bad length of " + test.length + " for " + input, 5, test.length);
            assertEquals("bytes 2-6/10", test.contentRange);
        } finally {
            test.close();
        }
        input = "bytes=2-2";
        test = new ByteRangeInputStream(NullInputStream.NULL_STREAM, 10, input);
        try {
            assertEquals("bad offset of " + test.offset + " for " + input, 2, test.offset);
            assertEquals("bad length of " + test.length + " for " + input, 1, test.length);
            assertEquals("bytes 2-2/10", test.contentRange);
        } finally {
            test.close();
        }

    }

    @Test(expected = IndexOutOfBoundsException.class)
    public void testBadRangeHeaderOffset() throws IOException {
        @SuppressWarnings({ "unused", "resource" })
        ByteRangeInputStream test = new ByteRangeInputStream(NullInputStream.NULL_STREAM, 10, "bytes=10");
    }

    @Test(expected = IndexOutOfBoundsException.class)
    public void testBadRangeHeaderOrder() throws IOException {
        @SuppressWarnings({ "unused", "resource" })
        ByteRangeInputStream test = new ByteRangeInputStream(NullInputStream.NULL_STREAM, 10, "bytes=6-2");
    }

    @SuppressWarnings("resource")
    @Test
    public void testSkippedBytes() throws IndexOutOfBoundsException, IOException {
        String data = "1234567890";
        String input = "bytes=3-12";
        InputStream bytes = new ByteArrayInputStream(data.getBytes(Charset.forName("UTF-8")));
        ByteRangeInputStream test = new ByteRangeInputStream(bytes, 10, input);
        assertEquals("bad offset of " + test.offset + " for " + input, 3, test.offset);
        assertEquals("bad length of " + test.length + " for " + input, 7, test.length);
        assertEquals("bytes 3-9/10", test.contentRange);
        bytes.reset();
        test = new ByteRangeInputStream(bytes, 10, "bytes=0-8");
        assertEquals("123456789", IOUtils.toString(test));
        bytes.reset();
        InputStream bytes2 = new ByteArrayInputStream(data.getBytes(Charset.forName("UTF-8")));
        test = new ByteRangeInputStream(bytes, 10, "bytes=0-2");
        ByteRangeInputStream test2 = new ByteRangeInputStream(bytes2, 10, "bytes=-7");
        SequenceInputStream test3 = new SequenceInputStream(test, test2);
        String actual = IOUtils.toString(test3);
        assertEquals(data, actual);
    }

    public static junit.framework.Test suite() {
        return new JUnit4TestAdapter(TestByteRangeInputStream.class);
    }

}