1 /*
2  * Copyright 2003 The Apache Software Foundation
3  *
4  *  Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  *  Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.mockito.cglib.transform;
17 
18 import org.mockito.asm.*;
19 
20 public class MethodFilterTransformer extends AbstractClassTransformer {
21     private MethodFilter filter;
22     private ClassTransformer pass;
23     private ClassVisitor direct;
24 
MethodFilterTransformer(MethodFilter filter, ClassTransformer pass)25     public MethodFilterTransformer(MethodFilter filter, ClassTransformer pass) {
26         this.filter = filter;
27         this.pass = pass;
28         super.setTarget(pass);
29     }
30 
visitMethod(int access, String name, String desc, String signature, String[] exceptions)31     public MethodVisitor visitMethod(int access,
32                                      String name,
33                                      String desc,
34                                      String signature,
35                                      String[] exceptions) {
36         return (filter.accept(access, name, desc, signature, exceptions) ? pass : direct).visitMethod(access, name, desc, signature, exceptions);
37     }
38 
setTarget(ClassVisitor target)39     public void setTarget(ClassVisitor target) {
40         pass.setTarget(target);
41         direct = target;
42     }
43 }
44