LineReaderUtilTest

/*
 * #%L
 * IsisFish
 *
 * $Id: LineReaderUtilTest.java 4156 2014-12-09 11:27:18Z echatellier $
 * $HeadURL: https://svn.codelutin.com/isis-fish/tags/isis-fish-4.4.0.2/src/test/java/fr/ifremer/isisfish/logging/io/LineReaderUtilTest.java $
 * %%
 * Copyright (C) 2006 - 2010 Ifremer, Code Lutin
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

package fr.ifremer.isisfish.logging.io;

import java.io.File;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import fr.ifremer.isisfish.AbstractIsisFishTest;
import fr.ifremer.isisfish.logging.LogLevel;
import fr.ifremer.isisfish.logging.LogLevelUtil;
import fr.ifremer.isisfish.logging.LoggerHelper;

/**
 * Tests unitaires sur les implantations de LineReader
 *
 * @author chemit
 */

public class LineReaderUtilTest extends AbstractIsisFishTest {

    protected static final Log log = LogFactory.getLog(LineReaderUtilTest.class);

    protected static File testDir;

    public static File getTestDir() {
        if (testDir == null) {
            testDir = new File(getTestDirectory(), "bench_" + System.nanoTime());
            if (!testDir.mkdirs()) {
                throw new IllegalStateException("could not create directory " + testDir);
            }
        }
        return testDir;
    }

    final static int[] sizes = new int[]{
            10,
            50,
            100,
            1000,
            5000,
            //10000,
            //50000,
            //100000,
            //200000,
            //300000,
            //500000,
            //1000000,
            //2000000
    };
    File f;

    @Before
    public void setUp() throws Exception {
        System.out.println("benchdir " + getTestDir());
        String filename = "LineReaderTest_30.log";
        f = new File(testDir, filename);

        if (!f.exists()) {
            LoggerHelper.createMock(testDir, filename, 5);
        }
    }

    @Test
    public void testLevelReader() throws Exception {

        long nbLines;
        LineReader reader, reader2;

        reader = new LineReader(f, new FileOffsetReader(new File(f.getParentFile(), "offsets")));
        reader.getOffsetReader().deleteOffsetFile();
        reader.open();
        nbLines = reader.getNbLines();
        Assert.assertEquals(30, nbLines);
        log.info("found " + nbLines + " in reader " + reader);

        reader2 = new LineReaderUtil.LevelLineReader(reader, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + LogLevel.DEBUG.mask())), LogLevel.FATAL);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(5, nbLines);
        reader2.close();

        reader2 = new LineReaderUtil.LevelLineReader(reader, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + LogLevel.DEBUG.mask())), LogLevel.DEBUG);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(5, nbLines);
        reader2.close();

        reader2 = new LineReaderUtil.LevelLineReader(reader, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + LogLevel.INFO.mask())), LogLevel.INFO);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(5, nbLines);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.close();

        reader.getOffsetReader().deleteOffsetFile();
        reader.close();
    }

    @Test
    public void testLevelsReader() throws Exception {

        LineReader reader, reader2, reader3;
        int levels;
        long nbLines;

        reader = new LineReader(f, new FileOffsetReader(new File(f.getParentFile(), "offsets")));
        reader.getOffsetReader().deleteOffsetFile();
        reader.open();
        nbLines = reader.getNbLines();
        Assert.assertEquals(30, nbLines);
        log.info("found " + nbLines + " in reader " + reader);

        levels = LogLevelUtil.createSet(LogLevel.DEBUG, LogLevel.INFO);
        reader2 = new LineReaderUtil.LevelsLineReader(reader, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + levels)), LogLevel.DEBUG, LogLevel.INFO);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(10, nbLines);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.close();

        levels = LogLevelUtil.createSet(LogLevel.WARN, LogLevel.INFO);
        reader2 = new LineReaderUtil.LevelsLineReader(reader, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + levels)), LogLevel.INFO, LogLevel.WARN);
        reader2.getOffsetReader().deleteOffsetFile();
        Assert.assertTrue(reader2.getOffsetReader().needCreate());
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(10, nbLines);

        levels = LogLevelUtil.createSet(LogLevel.INFO);
        reader3 = new LineReaderUtil.LevelsLineReader(reader2, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + levels)), LogLevel.INFO);
        reader3.getOffsetReader().deleteOffsetFile();
        Assert.assertTrue(reader3.getOffsetReader().needCreate());
        reader3.open();
        nbLines = reader3.getNbLines();
        log.info("found " + nbLines + " in reader " + reader3);
        Assert.assertEquals(5, nbLines);

        reader2.close();
        reader3.close();

        levels = LogLevelUtil.createSet(LogLevel.INFO, LogLevel.WARN);
        reader2 = new LineReaderUtil.LevelsLineReader(reader, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + levels)), LogLevel.WARN, LogLevel.INFO);
        Assert.assertFalse(reader2.getOffsetReader().needCreate());
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(10, nbLines);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.close();

        levels = LogLevelUtil.createSet(LogLevel.DEBUG);
        reader2 = new LineReaderUtil.LevelsLineReader(reader, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + levels)), LogLevel.DEBUG);
        Assert.assertFalse(reader2.getOffsetReader().needCreate());
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(5, nbLines);
        reader2.close();

        levels = LogLevelUtil.createSet(LogLevel.DEBUG, LogLevel.DEBUG);
        reader2 = new LineReaderUtil.LevelsLineReader(reader, new FileOffsetReader(new File(reader.getFile().getParentFile(), "leveloffsets_" + levels)), LogLevel.DEBUG, LogLevel.DEBUG);
        Assert.assertFalse(reader2.getOffsetReader().needCreate());
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(5, nbLines);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.close();

        reader.getOffsetReader().deleteOffsetFile();

        reader.close();
    }

    @Test
    public void testPatternReader() throws Exception {

        LineReader reader, reader2;
        long nbLines;

        reader = new LineReader(f, new FileOffsetReader(new File(f.getParentFile(), "offsets")));
        reader.getOffsetReader().deleteOffsetFile();
        reader.open();
        nbLines = reader.getNbLines();
        Assert.assertEquals(30, nbLines);
        log.info("found " + nbLines + " in reader " + reader);

        reader2 = new LineReaderUtil.PatternLineReader(reader, new MemoryOffsetReader(100), "DEBUG.*", 0);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(5, nbLines);
        reader2.close();

        reader2 = new LineReaderUtil.PatternLineReader(reader, new MemoryOffsetReader(100), "ADEBUG.*", 0);
        reader2.getOffsetReader().deleteOffsetFile();
        reader2.open();
        nbLines = reader2.getNbLines();
        log.info("found " + nbLines + " in reader " + reader2);
        Assert.assertEquals(0, nbLines);
        reader2.close();

        reader.getOffsetReader().deleteOffsetFile();
        reader.close();
    }

}