aboutsummaryrefslogtreecommitdiffstats
path: root/bcel-builder/testsrc/org/aspectj/apache/bcel/classfile/tests/BcelTestCase.java
blob: 87c6dc21e5a9669769e245ae6e33ee7c080bb64a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
/* *******************************************************************
 * Copyright (c) 2004 IBM
 * All rights reserved. 
 * This program and the accompanying materials are made available 
 * under the terms of the Common Public License v1.0 
 * which accompanies this distribution and is available at 
 * http://www.eclipse.org/legal/cpl-v10.html 
 *  
 * Contributors: 
 *     Andy Clement -     initial implementation {date}
 * ******************************************************************/

package org.aspectj.apache.bcel.classfile.tests;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import org.aspectj.apache.bcel.classfile.Attribute;
import org.aspectj.apache.bcel.classfile.JavaClass;
import org.aspectj.apache.bcel.classfile.Method;
import org.aspectj.apache.bcel.classfile.annotation.Annotation;
import org.aspectj.apache.bcel.generic.ConstantPoolGen;
import org.aspectj.apache.bcel.generic.ObjectType;
import org.aspectj.apache.bcel.generic.annotation.AnnotationGen;
import org.aspectj.apache.bcel.generic.annotation.ElementNameValuePairGen;
import org.aspectj.apache.bcel.generic.annotation.ElementValueGen;
import org.aspectj.apache.bcel.generic.annotation.SimpleElementValueGen;
import org.aspectj.apache.bcel.util.ClassPath;
import org.aspectj.apache.bcel.util.SyntheticRepository;

import junit.framework.TestCase;

/**
 * Super class for the Java5 tests, includes various helper methods.
 */

public class BcelTestCase extends TestCase {

	private boolean verbose = false;

	protected File createTestdataFile(String name) {
		return new File("testdata"+File.separator+name);
	}
	
	protected JavaClass getClassFromJar(String clazzname) throws ClassNotFoundException {
		SyntheticRepository repos = createRepos("testcode.jar");
		return repos.loadClass(clazzname);
	}
	
	protected Method getMethod(JavaClass cl,String methodname) {
		Method[] methods = cl.getMethods();
		for (int i = 0; i < methods.length; i++) {
			Method m = methods[i];
			if (m.getName().equals(methodname)) {
				return m;
			}
		}
		return null;
	}
	
	protected boolean wipe(String name) {
		return new File("testdata"+File.separator+name).delete();
	}
	
	protected boolean wipe(String dir, String name) {
		boolean b = wipe(dir+File.separator+name);
		String[] files = new File(dir).list();
		if (files==null || files.length==0) {
			new File(dir).delete(); // Why does this not succeed? stupid thing
		}
		return b;
	}
	

	public SyntheticRepository createRepos(String cpentry) {
		ClassPath cp = new ClassPath(
				"testdata"+File.separator+cpentry+File.pathSeparator+
				System.getProperty("java.class.path"));
		return SyntheticRepository.getInstance(cp);
	}	
	
	protected Attribute[] findAttribute(String name, JavaClass clazz) {
		Attribute[] all = clazz.getAttributes();
		List chosenAttrsList = new ArrayList();
		for (int i = 0; i < all.length; i++) {
			if (verbose) System.err.println("Attribute: "+all[i].getName());
			if (all[i].getName().equals(name)) chosenAttrsList.add(all[i]);
		}
		return (Attribute[])chosenAttrsList.toArray(new Attribute[]{});
	}
	
	protected Attribute findAttribute(String name, Attribute[] all) {
		List chosenAttrsList = new ArrayList();
		for (int i = 0; i < all.length; i++) {
			if (verbose) System.err.println("Attribute: "+all[i].getName());
			if (all[i].getName().equals(name)) chosenAttrsList.add(all[i]);
		}
		assertTrue("Should be one match: "+chosenAttrsList.size(),chosenAttrsList.size()==1);
		return (Attribute)chosenAttrsList.get(0);
	}

	protected String dumpAnnotations(Annotation[] as) {
		StringBuffer result = new StringBuffer();
		result.append("[");
		for (int i = 0; i < as.length; i++) {
			Annotation annotation = as[i];
			result.append(annotation.toShortString());
			if (i+1<as.length) result.append(",");
		}
		result.append("]");
		return result.toString();
	}
	
	protected String dumpAnnotations(AnnotationGen[] as) {
		StringBuffer result = new StringBuffer();
		result.append("[");
		for (int i = 0; i < as.length; i++) {
			AnnotationGen annotation = as[i];
			result.append(annotation.toShortString());
			if (i+1<as.length) result.append(",");
		}
		result.append("]");
		return result.toString();
	}
	
	protected String dumpAttributes(Attribute[] as) {
		StringBuffer result = new StringBuffer();
		result.append("AttributeArray:[");
		for (int i = 0; i < as.length; i++) {
			Attribute attr = as[i];
			result.append(attr.toString());
			if (i+1<as.length) result.append(",");
		}
		result.append("]");
		return result.toString();
	}

	public AnnotationGen createFruitAnnotation(ConstantPoolGen cp, String aFruit, boolean visibility) {
		SimpleElementValueGen evg = new SimpleElementValueGen(ElementValueGen.STRING,cp,aFruit);
		ElementNameValuePairGen nvGen = new ElementNameValuePairGen("fruit",evg,cp);
		ObjectType t = new ObjectType("SimpleStringAnnotation");
		List elements = new ArrayList();
		elements.add(nvGen);
		return new AnnotationGen(t,elements,visibility,cp);
	}
	
	

}