Apache
/
hadoop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

HADOOP-13470. GenericTestUtils$LogCapturer is flaky. (Contributed by Mingliang Liu)

This commit is contained in:
Mingliang Liu 2016-08-16 16:30:43 -07:00
parent ac779f2554
commit 00eb790622
2 changed files with 63 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/test/GenericTestUtils.java
View File

@ -43,10 +43,12 @@
import org.apache.hadoop.util.NativeCodeLoader; import org.apache.hadoop.util.NativeCodeLoader;
import org.apache.hadoop.util.StringUtils; import org.apache.hadoop.util.StringUtils;
import org.apache.hadoop.util.Time; import org.apache.hadoop.util.Time;
import org.apache.log4j.Appender;
import org.apache.log4j.Layout; import org.apache.log4j.Layout;
import org.apache.log4j.Level; import org.apache.log4j.Level;
import org.apache.log4j.LogManager; import org.apache.log4j.LogManager;
import org.apache.log4j.Logger; import org.apache.log4j.Logger;
import org.apache.log4j.PatternLayout;
import org.apache.log4j.WriterAppender; import org.apache.log4j.WriterAppender;
import org.junit.Assert; import org.junit.Assert;
import org.junit.Assume; import org.junit.Assume;
@ -197,36 +199,41 @@ public static class LogCapturer {
private StringWriter sw = new StringWriter(); private StringWriter sw = new StringWriter();
private WriterAppender appender; private WriterAppender appender;
private Logger logger; private Logger logger;
public static LogCapturer captureLogs(Log l) { public static LogCapturer captureLogs(Log l) {
Logger logger = ((Log4JLogger)l).getLogger(); Logger logger = ((Log4JLogger)l).getLogger();
LogCapturer c = new LogCapturer(logger); return new LogCapturer(logger);
return c; }
public static LogCapturer captureLogs(org.slf4j.Logger logger) {
return new LogCapturer(toLog4j(logger));
} }
private LogCapturer(Logger logger) { private LogCapturer(Logger logger) {
this.logger = logger; this.logger = logger;
Layout layout = Logger.getRootLogger().getAppender("stdout").getLayout(); Appender defaultAppender = Logger.getRootLogger().getAppender("stdout");
WriterAppender wa = new WriterAppender(layout, sw); if (defaultAppender == null) {
logger.addAppender(wa); defaultAppender = Logger.getRootLogger().getAppender("console");
}
final Layout layout = (defaultAppender == null) ? new PatternLayout() :
defaultAppender.getLayout();
this.appender = new WriterAppender(layout, sw);
logger.addAppender(this.appender);
} }
public String getOutput() { public String getOutput() {
return sw.toString(); return sw.toString();
} }
public void stopCapturing() { public void stopCapturing() {
logger.removeAppender(appender); logger.removeAppender(appender);
} }
public void clearOutput() { public void clearOutput() {
sw.getBuffer().setLength(0); sw.getBuffer().setLength(0);
} }
} }
/** /**
* Mockito answer helper that triggers one latch as soon as the * Mockito answer helper that triggers one latch as soon as the
* method is called, then waits on another before continuing. * method is called, then waits on another before continuing.

44
hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/test/TestGenericTestUtils.java
View File

@ -18,8 +18,16 @@
package org.apache.hadoop.test; package org.apache.hadoop.test;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Test; import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static org.junit.Assert.assertTrue;
public class TestGenericTestUtils extends GenericTestUtils { public class TestGenericTestUtils extends GenericTestUtils {
@Test @Test
@ -75,4 +83,40 @@ public String toString() {
} }
} }
@Test(timeout = 10000)
public void testLogCapturer() {
final Log log = LogFactory.getLog(TestGenericTestUtils.class);
LogCapturer logCapturer = LogCapturer.captureLogs(log);
final String infoMessage = "info message";
// test get output message
log.info(infoMessage);
assertTrue(logCapturer.getOutput().endsWith(
String.format(infoMessage + "%n")));
// test clear output
logCapturer.clearOutput();
assertTrue(logCapturer.getOutput().isEmpty());
// test stop capturing
logCapturer.stopCapturing();
log.info(infoMessage);
assertTrue(logCapturer.getOutput().isEmpty());
}
@Test(timeout = 10000)
public void testLogCapturerSlf4jLogger() {
final Logger logger = LoggerFactory.getLogger(TestGenericTestUtils.class);
LogCapturer logCapturer = LogCapturer.captureLogs(logger);
final String infoMessage = "info message";
// test get output message
logger.info(infoMessage);
assertTrue(logCapturer.getOutput().endsWith(
String.format(infoMessage + "%n")));
// test clear output
logCapturer.clearOutput();
assertTrue(logCapturer.getOutput().isEmpty());
// test stop capturing
logCapturer.stopCapturing();
logger.info(infoMessage);
assertTrue(logCapturer.getOutput().isEmpty());
}
} }