Package org.springframework.boot.logging

Source Code of org.springframework.boot.logging.LoggingApplicationListenerTests

/*
* Copyright 2012-2014 the original author or authors.
*
* Licensed 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.springframework.boot.logging;

import java.io.File;
import java.io.IOException;
import java.util.logging.LogManager;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.commons.logging.impl.SLF4JLogFactory;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.context.event.ApplicationStartedEvent;
import org.springframework.boot.logging.java.JavaLoggingSystem;
import org.springframework.boot.test.EnvironmentTestUtils;
import org.springframework.boot.test.OutputCapture;
import org.springframework.context.support.GenericApplicationContext;

import static org.hamcrest.Matchers.containsString;
import static org.hamcrest.Matchers.not;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;

/**
* Tests for {@link LoggingApplicationListener}.
*
* @author Dave Syer
* @author Phillip Webb
*/
public class LoggingApplicationListenerTests {

  private static final String[] NO_ARGS = {};

  @Rule
  public TemporaryFolder temporaryFolder = new TemporaryFolder();

  @Rule
  public OutputCapture outputCapture = new OutputCapture();

  private final LoggingApplicationListener initializer = new LoggingApplicationListener();

  private final Log logger = new SLF4JLogFactory().getInstance(getClass());

  private final SpringApplication springApplication = new SpringApplication();

  private final GenericApplicationContext context = new GenericApplicationContext();

  @Before
  public void init() throws SecurityException, IOException {
    LogManager.getLogManager().readConfiguration(
        JavaLoggingSystem.class.getResourceAsStream("logging.properties"));
    this.initializer.onApplicationEvent(new ApplicationStartedEvent(
        new SpringApplication(), NO_ARGS));
    new File("target/foo.log").delete();
    new File(tmpDir() + "/spring.log").delete();
  }

  @After
  public void clear() {
    System.clearProperty("LOG_FILE");
    System.clearProperty("LOG_PATH");
    System.clearProperty("PID");
    if (this.context != null) {
      this.context.close();
    }
  }

  private String tmpDir() {
    String path = this.context.getEnvironment().resolvePlaceholders(
        "${java.io.tmpdir}");
    path = path.replace("\\", "/");
    if (path.endsWith("/")) {
      path = path.substring(0, path.length() - 1);
    }
    return path;
  }

  @Test
  public void baseConfigLocation() {
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.info("Hello world");
    String output = this.outputCapture.toString().trim();
    assertTrue("Wrong output:\n" + output, output.contains("Hello world"));
    assertFalse("Wrong output:\n" + output, output.contains("???"));
    assertFalse(new File(tmpDir() + "/spring.log").exists());
  }

  @Test
  public void overrideConfigLocation() {
    EnvironmentTestUtils.addEnvironment(this.context,
        "logging.config: classpath:logback-nondefault.xml");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.info("Hello world");
    String output = this.outputCapture.toString().trim();
    assertTrue("Wrong output:\n" + output, output.contains("Hello world"));
    assertFalse("Wrong output:\n" + output, output.contains("???"));
    assertTrue("Wrong output:\n" + output, output.startsWith("LOG_FILE_IS_UNDEFINED"));
    assertTrue("Wrong output:\n" + output, output.endsWith("BOOTBOOT"));
  }

  @Test
  public void overrideConfigDoesNotExist() throws Exception {
    EnvironmentTestUtils.addEnvironment(this.context,
        "logging.config: doesnotexist.xml");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    // Should not throw
    this.logger.info("Hello world");
    String output = this.outputCapture.toString().trim();
    assertTrue("Wrong output:\n" + output, output.contains("Hello world"));
    assertFalse("Wrong output:\n" + output, output.contains("???"));
    assertFalse(new File(tmpDir() + "/spring.log").exists());
  }

  @Test
  public void addLogFileProperty() {
    EnvironmentTestUtils.addEnvironment(this.context,
        "logging.config: classpath:logback-nondefault.xml",
        "logging.file: target/foo.log");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    Log logger = LogFactory.getLog(LoggingApplicationListenerTests.class);
    logger.info("Hello world");
    String output = this.outputCapture.toString().trim();
    assertTrue("Wrong output:\n" + output, output.startsWith("target/foo.log"));
  }

