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