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