duke@1: /*
ohair@554:  * Copyright (c) 2006, Oracle and/or its affiliates. All rights reserved.
duke@1:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1:  *
duke@1:  * This code is free software; you can redistribute it and/or modify it
duke@1:  * under the terms of the GNU General Public License version 2 only, as
duke@1:  * published by the Free Software Foundation.
duke@1:  *
duke@1:  * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
duke@1:  * version 2 for more details (a copy is included in the LICENSE file that
duke@1:  * accompanied this code).
duke@1:  *
duke@1:  * You should have received a copy of the GNU General Public License version
duke@1:  * 2 along with this work; if not, write to the Free Software Foundation,
duke@1:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1:  *
ohair@554:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554:  * or visit www.oracle.com if you need additional information or have any
ohair@554:  * questions.
duke@1:  */
duke@1: 
duke@1: /*
duke@1:  * @test
duke@1:  * @bug 6439826 6411930 6380018 6392177
duke@1:  * @summary Exception issuing Diagnostic while processing generated errant code
duke@1:  */
duke@1: 
duke@1: import java.io.*;
duke@1: import java.util.*;
duke@1: import javax.annotation.processing.*;
duke@1: import javax.lang.model.*;
duke@1: import javax.lang.model.element.*;
duke@1: import javax.tools.*;
duke@1: import com.sun.source.util.*;
duke@1: import com.sun.tools.javac.api.*;
duke@1: import static javax.lang.model.util.ElementFilter.*;
duke@1: 
duke@1: 
duke@1: @SupportedAnnotationTypes("*")
duke@1: public class T6439826 extends AbstractProcessor {
duke@1:     public static void main(String... args) {
duke@1:         String testSrc = System.getProperty("test.src", ".");
duke@1:         String testClasses = System.getProperty("test.classes");
duke@1:         JavacTool tool = JavacTool.create();
duke@1:         MyDiagListener dl = new MyDiagListener();
duke@1:         StandardJavaFileManager fm = tool.getStandardFileManager(dl, null, null);
duke@1:         Iterable<? extends JavaFileObject> files =
duke@1:             fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, T6439826.class.getName()+".java")));
jjg@308:         Iterable<String> opts = Arrays.asList("-source","1.6",
jjg@308:                                               "-proc:only",
duke@1:                                               "-processor", "T6439826",
duke@1:                                               "-processorpath", testClasses);
duke@1:         StringWriter out = new StringWriter();
duke@1:         JavacTask task = tool.getTask(out, fm, dl, opts, null, files);
duke@1:         task.call();
duke@1:         String s = out.toString();
duke@1:         System.err.print(s);
duke@1:         // Expect the following 2 diagnostics, and no output to log
duke@1:         //   Foo.java:1: illegal character: \35
duke@1:         //   Foo.java:1: reached end of file while parsing
duke@1:         System.err.println(dl.count + " diagnostics; " + s.length() + " characters");
duke@1:         if (dl.count != 2 || s.length() != 0)
duke@1:             throw new AssertionError("unexpected output from compiler");
duke@1:     }
duke@1: 
duke@1:     public boolean process(Set<? extends TypeElement> annotations,
duke@1:                            RoundEnvironment roundEnv) {
duke@1:         Set<? extends TypeElement> elems = typesIn(roundEnv.getRootElements());
duke@1:         for (TypeElement e: elems) {
duke@1:             if (e.getSimpleName().toString().equals(T6439826.class.getName()))
duke@1:                 writeBadFile();
duke@1:         }
duke@1:         return false;
duke@1:     }
duke@1: 
darcy@495:     @Override
darcy@495:     public SourceVersion getSupportedSourceVersion() {
darcy@495:         return SourceVersion.latest();
darcy@495:     }
darcy@495: 
duke@1:     private void writeBadFile() {
duke@1:         Filer filer = processingEnv.getFiler();
duke@1:         Messager messager = processingEnv.getMessager();
duke@1:         try {
duke@1:             Writer out = filer.createSourceFile("Foo").openWriter();
duke@1:             out.write("class Foo #"); // write a file that generates a scanner error
duke@1:             out.close();
duke@1:         } catch (IOException e) {
duke@1:             messager.printMessage(Diagnostic.Kind.ERROR, e.toString());
duke@1:         }
duke@1:     }
duke@1: 
duke@1:     static class MyDiagListener implements DiagnosticListener {
duke@1:         public void report(Diagnostic d) {
duke@1:             System.err.println(d);
duke@1:             count++;
duke@1:         }
duke@1: 
duke@1:         public int count;
duke@1:     }
duke@1: }