jdk/test/javax/swing/JPopupMenu/4458079/bug4458079.java

/*
 * Copyright (c) 2013, 2017, 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
 * @bug 4458079
 * @library ../../regtesthelpers
 * @build Util
 * @summary Tests calling removeAll() from PopupMenuListener
 * @author Peter Zhelezniakov
 * @run main bug4458079
 */

import java.awt.Robot;
import java.awt.event.*;
import javax.swing.*;
import javax.swing.event.*;
import java.awt.event.KeyEvent;

public class bug4458079 extends JFrame implements PopupMenuListener {
    public JMenu menu;

    static volatile boolean itemASelected = false;
    public static void main(String[] args) throws Exception {
        Robot robot = new Robot();
        robot.waitForIdle();
        // move mouse outside menu to prevent auto selection
        robot.mouseMove(100,100);

        SwingUtilities.invokeAndWait(new Runnable() {
            public void run() {
                new bug4458079().createAndShowGUI();
            }
        });

        robot.setAutoDelay(50);

        Util.hitMnemonics(robot, KeyEvent.VK_M);

        robot.waitForIdle();
        Thread.sleep(1000);

        Util.hitKeys(robot, KeyEvent.VK_DOWN);
        Util.hitKeys(robot, KeyEvent.VK_ENTER);

        robot.waitForIdle();
        Thread.sleep(1000);

        if (!itemASelected) {
            throw new RuntimeException("Test failed: arrow key traversal in JMenu broken!");
        }
    }
    public void createAndShowGUI() {
        JMenuBar bar = new JMenuBar();
        menu = new JMenu("Menu");
        menu.add(new JMenuItem("1"));
        menu.add(new JMenuItem("2"));
        menu.setMnemonic(KeyEvent.VK_M);
        menu.getPopupMenu().addPopupMenuListener(this);
        bar.add(menu);

        setJMenuBar(bar);
        getContentPane().add(new JButton(""));
        setSize(300, 300);
        setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        pack();
        setLocationRelativeTo(null);
        setVisible(true);
    }

    public void rebuildMenu() {
        menu.removeAll();
        final String itemCommand = "A";
        JMenuItem item = new JMenuItem(itemCommand);
        item.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                JMenuItem item = ((JMenuItem)e.getSource());
                if (e.getActionCommand() == itemCommand) {
                    itemASelected = true;
                }
            }
        });
        menu.add(item);
        menu.add(new JMenuItem("B"));
    }

    public void popupMenuWillBecomeVisible(PopupMenuEvent e) {
        rebuildMenu();
    }

    public void popupMenuWillBecomeInvisible(PopupMenuEvent e) {}
    public void popupMenuCanceled(PopupMenuEvent e) {}
}