log4j-core/src/test/java/org/apache/logging/log4j/core/appender/rolling/RollingAppenderOnStartupTest.java - logging-log4j2 - Git at Google

 /*
  * 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.logging.log4j.core.appender.rolling;

 import java.io.File;
 import java.nio.file.DirectoryStream;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.nio.file.StandardCopyOption;
 import java.nio.file.attribute.BasicFileAttributeView;
 import java.nio.file.attribute.FileTime;
 import java.time.Instant;
 import java.time.temporal.ChronoUnit;
 import java.util.Arrays;
 import java.util.Collection;
 import java.util.List;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;

 import org.apache.logging.log4j.Logger;
 import org.apache.logging.log4j.junit.LoggerContextRule;
 import org.junit.AfterClass;
 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.Parameterized;

 import static org.junit.Assert.assertTrue;

 /**
  *
  */
 public class RollingAppenderOnStartupTest {

     private static final String SOURCE = "src/test/resources/__files";
     private static final String DIR = "target/onStartup";
     private static final String CONFIG = "log4j-test4.xml";
     private static final String FILENAME = "onStartup.log";

     private Logger logger;

     @Rule
     public LoggerContextRule loggerContextRule;

     public RollingAppenderOnStartupTest() {
         this.loggerContextRule = LoggerContextRule.createShutdownTimeoutLoggerContextRule(CONFIG);
     }

     @Before
     public void setUp() throws Exception {
         this.logger = this.loggerContextRule.getLogger(RollingAppenderOnStartupTest.class.getName());
     }

     @BeforeClass
     public static void beforeClass() throws Exception {
         if (Files.exists(Paths.get("target/onStartup"))) {
             try (DirectoryStream<Path> directoryStream = Files.newDirectoryStream(Paths.get(DIR))) {
                 for (final Path path : directoryStream) {
                     Files.delete(path);
                 }
                 Files.delete(Paths.get(DIR));
             }
         }
         Files.createDirectory(new File(DIR).toPath());
         Path target = Paths.get(DIR, FILENAME);
         Files.copy(Paths.get(SOURCE, FILENAME), target, StandardCopyOption.COPY_ATTRIBUTES);
         FileTime newTime = FileTime.from(Instant.now().minus(1, ChronoUnit.DAYS));
         Files.getFileAttributeView(target, BasicFileAttributeView.class).setTimes(newTime, newTime, newTime);
     }

     @AfterClass
     public static void afterClass() throws Exception {
         try (DirectoryStream<Path> directoryStream = Files.newDirectoryStream(Paths.get(DIR))) {
             boolean rolled = false;
             for (final Path path : directoryStream) {
                 if (!path.toFile().getName().endsWith(FILENAME)) {
                     rolled = true;
                 }
                 try (Stream<String> stream = Files.lines(path)) {
                     List<String> lines = stream.collect(Collectors.toList());
                     assertTrue("No header present for " + path.toFile().getName(), lines.get(0).startsWith("<!DOCTYPE HTML"));
                 }
                 Files.delete(path);
             }
             assertTrue("File did not roll", rolled);
         }
         Files.delete(Paths.get("target/onStartup"));
     }

     @Test
     public void testAppender() throws Exception {
         for (int i = 0; i < 100; ++i) {
             logger.debug("This is test message number " + i);
         }

     }
 }
	/*
	* 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.logging.log4j.core.appender.rolling;

	import java.io.File;
	import java.nio.file.DirectoryStream;
	import java.nio.file.Files;
	import java.nio.file.Path;
	import java.nio.file.Paths;
	import java.nio.file.StandardCopyOption;
	import java.nio.file.attribute.BasicFileAttributeView;
	import java.nio.file.attribute.FileTime;
	import java.time.Instant;
	import java.time.temporal.ChronoUnit;
	import java.util.Arrays;
	import java.util.Collection;
	import java.util.List;
	import java.util.stream.Collectors;
	import java.util.stream.Stream;

	import org.apache.logging.log4j.Logger;
	import org.apache.logging.log4j.junit.LoggerContextRule;
	import org.junit.AfterClass;
	import org.junit.Before;
	import org.junit.BeforeClass;
	import org.junit.Rule;
	import org.junit.Test;
	import org.junit.runner.RunWith;
	import org.junit.runners.Parameterized;

	import static org.junit.Assert.assertTrue;

	/**
	*
	*/
	public class RollingAppenderOnStartupTest {

	private static final String SOURCE = "src/test/resources/__files";
	private static final String DIR = "target/onStartup";
	private static final String CONFIG = "log4j-test4.xml";
	private static final String FILENAME = "onStartup.log";

	private Logger logger;

	@Rule
	public LoggerContextRule loggerContextRule;

	public RollingAppenderOnStartupTest() {
	this.loggerContextRule = LoggerContextRule.createShutdownTimeoutLoggerContextRule(CONFIG);
	}

	@Before
	public void setUp() throws Exception {
	this.logger = this.loggerContextRule.getLogger(RollingAppenderOnStartupTest.class.getName());
	}

	@BeforeClass
	public static void beforeClass() throws Exception {
	if (Files.exists(Paths.get("target/onStartup"))) {
	try (DirectoryStream<Path> directoryStream = Files.newDirectoryStream(Paths.get(DIR))) {
	for (final Path path : directoryStream) {
	Files.delete(path);
	}
	Files.delete(Paths.get(DIR));
	}
	}
	Files.createDirectory(new File(DIR).toPath());
	Path target = Paths.get(DIR, FILENAME);
	Files.copy(Paths.get(SOURCE, FILENAME), target, StandardCopyOption.COPY_ATTRIBUTES);
	FileTime newTime = FileTime.from(Instant.now().minus(1, ChronoUnit.DAYS));
	Files.getFileAttributeView(target, BasicFileAttributeView.class).setTimes(newTime, newTime, newTime);
	}

	@AfterClass
	public static void afterClass() throws Exception {
	try (DirectoryStream<Path> directoryStream = Files.newDirectoryStream(Paths.get(DIR))) {
	boolean rolled = false;
	for (final Path path : directoryStream) {
	if (!path.toFile().getName().endsWith(FILENAME)) {
	rolled = true;
	}
	try (Stream<String> stream = Files.lines(path)) {
	List<String> lines = stream.collect(Collectors.toList());
	assertTrue("No header present for " + path.toFile().getName(), lines.get(0).startsWith("<!DOCTYPE HTML"));
	}
	Files.delete(path);
	}
	assertTrue("File did not roll", rolled);
	}
	Files.delete(Paths.get("target/onStartup"));
	}

	@Test
	public void testAppender() throws Exception {
	for (int i = 0; i < 100; ++i) {
	logger.debug("This is test message number " + i);
	}

	}
	}