langtools/test/tools/javac/T6403466.java
author darcy
Mon, 15 Feb 2010 18:20:57 -0800
changeset 4934 c6a2cf3b6b82
parent 10 06bc494ca11e
child 5520 86e4b9a9da40
permissions -rw-r--r--
6634138: Source generated in last round not compiled Reviewed-by: jjg
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
10
06bc494ca11e Initial load
duke
parents:
diff changeset
     1
/*
06bc494ca11e Initial load
duke
parents:
diff changeset
     2
 * Copyright 2006 Sun Microsystems, Inc.  All Rights Reserved.
06bc494ca11e Initial load
duke
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
06bc494ca11e Initial load
duke
parents:
diff changeset
     4
 *
06bc494ca11e Initial load
duke
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
06bc494ca11e Initial load
duke
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
06bc494ca11e Initial load
duke
parents:
diff changeset
     7
 * published by the Free Software Foundation.
06bc494ca11e Initial load
duke
parents:
diff changeset
     8
 *
06bc494ca11e Initial load
duke
parents:
diff changeset
     9
 * This code is distributed in the hope that it will be useful, but WITHOUT
06bc494ca11e Initial load
duke
parents:
diff changeset
    10
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
06bc494ca11e Initial load
duke
parents:
diff changeset
    11
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
06bc494ca11e Initial load
duke
parents:
diff changeset
    12
 * version 2 for more details (a copy is included in the LICENSE file that
06bc494ca11e Initial load
duke
parents:
diff changeset
    13
 * accompanied this code).
06bc494ca11e Initial load
duke
parents:
diff changeset
    14
 *
06bc494ca11e Initial load
duke
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License version
06bc494ca11e Initial load
duke
parents:
diff changeset
    16
 * 2 along with this work; if not, write to the Free Software Foundation,
06bc494ca11e Initial load
duke
parents:
diff changeset
    17
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
06bc494ca11e Initial load
duke
parents:
diff changeset
    18
 *
06bc494ca11e Initial load
duke
parents:
diff changeset
    19
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
06bc494ca11e Initial load
duke
parents:
diff changeset
    20
 * CA 95054 USA or visit www.sun.com if you need additional information or
06bc494ca11e Initial load
duke
parents:
diff changeset
    21
 * have any questions.
06bc494ca11e Initial load
duke
parents:
diff changeset
    22
 */
06bc494ca11e Initial load
duke
parents:
diff changeset
    23
06bc494ca11e Initial load
duke
parents:
diff changeset
    24
/*
06bc494ca11e Initial load
duke
parents:
diff changeset
    25
 * @test
06bc494ca11e Initial load
duke
parents:
diff changeset
    26
 * @bug 6403466
06bc494ca11e Initial load
duke
parents:
diff changeset
    27
 * @summary javac TaskListener should be informed when annotation processing occurs
06bc494ca11e Initial load
duke
parents:
diff changeset
    28
 */
06bc494ca11e Initial load
duke
parents:
diff changeset
    29
06bc494ca11e Initial load
duke
parents:
diff changeset
    30
import com.sun.source.util.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    31
import java.io.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    32
import java.lang.annotation.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    33
import java.util.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    34
import javax.annotation.processing.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    35
import javax.lang.model.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    36
import javax.lang.model.element.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    37
import javax.lang.model.type.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    38
import javax.lang.model.util.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    39
import javax.tools.*;
06bc494ca11e Initial load
duke
parents:
diff changeset
    40
import com.sun.tools.javac.api.JavacTool;
06bc494ca11e Initial load
duke
parents:
diff changeset
    41
06bc494ca11e Initial load
duke
parents:
diff changeset
    42
@Wrap
06bc494ca11e Initial load
duke
parents:
diff changeset
    43
@SupportedAnnotationTypes("Wrap")
06bc494ca11e Initial load
duke
parents:
diff changeset
    44
public class T6403466 extends AbstractProcessor {
06bc494ca11e Initial load
duke
parents:
diff changeset
    45
06bc494ca11e Initial load
duke
parents:
diff changeset
    46
    static final String testSrcDir = System.getProperty("test.src");
06bc494ca11e Initial load
duke
parents:
diff changeset
    47
    static final String testClassDir = System.getProperty("test.classes");
06bc494ca11e Initial load
duke
parents:
diff changeset
    48
    static final String self = T6403466.class.getName();
06bc494ca11e Initial load
duke
parents:
diff changeset
    49
    static PrintWriter out = new PrintWriter(System.err, true);
06bc494ca11e Initial load
duke
parents:
diff changeset
    50
06bc494ca11e Initial load
duke
parents:
diff changeset
    51
    public static void main(String[] args) throws IOException {
06bc494ca11e Initial load
duke
parents:
diff changeset
    52
        JavacTool tool = JavacTool.create();
06bc494ca11e Initial load
duke
parents:
diff changeset
    53
06bc494ca11e Initial load
duke
parents:
diff changeset
    54
        StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
06bc494ca11e Initial load
duke
parents:
diff changeset
    55
        Iterable<? extends JavaFileObject> files =
06bc494ca11e Initial load
duke
parents:
diff changeset
    56
            fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrcDir, self + ".java")));
