01: /***
02: * ASM Guide
03: * Copyright (c) 2007 Eric Bruneton
04: * All rights reserved.
05: *
06: * Redistribution and use in source and binary forms, with or without
07: * modification, are permitted provided that the following conditions
08: * are met:
09: * 1. Redistributions of source code must retain the above copyright
10: * notice, this list of conditions and the following disclaimer.
11: * 2. Redistributions in binary form must reproduce the above copyright
12: * notice, this list of conditions and the following disclaimer in the
13: * documentation and/or other materials provided with the distribution.
14: * 3. Neither the name of the copyright holders nor the names of its
15: * contributors may be used to endorse or promote products derived from
16: * this software without specific prior written permission.
17: *
18: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19: * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21: * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
22: * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23: * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24: * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25: * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27: * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
28: * THE POSSIBILITY OF SUCH DAMAGE.
29: */package ch2.sec2;
30:
31: import java.lang.reflect.Method;
32:
33: import org.objectweb.asm.ClassVisitor;
34: import org.objectweb.asm.ClassWriter;
35: import org.objectweb.asm.util.CheckClassAdapter;
36:
37: import util.AbstractTestCase;
38:
39: /**
40: * ASM Guide example test class.
41: *
42: * @author Eric Bruneton
43: */
44: public class AddMethodAdapterTest extends AbstractTestCase {
45:
46: public void test() throws Exception {
47: ClassWriter cw = new ClassWriter(0);
48: CheckClassAdapter ca = new CheckClassAdapter(cw);
49: AddMethodAdapter am = new AddMethodAdapter(ca);
50: generateBasicClass(am);
51: checkClass(defineClass("C", cw.toByteArray()));
52: }
53:
54: protected void checkClass(Class c) throws Exception {
55: Object o = c.newInstance();
56: Method m = c.getMethod("getThis");
57: assertEquals(o, m.invoke(o));
58: }
59:
60: @Override
61: protected ClassVisitor getClassAdapter(ClassVisitor cv) {
62: return new AddMethodAdapter(cv);
63: }
64: }
|