1 /*

     2  * Copyright (c) 2004, 2007, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    20  * or visit www.oracle.com if you need additional information or have any

    21  * 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 }