LogTest.java

/*
 * A CCNx library test.
 *
 * Copyright (C) 2010, 2011 Palo Alto Research Center, Inc.
 *
 * This work is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License version 2 as published by the
 * Free Software Foundation. 
 * This work 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, write to the
 * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */
 
package org.ccnx.ccn.test.impl;

import java.io.ByteArrayOutputStream;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.logging.SimpleFormatter;
import java.util.logging.StreamHandler;

import org.ccnx.ccn.impl.support.Log;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;

public class LogTest {
	protected static ByteArrayOutputStream _baos = new ByteArrayOutputStream();
	protected static StreamHandler _sh = new StreamHandler(_baos, new SimpleFormatter());
	
	@BeforeClass
	public static void setUpBeforeClass() throws Exception {
		String app = Log.getApplicationClass();
		Logger logger = Logger.getLogger(app);
		logger.addHandler(_sh);
	}
	
	/**
	 * Write to the log and return number of bytes that show up
	 * in the log
	 */
	protected int writeLog(int facility, Level level, String msg) {
		int start, finish;
		start = _baos.toByteArray().length;
		Log.log(facility, level, msg);
		_sh.flush();
		finish = _baos.toByteArray().length;
		return finish - start;
	}
	
	protected void doTest(int facility, Level level) {
		
		// set level off and make sure no logging
		Log.setLevel(facility, Level.OFF);
		Assert.assertEquals(0, writeLog(facility, level, "test me off"));

		// set to ALL and make sure it does log
		Log.setLevel(facility, Level.ALL);
		Assert.assertTrue(writeLog(facility, level, "test me all") > 0);
		
		// set to level and make sure it does log
		Log.setLevel(facility, level);
		Assert.assertTrue(writeLog(facility, level, "test me level") > 0);
	}
	
	@Test
	public void testInfoStringObjectArray() {
		doTest(Log.FAC_DEFAULT, Level.INFO);
	}

	@Test
	public void testInfoIntStringObjectArray() {
		doTest(Log.FAC_DEFAULT, Level.INFO);
		doTest(Log.FAC_NETMANAGER, Level.INFO);
	}

	@Test
	public void testWarningIntStringObjectArray() {
		doTest(Log.FAC_DEFAULT, Level.WARNING);
		doTest(Log.FAC_NETMANAGER, Level.WARNING);
	}

	@Test
	public void testSevereIntStringObjectArray() {
		doTest(Log.FAC_DEFAULT, Level.SEVERE);
		doTest(Log.FAC_NETMANAGER, Level.SEVERE);
	}

	@Test
	public void testGetLevelInt() {
		Log.setLevel(Level.OFF);
		Assert.assertEquals(Level.OFF.intValue(), Log.getLevel().intValue());
		Assert.assertEquals(Level.OFF.intValue(), Log.getLevel(Log.FAC_DEFAULT).intValue());
		
		Log.setLevel(Log.FAC_NETMANAGER, Level.SEVERE);
		Assert.assertEquals(Level.SEVERE.intValue(), Log.getLevel(Log.FAC_NETMANAGER).intValue());
	}
	
	@Test
	public void testGetSetLevels() {
		Log.setLevel(Log.FAC_ALL, Level.WARNING);
		Log.setLevel(Log.FAC_NETMANAGER, Level.INFO);
		Log.setLevel(Log.FAC_IO, Level.FINE);
		
		Level [] save = Log.getLevels();
		Log.setLevel(Log.FAC_ALL, Level.FINER);
		Assert.assertEquals(Level.FINER, Log.getLevel(Log.FAC_PIPELINE));
		Assert.assertEquals(Level.FINER, Log.getLevel(Log.FAC_NETMANAGER));
		Assert.assertEquals(Level.FINER, Log.getLevel(Log.FAC_IO));
		
		Log.setLevels(save);
		Assert.assertEquals(Level.WARNING, Log.getLevel(Log.FAC_PIPELINE));
		Assert.assertEquals(Level.INFO, Log.getLevel(Log.FAC_NETMANAGER));
		Assert.assertEquals(Level.FINE, Log.getLevel(Log.FAC_IO) );
	}

}