hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/log/TestLogLevel.java

/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements.  See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership.  The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License.  You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.hadoop.log;

import java.io.*;
import java.net.*;

import org.apache.hadoop.http.HttpServer2;
import org.apache.hadoop.net.NetUtils;

import junit.framework.TestCase;

import org.apache.commons.logging.*;
import org.apache.commons.logging.impl.*;
import org.apache.log4j.*;
import org.junit.Assert;

public class TestLogLevel extends TestCase {
  static final PrintStream out = System.out;

  public void testDynamicLogLevel() throws Exception {
    String logName = TestLogLevel.class.getName();
    Log testlog = LogFactory.getLog(logName);

    //only test Log4JLogger
    if (testlog instanceof Log4JLogger) {
      Logger log = ((Log4JLogger)testlog).getLogger();
      log.debug("log.debug1");
      log.info("log.info1");
      log.error("log.error1");
      Assert.assertNotEquals("Get default Log Level which shouldn't be ERROR.",
          Level.ERROR, log.getEffectiveLevel());

      HttpServer2 server = new HttpServer2.Builder().setName("..")
          .addEndpoint(new URI("http://localhost:0")).setFindPort(true)
          .build();

      server.start();
      String authority = NetUtils.getHostPortString(server
          .getConnectorAddress(0));

      //servlet
      URL url = new URL("http://" + authority + "/logLevel?log=" + logName
          + "&level=" + Level.ERROR);
      out.println("*** Connecting to " + url);
      URLConnection connection = url.openConnection();
      connection.connect();

      BufferedReader in = new BufferedReader(new InputStreamReader(
          connection.getInputStream()));
      for(String line; (line = in.readLine()) != null; out.println(line));
      in.close();

      log.debug("log.debug2");
      log.info("log.info2");
      log.error("log.error2");
      assertEquals("Try setting log level: ERROR from servlet.", Level.ERROR,
          log.getEffectiveLevel());

      //command line
      String[] args = {"-setlevel", authority, logName, Level.DEBUG.toString()};
      LogLevel.main(args);
      log.debug("log.debug3");
      log.info("log.info3");
      log.error("log.error3");
      assertEquals("Try setting log level: DEBUG via command line", Level.DEBUG,
          log.getEffectiveLevel());

      // Test mixed upper case and lower case in level string.
      String[] args2 = {"-setlevel", authority, logName, "Info"};
      LogLevel.main(args2);
      log.debug("log.debug4");
      log.info("log.info4");
      log.error("log.error4");
      assertEquals("Try setting log level: Info via command line.", Level.INFO,
          log.getEffectiveLevel());

      // Test "Error" instead of "ERROR" should work for servlet
      URL newUrl = new URL("http://" + authority + "/logLevel?log=" + logName
          + "&level=" + "Error");
      out.println("*** Connecting to " + newUrl);
      connection = newUrl.openConnection();
      connection.connect();

      BufferedReader in2 = new BufferedReader(new InputStreamReader(
          connection.getInputStream()));
      for(String line; (line = in2.readLine()) != null; out.println(line));
      in2.close();

      log.debug("log.debug5");
      log.info("log.info5");
      log.error("log.error5");
      assertEquals("Try setting log level: Error via servlet.", Level.ERROR,
          log.getEffectiveLevel());
    }
    else {
      out.println(testlog.getClass() + " not tested.");
    }
  }
}