1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.bcel;
19
20 import org.apache.bcel.classfile.DescendingVisitor;
21 import org.apache.bcel.classfile.JavaClass;
22 import org.apache.bcel.visitors.CountingVisitor;
23 import org.junit.jupiter.api.BeforeEach;
24
25 public abstract class AbstractCounterVisitorTestCase extends AbstractTestCase {
26 private CountingVisitor visitor;
27
28 protected abstract JavaClass getTestClass() throws ClassNotFoundException;
29
30 public CountingVisitor getVisitor() {
31 if (visitor == null) {
32 visitor = new CountingVisitor();
33 }
34 return visitor;
35 }
36
37 @BeforeEach
38 public void setUp() throws ClassNotFoundException {
39 visitor = new CountingVisitor();
40 new DescendingVisitor(getTestClass(), getVisitor()).visit();
41 }
42
43 public void setVisitor(final CountingVisitor visitor) {
44 this.visitor = visitor;
45 }
46 }