1/*
2 * Copyright (c) 2006, 2015, 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 * @test
26 * @bug 6403468
27 * @summary Test that an erroneous return code results from raising an error.
28 * @author  Joseph D. Darcy
29 * @library /tools/javac/lib
30 * @modules jdk.compiler/com.sun.tools.javac.main
31 * @build JavacTestingAbstractProcessor CompileFail
32 * @compile TestReturnCode.java
33 *
34 * @compile                     -processor TestReturnCode -proc:only                                                                   Foo.java
35 * @run main CompileFail ERROR  -processor TestReturnCode -proc:only                                                    -AErrorOnFirst Foo.java
36 * @run main CompileFail ERROR  -processor TestReturnCode -proc:only                                      -AErrorOnLast                Foo.java
37 * @run main CompileFail ERROR  -processor TestReturnCode -proc:only                                      -AErrorOnLast -AErrorOnFirst Foo.java
38 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only                   -AExceptionOnFirst                              Foo.java
39 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only                   -AExceptionOnFirst               -AErrorOnFirst Foo.java
40 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only                   -AExceptionOnFirst -AErrorOnLast                Foo.java
41 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only                   -AExceptionOnFirst -AErrorOnLast -AErrorOnFirst Foo.java
42 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only -AExceptionOnLast                                                 Foo.java
43 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only -AExceptionOnLast                                  -AErrorOnFirst Foo.java
44 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only -AExceptionOnLast                    -AErrorOnLast                Foo.java
45 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only -AExceptionOnLast                    -AErrorOnLast -AErrorOnFirst Foo.java
46 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only -AExceptionOnLast -AExceptionOnFirst                              Foo.java
47 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only -AExceptionOnLast -AExceptionOnFirst               -AErrorOnFirst Foo.java
48 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only -AExceptionOnLast -AExceptionOnFirst -AErrorOnLast                Foo.java
49 * @run main CompileFail SYSERR -processor TestReturnCode -proc:only -AExceptionOnLast -AExceptionOnFirst -AErrorOnLast -AErrorOnFirst Foo.java
50 */
51
52import java.util.Set;
53import java.util.HashSet;
54import java.util.Arrays;
55import javax.annotation.processing.*;
56import javax.lang.model.SourceVersion;
57import javax.lang.model.element.*;
58import javax.lang.model.util.*;
59import static javax.tools.Diagnostic.Kind.*;
60
61
62/**
63 * This processor raises errors or throws exceptions on different
64 * rounds to allow the return code to be test.
65 */
66@SupportedOptions({"ErrorOnFirst",
67                   "ErrorOnLast",
68                   "ExceptionOnFirst",
69                   "ExceptionOnLast"})
70public class TestReturnCode extends JavacTestingAbstractProcessor {
71
72    private boolean errorOnFirst;
73    private boolean errorOnLast;
74    private boolean exceptionOnFirst;
75    private boolean exceptionOnLast;
76
77    public boolean process(Set<? extends TypeElement> annotations,
78                           RoundEnvironment roundEnv) {
79        if (!roundEnv.processingOver()) {
80            System.out.format("Variables: %b\t%b\t%b\t%b%n",
81                              errorOnFirst,
82                              errorOnLast,
83                              exceptionOnFirst,
84                              exceptionOnLast);
85            if (errorOnFirst)
86                messager.printMessage(ERROR, "Error on first round.");
87            if (exceptionOnFirst)
88                throw new RuntimeException("Exception on first round.");
89        } else {
90            if (errorOnLast)
91                messager.printMessage(ERROR, "Error on last round.");
92            if (exceptionOnLast)
93                throw new RuntimeException("Exception on last round.");
94        }
95        return true;
96    }
97
98    @Override
99    public void init(ProcessingEnvironment processingEnv) {
100        super.init(processingEnv);
101        Set<String> keySet = processingEnv.getOptions().keySet();
102        errorOnFirst      = keySet.contains("ErrorOnFirst");
103        errorOnLast     = keySet.contains("ErrorOnLast");
104        exceptionOnFirst  = keySet.contains("ExceptionOnFirst");
105        exceptionOnLast = keySet.contains("ExceptionOnLast");
106    }
107}
108