/*

 * Copyright (c) 2010, 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.

 */

/* @test

 * @key headful

 * @summary Test J2Ddemo.jar

 * @run main/timeout=200 J2DdemoTest

 */

import java.io.InputStream;

import java.io.IOException;

import java.io.File;

import java.io.BufferedInputStream;

import java.io.PrintStream;

public class J2DdemoTest {

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

        DemoRun test;

        /* Run the J2Ddemo.jar with the -runs=1 option */

        test = new DemoRun("demo/jfc/J2Ddemo/J2Ddemo.jar", "-runs=1");

        test.runit();

        /* Make sure patterns in output look ok */

        if (test.output_contains("ERROR")) {

            throw new RuntimeException("Test failed - ERROR seen in output");

        }

        if (test.output_contains("Exception")) {

            throw new RuntimeException("Test failed - Exception seen in output");

        }

        /* Must be a pass. */

        System.out.println("Test passed - cleanly terminated");

    }

    /*

     * Helper class to direct process output to a StringBuffer

     */

    static class MyInputStream implements Runnable {

        private String              name;

        private BufferedInputStream in;

        private StringBuffer        buffer;

        /* Create MyInputStream that saves all output to a StringBuffer */

        MyInputStream(String name, InputStream in) {

            this.name = name;

            this.in = new BufferedInputStream(in);

            buffer = new StringBuffer(4096);

            Thread thr = new Thread(this);

            thr.setDaemon(true);

            thr.start();

        }

        /* Dump the buffer */

        void dump(PrintStream x) {

            String str = buffer.toString();

            x.println("<beginning of " + name + " buffer>");

            x.println(str);

            x.println("<end of buffer>");

        }

        /* Check to see if a pattern is inside the output. */

        boolean contains(String pattern) {

            String str = buffer.toString();

            return str.contains(pattern);

        }

        /* Runs as a separate thread capturing all output in a StringBuffer */

        public void run() {

            try {

                byte b[] = new byte[100];

                for (;;) {

                    int n = in.read(b);

                    String str;

                    if (n < 0) {

                        break;

                    }

                    str = new String(b, 0, n);

                    buffer.append(str);

                    System.out.print(str);

                }

            } catch (IOException ioe) { /* skip */ }

        }

    }

    /*

     * Generic run of a demo jar file.

     */

    static class DemoRun {

        private String        demo_name;

        private String        demo_options;

        private MyInputStream output;

        private MyInputStream error;

        /* Create a Demo run process */

        public DemoRun(String name, String options)

        {

            demo_name    = name;

            demo_options = options;

        }

        /*

         * Execute the demo

         */

        public void runit()

        {

            String jre_home  = System.getProperty("java.home");

            String sdk_home  = (jre_home.endsWith("jre") ?

                                (jre_home + File.separator + "..") :

                                jre_home );

            String java      = sdk_home

                                 + File.separator + "bin"

                                 + File.separator + "java";

            /* VM options */

            String vm_opts[] = new String[0];

            String vopts = System.getProperty("test.vm.opts");

            if ( vopts != null && vopts.length()>0 ) {

                vm_opts = vopts.split("\\p{Space}+");

            } else {

                vm_opts = new String[0];

            }

            /* Command line */

            String cmd[] = new String[1 + vm_opts.length + 3];

            String cmdLine;

            int i;

            i = 0;

            cmdLine = "";

            cmdLine += (cmd[i++] = java);

            cmdLine += " ";

            for ( String vopt : vm_opts ) {

                cmdLine += (cmd[i++] = vopt);

                cmdLine += " ";

            }

            cmdLine += (cmd[i++] = "-jar");

            cmdLine += " ";

            cmdLine += (cmd[i++] = sdk_home + File.separator + demo_name);

            cmdLine += " ";

            cmdLine += (cmd[i++] = demo_options);

            /* Begin process */

            Process p;

            System.out.println("Starting: " + cmdLine);

            try {

                p = Runtime.getRuntime().exec(cmd);

            } catch ( IOException e ) {

                throw new RuntimeException("Test failed - exec got IO exception");

            }

            /* Save process output in StringBuffers */

            output = new MyInputStream("Input Stream", p.getInputStream());

            error  = new MyInputStream("Error Stream", p.getErrorStream());

            /* Wait for process to complete, and if exit code is non-zero we fail */

            int exitStatus;

            try {

                exitStatus = p.waitFor();

                if ( exitStatus != 0) {

                    System.out.println("Exit code is " + exitStatus);

                    error.dump(System.out);

                    output.dump(System.out);

                    throw new RuntimeException("Test failed - " +

                                        "exit return code non-zero " +

                                        "(exitStatus==" + exitStatus + ")");

                }

            } catch ( InterruptedException e ) {

                throw new RuntimeException("Test failed - process interrupted");

            }

            System.out.println("Completed: " + cmdLine);

        }

        /* Does the pattern appear in the output of this process */

        public boolean output_contains(String pattern)

        {

            return output.contains(pattern) || error.contains(pattern);

        }

    }

}