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 }