1 /*
   2  * Copyright (c) 2014, 2017, 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 import java.awt.Dialog;
  25 import java.awt.Frame;
  26 import java.io.*;
  27 import javax.swing.*;
  28 import sun.awt.SunToolkit;
  29 import java.util.concurrent.atomic.AtomicReference;
  30 
  31 /**
  32  * @test
  33  * @key headful
  34  * @bug 8043705
  35  * @summary Can't exit color chooser dialog when running as an applet
  36  * @modules java.desktop/sun.awt
  37  * @run main CloseDialogTest
  38  */
  39 
  40 public class CloseDialogTest {
  41 
  42     private static volatile Frame frame;
  43     private static volatile Dialog dialog;
  44     private static volatile InputStream testErrorStream;
  45     private static final PrintStream systemErrStream = System.err;
  46     private static final AtomicReference<Exception> caughtException
  47             = new AtomicReference<>();
  48 
  49     public static void main(String[] args) throws Exception {
  50 
  51         // redirect System err
  52         PipedOutputStream errorOutputStream = new PipedOutputStream();
  53         testErrorStream = new PipedInputStream(errorOutputStream);
  54         System.setErr(new PrintStream(errorOutputStream));
  55 
  56         ThreadGroup swingTG = new ThreadGroup(getRootThreadGroup(), "SwingTG");
  57         try {
  58             new Thread(swingTG, () -> {
  59                 SunToolkit.createNewAppContext();
  60                 SwingUtilities.invokeLater(() -> {
  61                     frame = new Frame();
  62                     frame.setSize(300, 300);
  63                     frame.setVisible(true);
  64 
  65                     dialog = new Dialog(frame);
  66                     dialog.setSize(200, 200);
  67                     dialog.setModal(true);
  68                     dialog.setVisible(true);
  69                 });
  70             }).start();
  71 
  72             Thread.sleep(400);
  73 
  74             Thread disposeThread = new Thread(swingTG, () ->
  75                     SwingUtilities.invokeLater(() -> {
  76                 try {
  77                     while (dialog == null || !dialog.isVisible()) {
  78                         Thread.sleep(100);
  79                     }
  80                     dialog.setVisible(false);
  81                     dialog.dispose();
  82                     frame.dispose();
  83                 } catch (Exception e) {
  84                     caughtException.set(e);
  85                 }
  86             }));
  87             disposeThread.start();
  88             disposeThread.join();
  89             Thread.sleep(500);
  90 
  91             // read System err
  92             final char[] buffer = new char[2048];
  93             System.err.print("END");
  94             System.setErr(systemErrStream);
  95             try (Reader in = new InputStreamReader(testErrorStream, "UTF-8")) {
  96                 int size = in.read(buffer, 0, buffer.length);
  97                 String errorString = new String(buffer, 0, size);
  98                 if (!errorString.startsWith("END")) {
  99                     System.err.println(errorString.
 100                             substring(0, errorString.length() - 4));
 101                     throw new RuntimeException("Error output is not empty!");
 102                 }
 103             }
 104         } finally {
 105             if (caughtException.get() != null) {
 106                 throw new RuntimeException("Failed. Caught exception!",
 107                         caughtException.get());
 108             }
 109         }
 110     }
 111 
 112     private static ThreadGroup getRootThreadGroup() {
 113         ThreadGroup threadGroup = Thread.currentThread().getThreadGroup();
 114         while (threadGroup.getParent() != null) {
 115             threadGroup = threadGroup.getParent();
 116         }
 117         return threadGroup;
 118     }
 119 }