1 /*
   2  * Copyright (c) 2010, 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 /* @test
  25    @bug 6894632
  26    @summary Removing rows from a DefaultTableModel with a RowSorter deselectes
  27    last row
  28    @author Semyon Sadetsky
  29 */
  30 
  31 import javax.swing.*;
  32 import javax.swing.table.DefaultTableModel;
  33 import javax.swing.table.TableModel;
  34 import javax.swing.table.TableRowSorter;
  35 import java.util.ArrayList;
  36 import java.util.List;
  37 
  38 public class bug6894632 {
  39     private static JTable table;
  40 
  41     public static void main(String[] args) throws Exception {
  42         SwingUtilities.invokeAndWait(new Runnable() {
  43             public void run() {
  44 
  45                 //init table with empty sort order
  46                 test(new ArrayList<RowSorter.SortKey>());
  47 
  48                 //init table as unsorted
  49                 List<RowSorter.SortKey> sortKeys = new ArrayList<>();
  50                 sortKeys.add(0, new RowSorter.SortKey(0, SortOrder.UNSORTED));
  51                 test(sortKeys);
  52             }
  53         });
  54 
  55         System.out.println("ok");
  56     }
  57 
  58     static void test(final List<RowSorter.SortKey> sortKeys) {
  59         final JFrame frame = new JFrame();
  60         try {
  61             frame.setUndecorated(true);
  62             frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
  63             table = new JTable();
  64             DefaultTableModel tableModel =
  65                     new DefaultTableModel(10, 1) {
  66                         public Object getValueAt(int row, int column) {
  67                             return row == getRowCount() - 1 ? row + "==last" :
  68                                     row;
  69                         }
  70                     };
  71             table.setModel(tableModel);
  72             TableRowSorter<TableModel> sorter =
  73                     new TableRowSorter<TableModel>(tableModel);
  74             sorter.setSortKeys(sortKeys);
  75             table.setRowSorter(sorter);
  76 
  77             frame.setContentPane(table);
  78             frame.pack();
  79             frame.setLocationRelativeTo(null);
  80             frame.setVisible(true);
  81 
  82             int lastRow = table.getRowCount() - 1;
  83 
  84             //select last row
  85             table.setRowSelectionInterval(lastRow, lastRow);
  86 
  87             //remove row above the last
  88             tableModel.removeRow(lastRow - 1);
  89             lastRow = table.getRowCount() - 1;
  90             if (lastRow != table.getSelectedRow()) {
  91                 throw new RuntimeException("last row must be still selected");
  92             }
  93 
  94             //sort table
  95             sortKeys.clear();
  96             sortKeys.add(0, new RowSorter.SortKey(0, SortOrder.ASCENDING));
  97             sorter.setSortKeys(sortKeys);
  98             //remove row above the last
  99             lastRow = table.getRowCount() - 1;
 100             tableModel.removeRow(lastRow - 1);
 101 
 102             if (!table.getValueAt(table.getSelectedRow(), 0).toString()
 103                     .endsWith("==last")) {
 104                 throw new RuntimeException(
 105                         "row ends with \"==last\" row must be still selected");
 106             }
 107 
 108             //make table unsorted again
 109             sortKeys.clear();
 110             sortKeys.add(0, new RowSorter.SortKey(0, SortOrder.UNSORTED));
 111             sorter.setSortKeys(sortKeys);
 112             //remove row above the last
 113             lastRow = table.getRowCount() - 1;
 114             tableModel.removeRow(lastRow - 1);
 115 
 116             lastRow = table.getRowCount() - 1;
 117             if (lastRow != table.getSelectedRow()) {
 118                 throw new RuntimeException(
 119                         "last row must be still selected");
 120             }
 121         } finally {
 122             frame.dispose();
 123         }
 124     }
 125 
 126 
 127 }