1 /** 2 * Copyright (c) 2004-2011 QOS.ch 3 * All rights reserved. 4 * 5 * Permission is hereby granted, free of charge, to any person obtaining 6 * a copy of this software and associated documentation files (the 7 * "Software"), to deal in the Software without restriction, including 8 * without limitation the rights to use, copy, modify, merge, publish, 9 * distribute, sublicense, and/or sell copies of the Software, and to 10 * permit persons to whom the Software is furnished to do so, subject to 11 * the following conditions: 12 * 13 * The above copyright notice and this permission notice shall be 14 * included in all copies or substantial portions of the Software. 15 * 16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 17 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 18 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 19 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE 20 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION 21 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION 22 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 23 * 24 */ 25 package org.slf4j.helpers; 26 27 import java.lang.reflect.InvocationHandler; 28 import java.lang.reflect.InvocationTargetException; 29 import java.lang.reflect.Method; 30 import java.lang.reflect.Proxy; 31 import java.util.ArrayList; 32 import java.util.Arrays; 33 import java.util.HashSet; 34 import java.util.List; 35 import java.util.Set; 36 37 import junit.framework.TestCase; 38 import org.slf4j.Logger; 39 40 /** 41 * @author Chetan Mehrotra 42 */ 43 public class SubstitutableLoggerTest extends TestCase { 44 private static final Set<String> EXCLUDED_METHODS = new HashSet<String>(Arrays.asList("getName")); 45 testDelegate()46 public void testDelegate() throws Exception { 47 SubstituteLogger log = new SubstituteLogger("foo"); 48 assertTrue(log.delegate() instanceof NOPLogger); 49 50 Set<String> expectedMethodSignatures = determineMethodSignatures(Logger.class); 51 LoggerInvocationHandler ih = new LoggerInvocationHandler(); 52 Logger proxyLogger = (Logger) Proxy.newProxyInstance(getClass().getClassLoader(), new Class[] { Logger.class }, ih); 53 log.setDelegate(proxyLogger); 54 55 invokeMethods(log); 56 57 // Assert that all methods are delegated 58 expectedMethodSignatures.removeAll(ih.getInvokedMethodSignatures()); 59 if (!expectedMethodSignatures.isEmpty()) { 60 fail("Following methods are not delegated " + expectedMethodSignatures.toString()); 61 } 62 } 63 invokeMethods(Logger proxyLogger)64 private void invokeMethods(Logger proxyLogger) throws InvocationTargetException, IllegalAccessException { 65 for (Method m : Logger.class.getDeclaredMethods()) { 66 if (!EXCLUDED_METHODS.contains(m.getName())) { 67 m.invoke(proxyLogger, new Object[m.getParameterTypes().length]); 68 } 69 } 70 } 71 72 private class LoggerInvocationHandler implements InvocationHandler { 73 private final Set<String> invokedMethodSignatures = new HashSet<String>(); 74 invoke(Object proxy, Method method, Object[] args)75 public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { 76 invokedMethodSignatures.add(getMethodSignature(method)); 77 if (method.getName().startsWith("is")) { 78 return true; 79 } 80 return null; 81 } 82 getInvokedMethodSignatures()83 public Set<String> getInvokedMethodSignatures() { 84 return invokedMethodSignatures; 85 } 86 } 87 determineMethodSignatures(Class<Logger> loggerClass)88 private static Set<String> determineMethodSignatures(Class<Logger> loggerClass) { 89 Set<String> methodSignatures = new HashSet<String>(); 90 for (Method m : loggerClass.getDeclaredMethods()) { 91 if (!EXCLUDED_METHODS.contains(m.getName())) { 92 methodSignatures.add(getMethodSignature(m)); 93 } 94 } 95 return methodSignatures; 96 } 97 getMethodSignature(Method m)98 private static String getMethodSignature(Method m) { 99 List<String> result = new ArrayList<String>(); 100 result.add(m.getName()); 101 for (Class<?> clazz : m.getParameterTypes()) { 102 result.add(clazz.getSimpleName()); 103 } 104 return result.toString(); 105 } 106 } 107