JBR-1417 [TEST] Added new regression test (JBR 11 does not support chain of popups)

(cherry picked from commit 41e89505be)
(cherry picked from commit 9fe5c778d9)
(cherry picked from commit 6ea9530d9f)
(cherry picked from commit d757108517)
(cherry picked from commit b99c1e7b5c)

test/jdk/jb/java/awt/Focus/ChainOfPopupsFocusTest.java

@@ -0,0 +1,186 @@
+/*
+ * Copyright 2000-2019 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import javax.swing.JComponent;
+import javax.swing.JFrame;
+import javax.swing.JScrollPane;
+import javax.swing.JTextArea;
+import javax.swing.KeyStroke;
+import javax.swing.Popup;
+import javax.swing.PopupFactory;
+import javax.swing.SwingUtilities;
+import javax.swing.WindowConstants;
+import java.awt.Component;
+import java.awt.Container;
+import java.awt.FocusTraversalPolicy;
+import java.awt.KeyboardFocusManager;
+import java.awt.Point;
+import java.awt.Robot;
+import java.awt.Window;
+import java.awt.event.ActionEvent;
+import java.awt.event.ActionListener;
+import java.awt.event.ComponentAdapter;
+import java.awt.event.ComponentEvent;
+import java.awt.event.KeyEvent;
+
+/**
+ * @test
+ * @summary Regression test for JBR-1417: JBR 11 does not support chain of popups
+ * @requires (jdk.version.major >= 8)
+ * @run main ChainOfPopupsFocusTest
+ */
+
+/*
+ * Description: Test checks that focus goes to the parent popup when child popup is closed.
+ * Test opens several popups one by one, setting current popup as a parent for the next one.
+ * Then the popups are closed one by one. Test fails if focus went somewhere else than the parent popup.
+ */
+
+public class ChainOfPopupsFocusTest implements Runnable, ActionListener {
+
+    private static final int DEPTH = 5;
+    private static final String FRAMENAME = "MainFrame";
+    private static final String POPUPNAME = "Popup";
+
+    private static volatile Component focusOwner;
+    private static volatile int popupsCount;
+    private static Robot robot;
+    private JFrame frame;
+
+    public static void main(String[] args) throws Exception {
+
+        robot = new Robot();
+        robot.setAutoDelay(50);
+
+        ChainOfPopupsFocusTest test = new ChainOfPopupsFocusTest();
+        SwingUtilities.invokeAndWait(test);
+        robot.delay(3000);
+
+        try {
+            SwingUtilities.invokeAndWait(() ->
+                    focusOwner = KeyboardFocusManager.getCurrentKeyboardFocusManager().getFocusOwner());
+            if(focusOwner == null || !focusOwner.getName().equals(FRAMENAME)) {
+                throw new RuntimeException("Test ERROR: " + FRAMENAME + " is not focused");
+            }
+
+            for (int count = 1; count <= DEPTH; count++) {
+                pressCtrlKey(KeyEvent.VK_N);
+                robot.delay(1000);
+            }
+
+            if(popupsCount != DEPTH) {
+                throw new RuntimeException("Test ERROR: Number of open popups is "
+                        + popupsCount + ", but " + DEPTH + " is expected");
+            }
+            for (int count = DEPTH-1; count >= 0; count--) {
+                pressCtrlKey(KeyEvent.VK_X);
+                robot.delay(1000);
+                SwingUtilities.invokeAndWait(() ->
+                        focusOwner = KeyboardFocusManager.getCurrentKeyboardFocusManager().getFocusOwner());
+                String focusedComponent = (focusOwner != null ? focusOwner.getName() : "Nothing");
+                String expectedComponent = (count == 0 ? FRAMENAME : POPUPNAME + count);
+                if(!focusedComponent.equals(expectedComponent)) {
+                    throw new RuntimeException("Test FAILED: "
+                            + focusedComponent + " is focused instead of " + expectedComponent);
+                }
+            }
+        } finally {
+            SwingUtilities.invokeAndWait(() -> test.frame.dispose());
+            robot.delay(2000);
+        }
+        System.out.println("Test PASSED");
+    }
+
+    private static void pressCtrlKey(int vk) {
+        robot.keyPress(KeyEvent.VK_CONTROL);
+        robot.keyPress(vk);
+        robot.keyRelease(vk);
+        robot.keyRelease(KeyEvent.VK_CONTROL);
+    }
+
+    @Override
+    public void run() {
+        JTextArea area = createTextArea(FRAMENAME);
+        frame = new JFrame(getClass().getSimpleName());
+        frame.add(new JScrollPane(area));
+        frame.pack();
+        frame.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);
+        frame.setLocationRelativeTo(null);
+        frame.setVisible(true);
+        frame.toFront();
+        area.requestFocusInWindow();
+    }
+
+    private JTextArea createTextArea(String locatedOn) {
+        JTextArea area = new JTextArea(20, 40);
+        area.setName(locatedOn);
+        area.registerKeyboardAction(this, "show", KeyStroke.getKeyStroke("control N"), JComponent.WHEN_FOCUSED);
+        area.registerKeyboardAction(event -> SwingUtilities.getWindowAncestor((Component) event.getSource()).setVisible(false), KeyStroke.getKeyStroke("control X"), JComponent.WHEN_FOCUSED);
+        area.addComponentListener(new ComponentAdapter() {
+            @Override
+            public void componentShown(ComponentEvent event) {
+                area.requestFocus();
+            }
+        });
+        return area;
+    }
+
+    @Override
+    public void actionPerformed(ActionEvent event) {
+        Component source = (Component) event.getSource();
+        switch (event.getActionCommand()) {
+            case "show":
+                Point point = source.getLocationOnScreen();
+                Component area = createTextArea(POPUPNAME + ++popupsCount);
+                Popup popup = PopupFactory.getSharedInstance().getPopup(source, new JScrollPane(area), point.x + 10, point.y + 10);
+                Window window = SwingUtilities.getWindowAncestor(area);
+                window.setAutoRequestFocus(true);
+                window.setFocusable(true);
+                window.setFocusableWindowState(true);
+                window.setFocusTraversalPolicy(new FocusTraversalPolicy() {
+                    @Override
+                    public Component getComponentAfter(Container container, Component component) {
+                        return area;
+                    }
+
+                    @Override
+                    public Component getComponentBefore(Container container, Component component) {
+                        return area;
+                    }
+
+                    @Override
+                    public Component getFirstComponent(Container container) {
+                        return area;
+                    }
+
+                    @Override
+                    public Component getLastComponent(Container container) {
+                        return area;
+                    }
+
+                    @Override
+                    public Component getDefaultComponent(Container container) {
+                        return area;
+                    }
+                });
+                popup.show();
+                break;
+            case "hide":
+                SwingUtilities.getWindowAncestor(source).setVisible(false);
+                break;
+        }
+    }
+}