src/test/java/org/apache/bcel/AbstractCounterVisitorTestCase.java


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

package org.apache.bcel;

import org.apache.bcel.classfile.DescendingVisitor;
import org.apache.bcel.classfile.JavaClass;
import org.apache.bcel.visitors.CounterVisitor;

public abstract class AbstractCounterVisitorTestCase extends AbstractTestCase
{
	protected abstract JavaClass getTestClass() throws ClassNotFoundException;

	private CounterVisitor visitor = null;

	public void setUp() throws ClassNotFoundException
	{
		visitor = new CounterVisitor();
		new DescendingVisitor(getTestClass(), getVisitor()).visit();
	}

	public CounterVisitor getVisitor()
	{
		if (visitor == null)
			visitor = new CounterVisitor();
		return visitor;
	}

	public void setVisitor(CounterVisitor visitor)
	{
		this.visitor = visitor;
	}
}