duke@1: /*
ksrini@1138:  * Copyright (c) 2005, 2011, 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 6304912
duke@1:  * @summary unit test for Log
duke@1:  */
duke@1: import java.io.InputStream;
duke@1: import java.io.IOException;
duke@1: import java.io.OutputStream;
duke@1: import java.net.URI;
duke@1: import javax.tools.JavaFileObject;
duke@1: import javax.tools.SimpleJavaFileObject;
jjg@50: import com.sun.tools.javac.file.JavacFileManager;
ksrini@1138: import com.sun.tools.javac.parser.EndPosTable;
duke@1: import com.sun.tools.javac.parser.Parser;
jjg@111: import com.sun.tools.javac.parser.ParserFactory;
duke@1: import com.sun.tools.javac.tree.JCTree;
duke@1: import com.sun.tools.javac.tree.TreeScanner;
duke@1: import com.sun.tools.javac.util.Context;
duke@1: import com.sun.tools.javac.util.Log;
duke@1: import com.sun.tools.javac.util.JCDiagnostic;
duke@1: import com.sun.tools.javac.util.Options;
duke@1: 
duke@1: public class TestLog
duke@1: {
duke@1:     public static void main(String... args) throws IOException {
duke@1:         test(false);
duke@1:         test(true);
duke@1:     }
duke@1: 
duke@1:     static void test(boolean genEndPos) throws IOException {
duke@1:         Context context = new Context();
duke@1: 
duke@1:         Options options = Options.instance(context);
duke@1:         options.put("diags", "%b:%s/%o/%e:%_%t%m|%p%m");
duke@1: 
duke@1:         Log log = Log.instance(context);
duke@1:         log.multipleErrors = true;
duke@1: 
duke@1:         JavacFileManager.preRegister(context);
jjg@111:         ParserFactory pfac = ParserFactory.instance(context);
duke@1: 
duke@1:         final String text =
duke@1:               "public class Foo {\n"
duke@1:             + "  public static void main(String[] args) {\n"
duke@1:             + "    if (args.length == 0)\n"
duke@1:             + "      System.out.println(\"no args\");\n"
duke@1:             + "    else\n"
duke@1:             + "      System.out.println(args.length + \" args\");\n"
duke@1:             + "  }\n"
duke@1:             + "}\n";
duke@1:         JavaFileObject fo = new StringJavaFileObject("Foo", text);
duke@1:         log.useSource(fo);
duke@1: 
jjg@111:         CharSequence cs = fo.getCharContent(true);
jjg@111:         Parser parser = pfac.newParser(cs, false, genEndPos, false);
jjg@111:         JCTree.JCCompilationUnit tree = parser.parseCompilationUnit();
duke@1:         log.setEndPosTable(fo, tree.endPositions);
duke@1: 
duke@1:         TreeScanner ts = new LogTester(log, tree.endPositions);
duke@1:         ts.scan(tree);
duke@1: 
duke@1:         check(log.nerrors, 4, "errors");
duke@1:         check(log.nwarnings, 4, "warnings");
duke@1:     }
duke@1: 
duke@1:     private static void check(int found, int expected, String name) {
duke@1:         if (found == expected)
duke@1:             System.err.println(found + " " + name + " found, as expected.");
duke@1:         else {
duke@1:             System.err.println("incorrect number of " + name + " found.");
duke@1:             System.err.println("expected: " + expected);
duke@1:             System.err.println("   found: " + found);
duke@1:             throw new IllegalStateException("test failed");
duke@1:         }
duke@1:     }
duke@1: 
duke@1:     private static class LogTester extends TreeScanner {
ksrini@1138:         LogTester(Log log, EndPosTable endPosTable) {
duke@1:             this.log = log;
ksrini@1138:             this.endPosTable = endPosTable;
duke@1:         }
duke@1: 
duke@1:         public void visitIf(JCTree.JCIf tree) {
duke@1:             JCDiagnostic.DiagnosticPosition nil = null;
duke@1:             // generate dummy messages to exercise the log API
duke@1:             log.error("not.stmt");
duke@1:             log.error(tree.pos, "not.stmt");
duke@1:             log.error(tree.pos(), "not.stmt");
duke@1:             log.error(nil, "not.stmt");
duke@1: 
duke@1:             log.warning("div.zero");
duke@1:             log.warning(tree.pos, "div.zero");
duke@1:             log.warning(tree.pos(), "div.zero");
duke@1:             log.warning(nil, "div.zero");
duke@1:         }
duke@1: 
duke@1:         private Log log;
ksrini@1138:         private EndPosTable endPosTable;
duke@1:     }
duke@1: 
duke@1:     private static class StringJavaFileObject extends SimpleJavaFileObject {
duke@1:         StringJavaFileObject(String name, String text) {
duke@1:             super(URI.create(name), JavaFileObject.Kind.SOURCE);
duke@1:             this.text = text;
duke@1:         }
duke@1:         public CharSequence getCharContent(boolean b) {
duke@1:             return text;
duke@1:         }
duke@1:         public InputStream openInputStream() {
duke@1:             throw new UnsupportedOperationException();
duke@1:         }
duke@1:         public OutputStream openOutputStream() {
duke@1:             throw new UnsupportedOperationException();
duke@1:         }
duke@1:         private String text;
duke@1:     }
duke@1: }