Mercurial > jdk8-mips64-public > langtools / file revision

 /*

  * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved.

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

  *

  * This code is free software; you can redistribute it and/or modify it

  * under the terms of the GNU General Public License version 2 only, as

  * published by the Free Software Foundation.

  *

  * This code is distributed in the hope that it will be useful, but WITHOUT

  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

  * version 2 for more details (a copy is included in the LICENSE file that

  * accompanied this code).

  *

  * You should have received a copy of the GNU General Public License version

  * 2 along with this work; if not, write to the Free Software Foundation,

  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

  *

  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

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

  * questions.

  */

 import java.io.File;

 import java.util.ArrayList;

 import java.util.List;

 import com.sun.tools.doclint.DocLint;

 import com.sun.tools.doclint.DocLint.BadArgs;

 import java.io.BufferedReader;

 import java.io.FileReader;

 import java.io.IOException;

 import java.io.PrintWriter;

 import java.io.Reader;

 import java.io.StringWriter;

 import java.util.regex.Matcher;

 import java.util.regex.Pattern;

 public class DocLintTester {

     public static void main(String... args) throws Exception {

         new DocLintTester().run(args);

     }

     public void run(String... args) throws Exception {

         String testSrc = System.getProperty("test.src");

         boolean badArgs = false;

         File refFile = null;

         List<String> opts = new ArrayList<String>();

         List<File> files = new ArrayList<File>();

         for (int i = 0; i < args.length; i++) {

             String arg = args[i];

             if (arg.equals("-ref")) {

                 refFile = new File(testSrc, args[++i]);

             } else if (arg.equals("-badargs")) {

                 badArgs = true;

             } else if (arg.startsWith("-Xmsgs")) {

                 opts.add(arg);

             } else if (arg.startsWith("-")) {

                 opts.add(arg);

                 if (i < args.length - 1 && !args[i+1].startsWith("-"))

                     opts.add(args[++i]);

             } else

                 files.add(new File(testSrc, arg));

         }

         check(opts, files, badArgs, refFile);

         if (errors > 0)

             throw new Exception(errors + " errors occurred");

     }

     void check(List<String> opts, List<File> files, boolean expectBadArgs, File refFile) throws Exception {

         List<String> args = new ArrayList<String>();

         args.addAll(opts);

         for (File file: files)

             args.add(file.getPath());

         StringWriter sw = new StringWriter();

         PrintWriter pw = new PrintWriter(sw);

         try {

             new DocLint().run(pw, args.toArray(new String[args.size()]));

             if (expectBadArgs)

                 error("expected exception not thrown");

         } catch (BadArgs e) {

             if (!expectBadArgs)

                 error("unexpected exception caught: " + e);

         }

         pw.flush();

         String out = normalizeNewlines(removeFileNames(sw.toString())).trim();

         if (out != null)

             System.err.println("Output:\n" + out);

         if (refFile == null) {

             if (!out.isEmpty())

                 error("unexpected output");

         } else {

             String expect = readFile(refFile);

             if (!expect.equals(out)) {

                 error("expected output not found");

                 System.err.println("EXPECT>>" + expect + "<<");

                 System.err.println(" FOUND>>" + out    + "<<");

             }

         }

     }

     String readFile(File file) throws IOException {

         StringBuilder sb = new StringBuilder();

         Reader in = new BufferedReader(new FileReader(file));

         try {

             char[] buf = new char[1024];

             int n;

             while ((n = in.read(buf)) != -1)

                 sb.append(buf, 0, n);

         } finally {

             in.close();

         }

         return sb.toString().trim();

     }

     private static final Pattern dirFileLine = Pattern.compile(

             "(?m)"                          // multi-line mode

             + "^(.*?)"                      // directory part of file name

             + "([-A-Za-z0-9.]+:[0-9]+:)");  // file name and line number

     String removeFileNames(String s) {

         Matcher m = dirFileLine.matcher(s);

         StringBuffer sb = new StringBuffer();

         while (m.find()) {

             m.appendReplacement(sb, "$2");

         }

         m.appendTail(sb);

         return sb.toString();

     }

     private static final String nl = System.getProperty("line.separator");

     String normalizeNewlines(String s) {

         return (nl.equals("\n") ? s : s.replace(nl, "\n"));

     }

     void error(String msg) {

         System.err.println("Error: " + msg);

         errors++;

     }

     int errors;

 }