1674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen/* 2674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * Copyright 2003 The Apache Software Foundation 3674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * 4674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * Licensed under the Apache License, Version 2.0 (the "License"); 5674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * you may not use this file except in compliance with the License. 6674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * You may obtain a copy of the License at 7674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * 8674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * http://www.apache.org/licenses/LICENSE-2.0 9674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * 10674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * Unless required by applicable law or agreed to in writing, software 11674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * distributed under the License is distributed on an "AS IS" BASIS, 12674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * See the License for the specific language governing permissions and 14674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * limitations under the License. 15674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen */ 16674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenpackage org.mockito.cglib.transform.impl; 17674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 18674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenimport org.mockito.asm.Attribute; 19674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenimport org.mockito.cglib.core.*; 20674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenimport org.mockito.cglib.transform.*; 21674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 22674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen/** 23674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * A {@link GeneratorStrategy} suitable for use with {@link org.mockito.cglib.Enhancer} which 24674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * causes all undeclared exceptions thrown from within a proxied method to be wrapped 25674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * in an alternative exception of your choice. 26674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen */ 27674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenpublic class UndeclaredThrowableStrategy extends DefaultGeneratorStrategy { 28674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen private ClassTransformer t; 29674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 30674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen /** 31674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * Create a new instance of this strategy. 32674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * @param wrapper a class which extends either directly or 33674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * indirectly from <code>Throwable</code> and which has at least one 34674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * constructor that takes a single argument of type 35674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * <code>Throwable</code>, for example 36674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * <code>java.lang.reflect.UndeclaredThrowableException.class</code> 37674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen */ 38674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen public UndeclaredThrowableStrategy(Class wrapper) { 39674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen t = new UndeclaredThrowableTransformer(wrapper); 40674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen t = new MethodFilterTransformer(TRANSFORM_FILTER, t); 41674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 42674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 43674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen private static final MethodFilter TRANSFORM_FILTER = new MethodFilter() { 44674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen public boolean accept(int access, String name, String desc, String signature, String[] exceptions) { 45674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen return !TypeUtils.isPrivate(access) && name.indexOf('$') < 0; 46674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 47674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen }; 48674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 49674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen protected ClassGenerator transform(ClassGenerator cg) throws Exception { 50674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen return new TransformingClassGenerator(cg, t); 51674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 52674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen} 53674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 54