RowSelectionTest.java example

Explorer
nebula.widgets.nattable-master
/*******************************************************************************
 * Copyright (c) 2012 Original authors and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Original authors and others - initial API and implementation
 ******************************************************************************/
package org.eclipse.nebula.widgets.nattable.selection;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.util.Collection;

import org.eclipse.nebula.widgets.nattable.coordinate.PositionCoordinate;
import org.eclipse.nebula.widgets.nattable.selection.command.SelectRowsCommand;
import org.eclipse.nebula.widgets.nattable.test.fixture.layer.DataLayerFixture;
import org.eclipse.nebula.widgets.nattable.util.ArrayUtil;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

public class RowSelectionTest {
    // Tests for column selection NTBL-225

    private SelectionLayer selectionLayer;

    @Before
    public void setUp() {
        this.selectionLayer = new SelectionLayer(new DataLayerFixture());
        // Selection grid origin as starting point
        this.selectionLayer.setSelectedCell(0, 0);
    }

    @After
    public void cleanUp() {
        this.selectionLayer.clear();
    }

    @Test
    public void shouldSelectAllCellsInARow() {
        final int columnCount = this.selectionLayer.getColumnCount();
        // User has clicked on second row header cell
        this.selectionLayer.selectRow(1, 2, false, false);

        // Selection anchor should be at row 2, column 0
        Assert.assertEquals(1, this.selectionLayer.getSelectionAnchor().getColumnPosition());
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().getRowPosition());

        // Last selected cell should be part of last column
        Assert.assertEquals(1, this.selectionLayer.getLastSelectedCellPosition().getColumnPosition());
        Assert.assertEquals(2, this.selectionLayer.getLastSelectedCellPosition().getRowPosition());

