| /* |
| * 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.filter; |
| |
| import org.apache.logging.log4j.Level; |
| import org.apache.logging.log4j.ThreadContext; |
| import org.apache.logging.log4j.core.Filter; |
| import org.apache.logging.log4j.core.LogEvent; |
| import org.apache.logging.log4j.core.config.Configuration; |
| import org.apache.logging.log4j.core.impl.Log4jLogEvent; |
| import org.apache.logging.log4j.core.util.KeyValuePair; |
| import org.apache.logging.log4j.core.junit.LoggerContextSource; |
| import org.apache.logging.log4j.test.junit.UsingThreadContextMap; |
| import org.apache.logging.log4j.message.SimpleMessage; |
| import org.junit.jupiter.api.Test; |
| |
| import java.util.Map; |
| |
| import static org.junit.jupiter.api.Assertions.*; |
| |
| @UsingThreadContextMap |
| public class DynamicThresholdFilterTest { |
| |
| @Test |
| public void testFilter() { |
| ThreadContext.put("userid", "testuser"); |
| ThreadContext.put("organization", "apache"); |
| final KeyValuePair[] pairs = new KeyValuePair[] { |
| new KeyValuePair("testuser", "DEBUG"), |
| new KeyValuePair("JohnDoe", "warn") }; |
| final DynamicThresholdFilter filter = DynamicThresholdFilter.createFilter("userid", pairs, Level.ERROR, null, |
| null); |
| filter.start(); |
| assertTrue(filter.isStarted()); |
| assertSame(Filter.Result.NEUTRAL, filter.filter(null, Level.DEBUG, null, (Object) null, (Throwable) null)); |
| assertSame(Filter.Result.NEUTRAL, filter.filter(null, Level.ERROR, null, (Object) null, (Throwable) null)); |
| ThreadContext.clearMap(); |
| ThreadContext.put("userid", "JohnDoe"); |
| ThreadContext.put("organization", "apache"); |
| LogEvent event = Log4jLogEvent.newBuilder().setLevel(Level.DEBUG).setMessage(new SimpleMessage("Test")).build(); |
| assertSame(Filter.Result.DENY, filter.filter(event)); |
| event = Log4jLogEvent.newBuilder().setLevel(Level.ERROR).setMessage(new SimpleMessage("Test")).build(); |
| assertSame(Filter.Result.NEUTRAL, filter.filter(event)); |
| ThreadContext.clearMap(); |
| } |
| |
| @Test |
| public void testFilterWorksWhenParamsArePassedAsArguments() { |
| ThreadContext.put("userid", "testuser"); |
| ThreadContext.put("organization", "apache"); |
| final KeyValuePair[] pairs = new KeyValuePair[] { |
| new KeyValuePair("testuser", "DEBUG"), |
| new KeyValuePair("JohnDoe", "warn") }; |
| final DynamicThresholdFilter filter = DynamicThresholdFilter.createFilter("userid", pairs, Level.ERROR, Filter.Result.ACCEPT, Filter.Result.NEUTRAL); |
| filter.start(); |
| assertTrue(filter.isStarted()); |
| final Object [] replacements = {"one", "two", "three"}; |
| assertSame(Filter.Result.ACCEPT, filter.filter(null, Level.DEBUG, null, "some test message", replacements)); |
| assertSame(Filter.Result.ACCEPT, filter.filter(null, Level.DEBUG, null, "some test message", "one", "two", "three")); |
| ThreadContext.clearMap(); |
| } |
| |
| @Test |
| @LoggerContextSource("log4j2-dynamicfilter.xml") |
| public void testConfig(final Configuration config) { |
| final Filter filter = config.getFilter(); |
| assertNotNull(filter, "No DynamicThresholdFilter"); |
| assertTrue(filter instanceof DynamicThresholdFilter, "Not a DynamicThresholdFilter"); |
| final DynamicThresholdFilter dynamic = (DynamicThresholdFilter) filter; |
| final String key = dynamic.getKey(); |
| assertNotNull(key, "Key is null"); |
| assertEquals("loginId", key, "Incorrect key value"); |
| final Map<String, Level> map = dynamic.getLevelMap(); |
| assertNotNull(map, "Map is null"); |
| assertEquals(1, map.size(), "Incorrect number of map elements"); |
| } |
| } |