06bc494ca11e Initial load
duke
parents:
diff changeset
    57
06bc494ca11e Initial load
duke
parents:
diff changeset
    58
        Iterable<String> options = Arrays.asList("-processorpath", testClassDir,
06bc494ca11e Initial load
duke
parents:
diff changeset
    59
                                                 "-processor", self,
06bc494ca11e Initial load
duke
parents:
diff changeset
    60
                                                 "-s", ".",
06bc494ca11e Initial load
duke
parents:
diff changeset
    61
                                                 "-d", ".");
06bc494ca11e Initial load
duke
parents:
diff changeset
    62
        JavacTask task = tool.getTask(out, fm, null, options, null, files);
06bc494ca11e Initial load
duke
parents:
diff changeset
    63
06bc494ca11e Initial load
duke
parents:
diff changeset
    64
        VerifyingTaskListener vtl = new VerifyingTaskListener(new File(testSrcDir, self + ".out"));
06bc494ca11e Initial load
duke
parents:
diff changeset
    65
        task.setTaskListener(vtl);
06bc494ca11e Initial load
duke
parents:
diff changeset
    66
06bc494ca11e Initial load
duke
parents:
diff changeset
    67
        if (!task.call())
06bc494ca11e Initial load
duke
parents:
diff changeset
    68
            throw new AssertionError("compilation failed");
06bc494ca11e Initial load
duke
parents:
diff changeset
    69
06bc494ca11e Initial load
duke
parents:
diff changeset
    70
        if (vtl.iter.hasNext() || vtl.errors)
06bc494ca11e Initial load
duke
parents:
diff changeset
    71
            throw new AssertionError("comparison against golden file failed.");
06bc494ca11e Initial load
duke
parents:
diff changeset
    72
    }
06bc494ca11e Initial load
duke
parents:
diff changeset
    73
06bc494ca11e Initial load
duke
parents:
diff changeset
    74
    public boolean process(Set<? extends TypeElement> annos, RoundEnvironment rEnv) {
4934
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    75
        if (!rEnv.processingOver()) {
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    76
            Filer filer = processingEnv.getFiler();
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    77
            for (TypeElement anno: annos) {
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    78
                Set<? extends Element> elts = rEnv.getElementsAnnotatedWith(anno);
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    79
                System.err.println("anno: " + anno);
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    80
                System.err.println("elts: " + elts);
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    81
                for (TypeElement te: ElementFilter.typesIn(elts)) {
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    82
                    try {
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    83
                        Writer out = filer.createSourceFile(te.getSimpleName() + "Wrapper").openWriter();
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    84
                        out.write("class " + te.getSimpleName() + "Wrapper { }");
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    85
                        out.close();
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    86
                    } catch (IOException ex) {
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    87
                        ex.printStackTrace();
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    88
                    }
10
06bc494ca11e Initial load
duke
parents:
diff changeset
    89
                }
4934
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    90
10
06bc494ca11e Initial load
duke
parents:
diff changeset
    91
            }
06bc494ca11e Initial load
duke
parents:
diff changeset
    92
        }
06bc494ca11e Initial load
duke
parents:
diff changeset
    93
        return true;
06bc494ca11e Initial load
duke
parents:
diff changeset
    94
    }
4934
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    95
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    96
    @Override
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    97
    public SourceVersion getSupportedSourceVersion() {
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    98
        return SourceVersion.latest();
c6a2cf3b6b82 6634138: Source generated in last round not compiled
darcy
parents: 10
diff changeset
    99
    }
10
06bc494ca11e Initial load
duke
parents:
diff changeset
   100
}
06bc494ca11e Initial load
duke
parents:
diff changeset
   101
06bc494ca11e Initial load
duke
parents:
diff changeset
   102
@Retention(RetentionPolicy.SOURCE)
06bc494ca11e Initial load
duke
parents:
diff changeset
   103
@Target(ElementType.TYPE)
06bc494ca11e Initial load
duke
parents:
diff changeset
   104
@interface Wrap {
06bc494ca11e Initial load
duke
parents:
diff changeset
   105
}
06bc494ca11e Initial load
duke
parents:
diff changeset
   106