  @Test
  public void addLogFilePropertyWithDefault() {
    assertFalse(new File("target/foo.log").exists());
    EnvironmentTestUtils.addEnvironment(this.context, "logging.file: target/foo.log");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    Log logger = LogFactory.getLog(LoggingApplicationListenerTests.class);
    logger.info("Hello world");
    assertTrue(new File("target/foo.log").exists());
  }

  @Test
  public void addLogPathProperty() {
    EnvironmentTestUtils.addEnvironment(this.context,
        "logging.config: classpath:logback-nondefault.xml",
        "logging.path: target/foo/");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    Log logger = LogFactory.getLog(LoggingApplicationListenerTests.class);
    logger.info("Hello world");
    String output = this.outputCapture.toString().trim();
    assertTrue("Wrong output:\n" + output, output.startsWith("target/foo/spring.log"));
  }

  @Test
  public void parseDebugArg() throws Exception {
    EnvironmentTestUtils.addEnvironment(this.context, "debug");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.debug("testatdebug");
    this.logger.trace("testattrace");
    assertThat(this.outputCapture.toString(), containsString("testatdebug"));
    assertThat(this.outputCapture.toString(), not(containsString("testattrace")));
  }

  @Test
  public void parseTraceArg() throws Exception {
    EnvironmentTestUtils.addEnvironment(this.context, "trace");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.debug("testatdebug");
    this.logger.trace("testattrace");
    assertThat(this.outputCapture.toString(), containsString("testatdebug"));
    assertThat(this.outputCapture.toString(), containsString("testattrace"));
  }

  @Test
  public void parseLevels() throws Exception {
    EnvironmentTestUtils.addEnvironment(this.context,
        "logging.level.org.springframework.boot=TRACE");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.debug("testatdebug");
    this.logger.trace("testattrace");
    assertThat(this.outputCapture.toString(), containsString("testatdebug"));
    assertThat(this.outputCapture.toString(), containsString("testattrace"));
  }

  @Test
  public void parseLevelsWithPlaceholder() throws Exception {
    EnvironmentTestUtils.addEnvironment(this.context, "foo=TRACE",
        "logging.level.org.springframework.boot=${foo}");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.debug("testatdebug");
    this.logger.trace("testattrace");
    assertThat(this.outputCapture.toString(), containsString("testatdebug"));
    assertThat(this.outputCapture.toString(), containsString("testattrace"));
  }

  @Test
  public void parseLevelsFails() throws Exception {
    EnvironmentTestUtils.addEnvironment(this.context,
        "logging.level.org.springframework.boot=GARBAGE");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.debug("testatdebug");
    assertThat(this.outputCapture.toString(), not(containsString("testatdebug")));
    assertThat(this.outputCapture.toString(),
        containsString("Cannot set level: GARBAGE"));
  }

  @Test
  public void parseLevelsNone() throws Exception {
    EnvironmentTestUtils.addEnvironment(this.context,
        "logging.level.org.springframework.boot=OFF");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.debug("testatdebug");
    this.logger.fatal("testatfatal");
    assertThat(this.outputCapture.toString(), not(containsString("testatdebug")));
    assertThat(this.outputCapture.toString(), not(containsString("testatfatal")));
  }

  @Test
  public void parseArgsDisabled() throws Exception {
    this.initializer.setParseArgs(false);
    EnvironmentTestUtils.addEnvironment(this.context, "debug");
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.debug("testatdebug");
    assertThat(this.outputCapture.toString(), not(containsString("testatdebug")));
  }

  @Test
  public void parseArgsDoesntReplace() throws Exception {
    this.initializer.setSpringBootLogging(LogLevel.ERROR);
    this.initializer.setParseArgs(false);
    this.initializer.onApplicationEvent(new ApplicationStartedEvent(
        this.springApplication, new String[] { "--debug" }));
    this.initializer.initialize(this.context.getEnvironment(),
        this.context.getClassLoader());
    this.logger.debug("testatdebug");
    assertThat(this.outputCapture.toString(), not(containsString("testatdebug")));
  }
}
TOP

Related Classes of org.springframework.boot.logging.LoggingApplicationListenerTests

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.