| /* |
| * 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.async; |
| |
| import org.apache.logging.log4j.Level; |
| import org.apache.logging.log4j.categories.AsyncLoggers; |
| import org.apache.logging.log4j.util.PropertiesUtil; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.experimental.categories.Category; |
| |
| import static org.junit.Assert.*; |
| |
| /** |
| * Tests the AsyncQueueFullPolicyFactory class. |
| */ |
| @Category(AsyncLoggers.class) |
| public class AsyncQueueFullPolicyFactoryTest { |
| |
| @Before |
| public void setUp() throws Exception { |
| System.clearProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_ASYNC_EVENT_ROUTER); |
| System.clearProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_DISCARDING_THRESHOLD_LEVEL); |
| PropertiesUtil.getProperties().reload(); |
| } |
| |
| @Test |
| public void testCreateReturnsDefaultRouterByDefault() throws Exception { |
| final AsyncQueueFullPolicy router = AsyncQueueFullPolicyFactory.create(); |
| assertEquals(DefaultAsyncQueueFullPolicy.class, router.getClass()); |
| } |
| |
| @Test |
| public void testCreateReturnsDiscardingRouterIfSpecified() throws Exception { |
| System.setProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_ASYNC_EVENT_ROUTER, |
| AsyncQueueFullPolicyFactory.PROPERTY_VALUE_DISCARDING_ASYNC_EVENT_ROUTER); |
| assertEquals(DiscardingAsyncQueueFullPolicy.class, AsyncQueueFullPolicyFactory.create().getClass()); |
| |
| System.setProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_ASYNC_EVENT_ROUTER, |
| DiscardingAsyncQueueFullPolicy.class.getSimpleName()); |
| assertEquals(DiscardingAsyncQueueFullPolicy.class, AsyncQueueFullPolicyFactory.create().getClass()); |
| |
| System.setProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_ASYNC_EVENT_ROUTER, |
| DiscardingAsyncQueueFullPolicy.class.getName()); |
| assertEquals(DiscardingAsyncQueueFullPolicy.class, AsyncQueueFullPolicyFactory.create().getClass()); |
| } |
| |
| @Test |
| public void testCreateDiscardingRouterDefaultThresholdLevelInfo() throws Exception { |
| System.setProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_ASYNC_EVENT_ROUTER, |
| AsyncQueueFullPolicyFactory.PROPERTY_VALUE_DISCARDING_ASYNC_EVENT_ROUTER); |
| assertEquals(Level.INFO, ((DiscardingAsyncQueueFullPolicy) AsyncQueueFullPolicyFactory.create()). |
| getThresholdLevel()); |
| } |
| |
| @Test |
| public void testCreateDiscardingRouterThresholdLevelCustomizable() throws Exception { |
| System.setProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_ASYNC_EVENT_ROUTER, |
| AsyncQueueFullPolicyFactory.PROPERTY_VALUE_DISCARDING_ASYNC_EVENT_ROUTER); |
| |
| for (final Level level : Level.values()) { |
| System.setProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_DISCARDING_THRESHOLD_LEVEL, |
| level.name()); |
| assertEquals(level, ((DiscardingAsyncQueueFullPolicy) AsyncQueueFullPolicyFactory.create()). |
| getThresholdLevel()); |
| } |
| } |
| |
| static class CustomRouterDefaultConstructor implements AsyncQueueFullPolicy { |
| public CustomRouterDefaultConstructor() { |
| } |
| |
| @Override |
| public EventRoute getRoute(final long backgroundThreadId, final Level level) { |
| return null; |
| } |
| } |
| |
| static class DoesNotImplementInterface { |
| } |
| |
| @Test |
| public void testCreateReturnsCustomRouterIfSpecified() throws Exception { |
| System.setProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_ASYNC_EVENT_ROUTER, |
| CustomRouterDefaultConstructor.class.getName()); |
| assertEquals(CustomRouterDefaultConstructor.class, AsyncQueueFullPolicyFactory.create().getClass()); |
| } |
| |
| @Test |
| public void testCreateReturnsDefaultRouterIfSpecifiedCustomRouterFails() throws Exception { |
| System.setProperty(AsyncQueueFullPolicyFactory.PROPERTY_NAME_ASYNC_EVENT_ROUTER, |
| DoesNotImplementInterface.class.getName()); |
| assertEquals(DefaultAsyncQueueFullPolicy.class, AsyncQueueFullPolicyFactory.create().getClass()); |
| } |
| } |