/*
 * Copyright (c) 2015, 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 6894632
 * @summary Removing rows from a DefaultTableModel with a RowSorter deselectes
 * last row
 * @author Semyon Sadetsky
 */

import javax.swing.*;
import javax.swing.table.DefaultTableModel;
import javax.swing.table.TableModel;
import javax.swing.table.TableRowSorter;
import java.util.ArrayList;
import java.util.List;

public class bug6894632 {
    private static JTable table;

    public static void main(String[] args) throws Exception {
        SwingUtilities.invokeAndWait(new Runnable() {
            public void run() {

                //init table with empty sort order
                test(new ArrayList<RowSorter.SortKey>());

                //init table as unsorted
                List<RowSorter.SortKey> sortKeys = new ArrayList<>();
                sortKeys.add(0, new RowSorter.SortKey(0, SortOrder.UNSORTED));
                test(sortKeys);
            }
        });

        System.out.println("ok");
    }

    static void test(final List<RowSorter.SortKey> sortKeys) {
        final JFrame frame = new JFrame();
        try {
            frame.setUndecorated(true);
            frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
            table = new JTable();
            DefaultTableModel tableModel =
                    new DefaultTableModel(10, 1) {
                        public Object getValueAt(int row, int column) {
                            return row == getRowCount() - 1 ? row + "==last" :
                                    row;
                        }
                    };
            table.setModel(tableModel);
            TableRowSorter<TableModel> sorter =
                    new TableRowSorter<TableModel>(tableModel);
            sorter.setSortKeys(sortKeys);
            table.setRowSorter(sorter);

            frame.setContentPane(table);
            frame.pack();
            frame.setLocationRelativeTo(null);
            frame.setVisible(true);

            int lastRow = table.getRowCount() - 1;

            //select last row
            table.setRowSelectionInterval(lastRow, lastRow);

            //remove row above the last
            tableModel.removeRow(lastRow - 1);
            lastRow = table.getRowCount() - 1;
            if (lastRow != table.getSelectedRow()) {
                throw new RuntimeException("last row must be still selected");
            }

            //sort table
            sortKeys.clear();
            sortKeys.add(0, new RowSorter.SortKey(0, SortOrder.ASCENDING));
            sorter.setSortKeys(sortKeys);
            //remove row above the last
            lastRow = table.getRowCount() - 1;
            tableModel.removeRow(lastRow - 1);

            if (!table.getValueAt(table.getSelectedRow(), 0).toString()
                    .endsWith("==last")) {
                throw new RuntimeException(
                        "row ends with \"==last\" row must be still selected");
            }

            //make table unsorted again
            sortKeys.clear();
            sortKeys.add(0, new RowSorter.SortKey(0, SortOrder.UNSORTED));
            sorter.setSortKeys(sortKeys);
            //remove row above the last
            lastRow = table.getRowCount() - 1;
            tableModel.removeRow(lastRow - 1);

            lastRow = table.getRowCount() - 1;
            if (lastRow != table.getSelectedRow()) {
                throw new RuntimeException(
                        "last row must be still selected");
            }
        } finally {
            frame.dispose();
        }
    }


}