06bc494ca11e Initial load
duke
parents:
diff changeset
   107
06bc494ca11e Initial load
duke
parents:
diff changeset
   108
class VerifyingTaskListener implements TaskListener {
06bc494ca11e Initial load
duke
parents:
diff changeset
   109
    VerifyingTaskListener(File ref) throws IOException {
06bc494ca11e Initial load
duke
parents:
diff changeset
   110
        BufferedReader in = new BufferedReader(new FileReader(ref));
06bc494ca11e Initial load
duke
parents:
diff changeset
   111
        String line;
06bc494ca11e Initial load
duke
parents:
diff changeset
   112
        List<String> lines = new ArrayList<String>();
06bc494ca11e Initial load
duke
parents:
diff changeset
   113
        while ((line = in.readLine()) != null)
06bc494ca11e Initial load
duke
parents:
diff changeset
   114
            lines.add(line);
06bc494ca11e Initial load
duke
parents:
diff changeset
   115
        in.close();
06bc494ca11e Initial load
duke
parents:
diff changeset
   116
        iter = lines.iterator();
06bc494ca11e Initial load
duke
parents:
diff changeset
   117
    }
06bc494ca11e Initial load
duke
parents:
diff changeset
   118
06bc494ca11e Initial load
duke
parents:
diff changeset
   119
    public void started(TaskEvent e) {
06bc494ca11e Initial load
duke
parents:
diff changeset
   120
        check("Started " + toString(e));
06bc494ca11e Initial load
duke
parents:
diff changeset
   121
    }
06bc494ca11e Initial load
duke
parents:
diff changeset
   122
    public void finished(TaskEvent e) {
06bc494ca11e Initial load
duke
parents:
diff changeset
   123
        check("Finished " + toString(e));
06bc494ca11e Initial load
duke
parents:
diff changeset
   124
    }
06bc494ca11e Initial load
duke
parents:
diff changeset
   125
06bc494ca11e Initial load
duke
parents:
diff changeset
   126
    // similar to TaskEvent.toString(), but just prints basename of the file
06bc494ca11e Initial load
duke
parents:
diff changeset
   127
    private String toString(TaskEvent e) {
06bc494ca11e Initial load
duke
parents:
diff changeset
   128
        JavaFileObject file = e.getSourceFile();
06bc494ca11e Initial load
duke
parents:
diff changeset
   129
        return "TaskEvent["
06bc494ca11e Initial load
duke
parents:
diff changeset
   130
            + e.getKind() + ","
06bc494ca11e Initial load
duke
parents:
diff changeset
   131
            + (file == null ? null : new File(file.toUri().getPath()).getName()) + ","
06bc494ca11e Initial load
duke
parents:
diff changeset
   132
            // the compilation unit is identified by the file
06bc494ca11e Initial load
duke
parents:
diff changeset
   133
            + e.getTypeElement() + "]";
06bc494ca11e Initial load
duke
parents:
diff changeset
   134
    }
06bc494ca11e Initial load
duke
parents:
diff changeset
   135
06bc494ca11e Initial load
duke
parents:
diff changeset
   136
    private void check(String s) {
06bc494ca11e Initial load
duke
parents:
diff changeset
   137
        System.out.println(s); // write a reference copy of expected output to stdout
06bc494ca11e Initial load
duke
parents:
diff changeset
   138
06bc494ca11e Initial load
duke
parents:
diff changeset
   139
        String ref = iter.hasNext() ? iter.next() : null;
06bc494ca11e Initial load
duke
parents:
diff changeset
   140
        line++;
06bc494ca11e Initial load
duke
parents:
diff changeset
   141
        if (!s.equals(ref)) {
06bc494ca11e Initial load
duke
parents:
diff changeset
   142
            if (ref != null)
06bc494ca11e Initial load
duke
parents:
diff changeset
   143
                System.err.println(line + ": expected: " + ref);
06bc494ca11e Initial load
duke
parents:
diff changeset
   144
            System.err.println(line + ":    found: " + s);
06bc494ca11e Initial load
duke
parents:
diff changeset
   145
            errors = true;
06bc494ca11e Initial load
duke
parents:
diff changeset
   146
        }
06bc494ca11e Initial load
duke
parents:
diff changeset
   147
    }
06bc494ca11e Initial load
duke
parents:
diff changeset
   148
06bc494ca11e Initial load
duke
parents:
diff changeset
   149
    Iterator<String> iter;
06bc494ca11e Initial load
duke
parents:
diff changeset
   150
    int line;
06bc494ca11e Initial load
duke
parents:
diff changeset
   151
    boolean errors;
06bc494ca11e Initial load
duke
parents:
diff changeset
   152
}