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 }