1 /*

     2  * Copyright (c) 2012, 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 /*

    25  * @test

    26  * @bug 4330357

    27  * @summary Tests that real editor in JTree cleans up after editing was stopped

    28  * @library ../../regtesthelpers

    29  * @build Util

    30  * @author Peter Zhelezniakov

    31  * @run main bug4330357

    32  */

    33 import java.awt.*;

    34 import java.awt.event.*;

    35 import javax.swing.*;

    36 import javax.swing.tree.*;

    37 import sun.awt.SunToolkit;

    38 

    39 public class bug4330357 {

    40 

    41     private static JTree tree;

    42     private static JButton button;

    43     private static Robot robot;

    44 

    45     public static void main(String[] args) throws Exception {

    46         SunToolkit toolkit = (SunToolkit) Toolkit.getDefaultToolkit();

    47         robot = new Robot();

    48         robot.setAutoDelay(50);

    49 

    50         UIManager.setLookAndFeel("javax.swing.plaf.metal.MetalLookAndFeel");

    51 

    52         javax.swing.SwingUtilities.invokeAndWait(new Runnable() {

    53 

    54             public void run() {

    55                 createAndShowGUI();

    56             }

    57         });

    58 

    59         toolkit.realSync();

    60 

    61         clickMouse(getTreeRowClickPoint(1));

    62         Util.hitKeys(robot, KeyEvent.VK_F2);

    63         Util.hitKeys(robot, KeyEvent.VK_A, KeyEvent.VK_B, KeyEvent.VK_C);

    64         toolkit.realSync();

    65 

    66         if (!hasComponent(JTextField.class)) {

    67             throw new RuntimeException("Cell editor is missed for path: color");

    68         }

    69 

    70 

    71         clickMouse(getButtonClickPoint());

    72         toolkit.realSync();

    73 

    74         clickMouse(getTreeRowClickPoint(2));

    75         Util.hitKeys(robot, KeyEvent.VK_F2);

    76         toolkit.realSync();

    77 

    78         if (!hasComponent(JComboBox.class)) {

    79             throw new RuntimeException("Cell editor is missed for path: sports");

    80         }

    81 

    82         if (hasComponent(JTextField.class)) {

    83             throw new RuntimeException("Cell editor is wrongly shown for path: color");

    84         }

    85     }

    86 

    87     static void clickMouse(Point point) {

    88         robot.mouseMove(point.x, point.y);

    89         robot.mousePress(InputEvent.BUTTON1_MASK);

    90         robot.mouseRelease(InputEvent.BUTTON1_MASK);

    91     }

    92 

    93     private static Point getTreeRowClickPoint(final int row) throws Exception {

    94         final Point[] result = new Point[1];

    95 

    96         SwingUtilities.invokeAndWait(new Runnable() {

    97 

    98             @Override

    99             public void run() {

   100 

   101                 Rectangle rect = tree.getRowBounds(row);

   102                 Point p = new Point(rect.x + rect.width / 2, rect.y + 2);

   103                 SwingUtilities.convertPointToScreen(p, tree);

   104                 result[0] = p;

   105             }

   106         });

   107 

   108         return result[0];

   109     }

   110 

   111     private static Point getButtonClickPoint() throws Exception {

   112         final Point[] result = new Point[1];

   113 

   114         SwingUtilities.invokeAndWait(new Runnable() {

   115 

   116             @Override

   117             public void run() {

   118                 Point p = button.getLocationOnScreen();

   119                 Dimension size = button.getSize();

   120                 result[0] = new Point(p.x + size.width / 2, p.y + size.height / 2);

   121             }

   122         });

   123         return result[0];

   124     }

   125 

   126     static boolean hasComponent(final Class cls) throws Exception {

   127         final boolean[] result = new boolean[1];

   128 

   129         SwingUtilities.invokeAndWait(new Runnable() {

   130 

   131             @Override

   132             public void run() {

   133                 result[0] = Util.findSubComponent(tree, cls.getName()) != null;

   134             }

   135         });

   136 

   137         return result[0];

   138     }

   139 

   140     private static void createAndShowGUI() {

   141         JFrame frame = new JFrame("Test");

   142         frame.setSize(200, 200);

   143         frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

   144 

   145         tree = new JTree();

   146         tree.setEditable(true);

   147 

   148         final TestEditor testEditor = new TestEditor();

   149         tree.setCellEditor(new DefaultTreeCellEditor(tree,

   150                 (DefaultTreeCellRenderer) tree.getCellRenderer(),

   151                 testEditor));

   152 

   153         button = new JButton("stop");

   154 

   155         button.addActionListener(new ActionListener() {

   156 

   157             public void actionPerformed(ActionEvent ae) {

   158                 testEditor.stopCellEditing();

   159             }

   160         });

   161 

   162         frame.getContentPane().add(new JScrollPane(tree), BorderLayout.CENTER);

   163         frame.getContentPane().add(button, BorderLayout.SOUTH);

   164         frame.setVisible(true);

   165     }

   166 

   167     static class TestEditor extends AbstractCellEditor implements TreeCellEditor {

   168 

   169         private JComboBox comboBox;

   170         private JTextField textField;

   171         private boolean comboBoxActive;

   172 

   173         TestEditor() {

   174             comboBox = new JComboBox(new String[]{"one", "two"});

   175             textField = new JTextField();

   176         }

   177 

   178         public Component getTreeCellEditorComponent(JTree tree, Object value,

   179                 boolean isSelected,

   180                 boolean expanded,

   181                 boolean leaf, int row) {

   182             if (row % 2 == 0) {

   183                 comboBoxActive = true;

   184                 return comboBox;

   185             }

   186             comboBoxActive = false;

   187             return textField;

   188         }

   189 

   190         public Object getCellEditorValue() {

   191             if (comboBoxActive) {

   192                 return comboBox.getSelectedItem();

   193             }

   194             return textField.getText();

   195         }

   196     }

   197 }