com.amalto.core.servlet.FileChunkLoaderTest.java Source code

Introduction

Here is the source code for com.amalto.core.servlet.FileChunkLoaderTest.java

Source

/*
 * Copyright (C) 2006-2016 Talend Inc. - www.talend.com
 * 
 * This source code is available under agreement available at
 * %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
 * 
 * You should have received a copy of the agreement along with this program; if not, write to Talend SA 9 rue Pages
 * 92150 Suresnes, France
 */
package com.amalto.core.servlet;

import static org.junit.Assert.*;

import java.io.File;
import java.net.URL;

import org.apache.commons.io.output.ByteArrayOutputStream;
import org.junit.Test;

import com.amalto.core.servlet.FileChunkLoader.FileChunkInfo;

@SuppressWarnings("nls")
public class FileChunkLoaderTest {

    @Test
    public void test() throws Exception {
        File test = getFile("com/amalto/core/servlet/test.log");

        FileChunkLoader loader = new FileChunkLoader(test);
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        long position = 0;
        FileChunkInfo chunkInfo = null;

        //
        for (int i = 0; i < 11; i++) {
            chunkInfo = loader.loadChunkTo(baos, position, 2);
            if (position == 0) {
                String result = baos.toString();
                assertTrue(result.endsWith("Service (JTA version) - JBoss Inc.\r\n"));
                assertEquals(238, chunkInfo.nextPosition);
            }
            position = chunkInfo.nextPosition;
        }
        assertEquals(2406, chunkInfo.nextPosition);
        assertEquals(1, chunkInfo.lines); // last line does not contains '\n'
        String result = baos.toString();
        assertTrue(result.endsWith("startup in 34 ms"));

        //
        baos.reset();
        chunkInfo = loader.loadChunkTo(baos, 0, 0);
        assertEquals(0, chunkInfo.nextPosition);
        assertEquals(0, chunkInfo.lines);

        //
        baos.reset();
        chunkInfo = loader.loadChunkTo(baos, 0, 100);
        assertEquals(2406, chunkInfo.nextPosition);
        assertEquals(21, chunkInfo.lines);

        // tail
        baos.reset();
        chunkInfo = loader.loadChunkTo(baos, -1, 10);
        assertEquals(2406, chunkInfo.nextPosition);
        assertEquals(6, chunkInfo.lines);
        assertTrue(result.endsWith("startup in 34 ms"));

        // tail
        baos.reset();
        chunkInfo = loader.loadChunkTo(baos, -1, 100);
        assertEquals(2406, chunkInfo.nextPosition);
        assertEquals(21, chunkInfo.lines);
    }

    private File getFile(String filename) {
        URL url = getClass().getClassLoader().getResource(filename);
        assertNotNull(url);
        return new File(url.getFile());
    }
}