1 /*

     2  * Copyright (c) 2013, 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 package jdk.testlibrary;

    25 

    26 /*

    27  * @test

    28  * @summary Test the OutputAnalyzer reporting functionality,

    29  *     such as printing additional diagnostic info

    30  *     (exit code, stdout, stderr, command line, etc.)

    31  * @modules java.management

    32  * @build jdk.testlibrary.*

    33  * @run main jdk.testlibrary.OutputAnalyzerReportingTest

    34  */

    35 

    36 import java.io.ByteArrayOutputStream;

    37 import java.io.PrintStream;

    38 

    39 public class OutputAnalyzerReportingTest {

    40 

    41     public static void main(String[] args) throws Exception {

    42         // Create the output analyzer under test

    43         String stdout = "aaaaaa";

    44         String stderr = "bbbbbb";

    45         OutputAnalyzer output = new OutputAnalyzer(stdout, stderr);

    46 

    47         // Expected summary values should be the same for all cases,

    48         // since the outputAnalyzer object is the same

    49         String expectedExitValue = "-1";

    50         String expectedSummary =

    51                 " stdout: [" + stdout + "];\n" +

    52                 " stderr: [" + stderr + "]\n" +

    53                 " exitValue = " + expectedExitValue + "\n";

    54 

    55 

    56         DiagnosticSummaryTestRunner testRunner =

    57                 new DiagnosticSummaryTestRunner();

    58 

    59         // should have exit value

    60         testRunner.init(expectedSummary);

    61         int unexpectedExitValue = 2;

    62         try {

    63             output.shouldHaveExitValue(unexpectedExitValue);

    64         } catch (RuntimeException e) { }

    65         testRunner.closeAndCheckResults();

    66 

    67         // should not contain

    68         testRunner.init(expectedSummary);

    69         try {

    70             output.shouldNotContain(stdout);

    71         } catch (RuntimeException e) { }

    72         testRunner.closeAndCheckResults();

    73 

    74         // should contain

    75         testRunner.init(expectedSummary);

    76         try {

    77             output.shouldContain("unexpected-stuff");

    78         } catch (RuntimeException e) { }

    79         testRunner.closeAndCheckResults();

    80 

    81         // should not match

    82         testRunner.init(expectedSummary);

    83         try {

    84             output.shouldNotMatch("[a]");

    85         } catch (RuntimeException e) { }

    86         testRunner.closeAndCheckResults();

    87 

    88         // should match

    89         testRunner.init(expectedSummary);

    90         try {

    91             output.shouldMatch("[qwerty]");

    92         } catch (RuntimeException e) { }

    93         testRunner.closeAndCheckResults();

    94 

    95     }

    96 

    97     private static class DiagnosticSummaryTestRunner {

    98         private ByteArrayOutputStream byteStream =

    99                 new ByteArrayOutputStream(10000);

   100 

   101         private String expectedSummary = "";

   102         private PrintStream errStream;

   103 

   104 

   105         public void init(String expectedSummary) {

   106             this.expectedSummary = expectedSummary;

   107             byteStream.reset();

   108             errStream = new PrintStream(byteStream);

   109             System.setErr(errStream);

   110         }

   111 

   112         public void closeAndCheckResults() {

   113             // check results

   114             errStream.close();

   115             String stdErrStr = byteStream.toString();

   116             if (!stdErrStr.contains(expectedSummary)) {

   117                 throw new RuntimeException("The output does not contain "

   118                     + "the diagnostic message, or the message is incorrect");

   119             }

   120         }

   121     }

   122 

   123 }