|
13 | 13 | */ |
14 | 14 | package ch.qos.logback.classic; |
15 | 15 |
|
| 16 | +import ch.qos.logback.classic.spi.ILoggingEvent; |
| 17 | +import ch.qos.logback.classic.turbo.MDCFilter; |
16 | 18 | import ch.qos.logback.classic.turbo.MarkerFilter; |
17 | 19 | import ch.qos.logback.classic.turbo.TurboFilter; |
| 20 | +import ch.qos.logback.core.read.ListAppender; |
18 | 21 | import ch.qos.logback.core.spi.FilterReply; |
| 22 | +import ch.qos.logback.core.testUtil.RandomUtil; |
19 | 23 | import org.junit.jupiter.api.BeforeEach; |
20 | 24 | import org.junit.jupiter.api.Test; |
| 25 | +import org.slf4j.MDC; |
21 | 26 | import org.slf4j.Marker; |
22 | 27 | import org.slf4j.MarkerFactory; |
23 | 28 |
|
|
29 | 34 | public class TurboFilteringInLoggerTest { |
30 | 35 |
|
31 | 36 | static final String BLUE = "BLUE"; |
32 | | - LoggerContext context; |
| 37 | + LoggerContext loggerContext; |
33 | 38 | Logger logger; |
34 | 39 | Marker blueMarker = MarkerFactory.getMarker(BLUE); |
35 | 40 |
|
| 41 | + int diff = RandomUtil.getPositiveInt(); |
| 42 | + String key = "tfiolKey" + diff; |
| 43 | + String value = "val" + diff; |
| 44 | + |
| 45 | + ListAppender<ILoggingEvent> listAppender = new ListAppender<>(); |
| 46 | + |
| 47 | + MDCFilter mdcFilter = new MDCFilter(); |
36 | 48 | @BeforeEach |
37 | 49 | public void setUp() throws Exception { |
38 | | - context = new LoggerContext(); |
39 | | - context.setName("test"); |
40 | | - context.start(); |
41 | | - logger = context.getLogger(TurboFilteringInLoggerTest.class); |
| 50 | + loggerContext = new LoggerContext(); |
| 51 | + loggerContext.setName("test"); |
| 52 | + loggerContext.start(); |
| 53 | + logger = loggerContext.getLogger(TurboFilteringInLoggerTest.class); |
| 54 | + |
| 55 | + Logger root = loggerContext.getLogger(Logger.ROOT_LOGGER_NAME); |
| 56 | + root.setLevel(Level.ERROR); |
| 57 | + listAppender.start(); |
| 58 | + root.addAppender(listAppender); |
| 59 | + |
42 | 60 | } |
43 | 61 |
|
| 62 | + private void addMDCFilter() { |
| 63 | + |
| 64 | + mdcFilter.setOnMatch("ACCEPT"); |
| 65 | + mdcFilter.setOnMismatch("DENY"); |
| 66 | + mdcFilter.setMDCKey(key); |
| 67 | + mdcFilter.setValue(value); |
| 68 | + mdcFilter.start(); |
| 69 | + loggerContext.addTurboFilter(mdcFilter); |
| 70 | + } |
44 | 71 | private void addYesFilter() { |
45 | 72 | YesFilter filter = new YesFilter(); |
46 | 73 | filter.start(); |
47 | | - context.addTurboFilter(filter); |
| 74 | + loggerContext.addTurboFilter(filter); |
48 | 75 | } |
49 | 76 |
|
50 | 77 | private void addNoFilter() { |
51 | 78 | NoFilter filter = new NoFilter(); |
52 | 79 | filter.start(); |
53 | | - context.addTurboFilter(filter); |
| 80 | + loggerContext.addTurboFilter(filter); |
54 | 81 | } |
55 | 82 |
|
56 | 83 | private void addAcceptBLUEFilter() { |
57 | 84 | MarkerFilter filter = new MarkerFilter(); |
58 | 85 | filter.setMarker(BLUE); |
59 | 86 | filter.setOnMatch("ACCEPT"); |
60 | 87 | filter.start(); |
61 | | - context.addTurboFilter(filter); |
| 88 | + loggerContext.addTurboFilter(filter); |
62 | 89 | } |
63 | 90 |
|
64 | 91 | private void addDenyBLUEFilter() { |
65 | 92 | MarkerFilter filter = new MarkerFilter(); |
66 | 93 | filter.setMarker(BLUE); |
67 | 94 | filter.setOnMatch("DENY"); |
68 | 95 | filter.start(); |
69 | | - context.addTurboFilter(filter); |
| 96 | + loggerContext.addTurboFilter(filter); |
70 | 97 | } |
71 | 98 |
|
72 | 99 | @Test |
@@ -156,11 +183,21 @@ public void testIsErrorEnabledWithDenyBlueFilter() { |
156 | 183 | @Test |
157 | 184 | public void testLoggingContextReset() { |
158 | 185 | addYesFilter(); |
159 | | - assertNotNull(context.getTurboFilterList().get(0)); |
160 | | - context.reset(); |
161 | | - assertEquals(0, context.getTurboFilterList().size()); |
| 186 | + assertNotNull(loggerContext.getTurboFilterList().get(0)); |
| 187 | + loggerContext.reset(); |
| 188 | + assertEquals(0, loggerContext.getTurboFilterList().size()); |
162 | 189 | } |
163 | 190 |
|
| 191 | + @Test |
| 192 | + public void fluentAPI() { |
| 193 | + addMDCFilter(); |
| 194 | + Logger logger = loggerContext.getLogger(this.getClass()); |
| 195 | + logger.atDebug().log("hello 1"); |
| 196 | + assertEquals(0, listAppender.list.size()); |
| 197 | + MDC.put(key, value); |
| 198 | + logger.atDebug().log("hello 2"); |
| 199 | + assertEquals(1, listAppender.list.size()); |
| 200 | + } |
164 | 201 | } |
165 | 202 |
|
166 | 203 | class YesFilter extends TurboFilter { |
|
0 commit comments