test/tools/apt/lib/TestProcessorFactory.java

changeset 1
9a66ca7c79fa
child 554
9d9f26857129
equal deleted inserted replaced
-1:000000000000 1:9a66ca7c79fa
1 /*
2 * Copyright 2004-2007 Sun Microsystems, Inc. All Rights Reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
20 * CA 95054 USA or visit www.sun.com if you need additional information or
21 * have any questions.
22 */
23
24
25 import java.util.*;
26 import com.sun.mirror.apt.*;
27 import com.sun.mirror.declaration.*;
28 import com.sun.mirror.util.*;
29
30
31 /**
32 * A factory for generating the TestProcessor annotation processor, which
33 * processes the @Test annotation.
34 *
35 * @author Scott Seligman
36 */
37 public class TestProcessorFactory implements AnnotationProcessorFactory {
38
39 public Collection<String> supportedOptions() {
40 return new ArrayList<String>();
41 }
42
43 public Collection<String> supportedAnnotationTypes() {
44 ArrayList<String> res = new ArrayList<String>();
45 res.add("Test");
46 res.add("Ignore");
47 return res;
48 }
49
50 public AnnotationProcessor getProcessorFor(
51 Set<AnnotationTypeDeclaration> as,
52 AnnotationProcessorEnvironment env) {
53 // The tester that's running.
54 Tester tester = Tester.activeTester;
55
56 try {
57 // Find the tester's class declaration.
58 ClassDeclaration testerDecl = null;
59 for (TypeDeclaration decl : env.getSpecifiedTypeDeclarations()) {
60 if (decl.getQualifiedName().equals(
61 tester.getClass().getName())) {
62 testerDecl = (ClassDeclaration) decl;
63 break;
64 }
65 }
66
67 // Give the tester access to its own declaration and to the env.
68 tester.thisClassDecl = testerDecl;
69 tester.env = env;
70
71 // Initializer the tester.
72 tester.init();
73
74 return new TestProcessor(env, tester);
75
76 } catch (Exception e) {
77 throw new Error("Couldn't create test annotation processor", e);
78 }
79 }
80 }

mercurial