org.apache.hadoop.hbase.http.log.TestLogLevel.java Source code

Introduction

Here is the source code for org.apache.hadoop.hbase.http.log.TestLogLevel.java

Source

/**
* 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.hbase.http.log;

import static org.junit.Assert.assertTrue;

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

import org.apache.hadoop.hbase.testclassification.MiscTests;
import org.apache.hadoop.hbase.testclassification.SmallTests;
import org.apache.hadoop.hbase.http.HttpServer;
import org.apache.hadoop.net.NetUtils;
import org.apache.commons.logging.*;
import org.apache.commons.logging.impl.*;
import org.apache.log4j.*;
import org.junit.Test;
import org.junit.experimental.categories.Category;

@Category({ MiscTests.class, SmallTests.class })
public class TestLogLevel {
    static final PrintStream out = System.out;

    @Test(timeout = 60000)
    @SuppressWarnings("deprecation")
    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");
            assertTrue(!Level.ERROR.equals(log.getEffectiveLevel()));

            HttpServer server = null;
            try {
                server = new HttpServer.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);
                HttpURLConnection connection = (HttpURLConnection) url.openConnection();
                connection.connect();

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

                log.debug("log.debug2");
                log.info("log.info2");
                log.error("log.error2");
                assertTrue(Level.ERROR.equals(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");
                assertTrue(Level.DEBUG.equals(log.getEffectiveLevel()));
            } finally {
                if (server != null) {
                    server.stop();
                }
            }
        } else {
            out.println(testlog.getClass() + " not tested.");
        }
    }
}