        // Cells in between should have been selected
        Assert.assertEquals(columnCount, this.selectionLayer.getSelectedColumnPositions().length);
    }

    @Test
    public void shouldExtendSelectionUpWithShiftKey() {
        // User selects cell
        this.selectionLayer.selectCell(2, 2, false, false);
        // User selects column using shift key mask
        this.selectionLayer.selectRow(1, 1, true, false);

        // Selection Anchor should not have changed
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().columnPosition);
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().getRowPosition());

        assertCellsSelectedBetween(2, 1);
    }

    @Test
    public void shouldExtendSelectionDownWithShiftKey() {
        // User selects cell
        this.selectionLayer.selectCell(2, 2, false, false);
        // User selects column using shift key mask
        this.selectionLayer.selectRow(1, 4, true, false);

        // Selection Anchor should not have changed
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().columnPosition);
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().getRowPosition());

        assertCellsSelectedBetween(2, 4);
    }

    @Test
    public void shouldAppendSelectionUpWithShiftKey() {
        // User selects cell
        this.selectionLayer.selectCell(2, 2, false, false);
        // User selects column using shift key mask
        this.selectionLayer.selectRow(1, 1, true, false);

        // Selection Anchor should not have changed
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().columnPosition);
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().getRowPosition());

        assertCellsSelectedBetween(2, 1);

        this.selectionLayer.selectRow(1, 0, true, false);
        assertCellsSelectedBetween(2, 0);
    }

    @Test
    public void shouldAppendSelectionDownWithShiftKey() {
        // User selects cell
        this.selectionLayer.selectCell(2, 2, false, false);
        // User selects column using shift key mask
        this.selectionLayer.selectRow(1, 3, true, false);

        // Selection Anchor should not have changed
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().columnPosition);
        Assert.assertEquals(2, this.selectionLayer.getSelectionAnchor().getRowPosition());

        assertCellsSelectedBetween(2, 3);

        this.selectionLayer.selectRow(1, 5, true, false);
        assertCellsSelectedBetween(2, 5);
    }

    private void assertCellsSelectedBetween(int startRowPosition, int endRowPosition) {
        for (int row = startRowPosition; row <= endRowPosition; row++) {
            for (int col = 0; col <= 4; col++) {
                assertTrue("[" + col + ", " + row + "] not selected",
                        this.selectionLayer.isCellPositionSelected(col, row));
            }
        }
    }

    @Test
    public void shouldExtendSelectionWithAllCellsInARowUsingTheCtrlKey() {
        final int columnCount = this.selectionLayer.getColumnCount();

        // User has selected 3 non-consecutive cells
        this.selectionLayer.selectCell(2, 2, false, false);
        this.selectionLayer.selectCell(3, 2, false, true);
        this.selectionLayer.selectCell(2, 0, false, true);

        // User has clicked on second row header cell
        this.selectionLayer.selectRow(1, 1, false, true);

        // Selection anchor should be at row 1, col 0
        Assert.assertEquals(1, this.selectionLayer.getSelectionAnchor().getColumnPosition());
        Assert.assertEquals(1, this.selectionLayer.getSelectionAnchor().getRowPosition());

        // Last selected cell should be part of last column
        Assert.assertEquals(1, this.selectionLayer.getLastSelectedCellPosition().getRowPosition());
        final int lastColumnPosition = this.selectionLayer.getLastSelectedCellPosition().getColumnPosition();
        Assert.assertEquals(1, lastColumnPosition);

        // Cells in row should have been selected
        final int[] selectedColumns = this.selectionLayer.getSelectedColumnPositions();
        Assert.assertEquals(columnCount, selectedColumns.length);
        Assert.assertTrue(this.selectionLayer.isCellPositionSelected(4, 1));

        // Test extending column selection to the right of previous column
        // selection
        this.selectionLayer.selectRow(1, 3, false, true);

        // Selection model should contain all previously selected cells
        Assert.assertTrue(this.selectionLayer.isCellPositionSelected(2, 2));
        Assert.assertTrue(this.selectionLayer.isCellPositionSelected(3, 2));
        Assert.assertTrue(this.selectionLayer.isCellPositionSelected(2, 0));

        // 3rd row cells should be selected
        Assert.assertTrue(this.selectionLayer.isCellPositionSelected(1, 3));
        Assert.assertTrue(this.selectionLayer.isCellPositionSelected(3, 3));
    }

    @Test
    public void onlyOneCellSelectedAtAnyTime() {
        this.selectionLayer.getSelectionModel().setMultipleSelectionAllowed(false);

        this.selectionLayer.clear();
        this.selectionLayer.doCommand(
                new SelectRowsCommand(this.selectionLayer, 1, 0, false, true));

        Collection<PositionCoordinate> cells =
                ArrayUtil.asCollection(this.selectionLayer.getSelectedCellPositions());
        assertEquals(1, cells.size());
        assertEquals(1, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(1, this.selectionLayer.getSelectedRowCount());

        // select another row with control mask
        this.selectionLayer.doCommand(
                new SelectRowsCommand(this.selectionLayer, 1, 2, false, true));

        cells = ArrayUtil.asCollection(this.selectionLayer.getSelectedCellPositions());
        assertEquals(1, cells.size());
        assertEquals(1, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(1, this.selectionLayer.getSelectedRowCount());

        // select additional rows with shift mask
        this.selectionLayer.doCommand(new SelectRowsCommand(this.selectionLayer, 1, 5,
                true, false));

        cells = ArrayUtil.asCollection(this.selectionLayer.getSelectedCellPositions());
        assertEquals(1, cells.size());
        assertEquals(1, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(1, this.selectionLayer.getSelectedRowCount());
    }

    @Test
    public void testMultiSelectionRestore() {
        this.selectionLayer.clear();
        this.selectionLayer.doCommand(
                new SelectRowsCommand(this.selectionLayer, 1, 0, false, false));

        assertEquals(1, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(1, this.selectionLayer.getSelectedRowCount());

        this.selectionLayer.doCommand(
                new SelectRowsCommand(this.selectionLayer, 1, 2, true, false));

        assertEquals(1, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(3, this.selectionLayer.getSelectedRowCount());

        this.selectionLayer.doCommand(
                new SelectRowsCommand(this.selectionLayer, 1, 2, false, true));

        assertEquals(1, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(2, this.selectionLayer.getSelectedRowCount());

        this.selectionLayer.doCommand(
                new SelectRowsCommand(this.selectionLayer, 1, 1, false, true));

        assertEquals(1, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(1, this.selectionLayer.getSelectedRowCount());

        this.selectionLayer.doCommand(
                new SelectRowsCommand(this.selectionLayer, 1, 0, false, true));

        assertEquals(0, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(0, this.selectionLayer.getSelectedRowCount());

        this.selectionLayer.doCommand(
                new SelectRowsCommand(this.selectionLayer, 1, 0, false, true));

        assertEquals(1, this.selectionLayer.getSelectedRowPositions().size());
        assertEquals(1, this.selectionLayer.getSelectedRowCount());
    }
}