/*******************************************************************************
* 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.search.strategy;
import org.eclipse.nebula.widgets.nattable.config.ConfigRegistry;
import org.eclipse.nebula.widgets.nattable.config.DefaultNatTableStyleConfiguration;
import org.eclipse.nebula.widgets.nattable.coordinate.PositionCoordinate;
import org.eclipse.nebula.widgets.nattable.data.IDataProvider;
import org.eclipse.nebula.widgets.nattable.grid.command.ClientAreaResizeCommand;
import org.eclipse.nebula.widgets.nattable.grid.layer.DefaultGridLayer;
import org.eclipse.nebula.widgets.nattable.search.CellValueAsStringComparator;
import org.eclipse.nebula.widgets.nattable.search.ISearchDirection;
import org.eclipse.nebula.widgets.nattable.search.strategy.SelectionSearchStrategy;
import org.eclipse.nebula.widgets.nattable.selection.SelectionLayer;
import org.eclipse.nebula.widgets.nattable.selection.command.SelectAllCommand;
import org.eclipse.nebula.widgets.nattable.selection.command.SelectCellCommand;
import org.eclipse.nebula.widgets.nattable.test.fixture.layer.GridLayerFixture;
import org.eclipse.nebula.widgets.nattable.util.IClientAreaProvider;
import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Rectangle;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Shell;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
public class SelectionSearchStrategyTest {
private static final String CELL_VALUE = "EVEN_BODY_CELL";
private DefaultGridLayer gridLayer;
private ConfigRegistry configRegistry;
private IDataProvider bodyDataProvider;
@Before
public void setUp() {
bodyDataProvider = new IDataProvider() {
public int getColumnCount() {
return GridLayerFixture.bodyDataProvider.getColumnCount();
}
public int getRowCount() {
return GridLayerFixture.bodyDataProvider.getRowCount();
}
public Object getDataValue(int columnIndex, int rowIndex) {
if (columnIndex == 0 || columnIndex == 9) {
return CELL_VALUE;
}
return (String) GridLayerFixture.bodyDataProvider.getDataValue(
columnIndex, rowIndex);
}
public void setDataValue(int columnIndex, int rowIndex,
Object newValue) {
throw new UnsupportedOperationException();
}
};
gridLayer = new DefaultGridLayer(bodyDataProvider,
GridLayerFixture.colHeaderDataProvider,
GridLayerFixture.rowHeaderDataProvider,
GridLayerFixture.cornerDataProvider);
gridLayer.setClientAreaProvider(new IClientAreaProvider() {
public Rectangle getClientArea() {
return new Rectangle(0, 0, 1050, 250);
}
});
gridLayer.doCommand(new ClientAreaResizeCommand(new Shell(Display
.getDefault(), SWT.V_SCROLL | SWT.H_SCROLL)));
configRegistry = new ConfigRegistry();
new DefaultNatTableStyleConfiguration()
.configureRegistry(configRegistry);
}
@Test
public void shouldAccessWhatIsInSelection() {
// Select range of cells
int startColumnPosition = 2;
int startRowPosition = 1;
final int lastColumn = 7;
final int lastRow = 5;
for (int columnPosition = startColumnPosition; columnPosition < lastColumn; columnPosition++) {
for (int rowPosition = startRowPosition; rowPosition < lastRow; rowPosition++) {
gridLayer.doCommand(new SelectCellCommand(gridLayer,
columnPosition, rowPosition, false, true));
}
}
// We should get 20 Cells from the body
SelectionSearchStrategy selectionStrategy = new SelectionSearchStrategy(
configRegistry);
PositionCoordinate[] cellsToSearch = selectionStrategy
.getSelectedCells(gridLayer.getBodyLayer().getSelectionLayer());
Assert.assertEquals(20, cellsToSearch.length);
}
@Test
public void shouldOnlySearchWhatIsSelected() {
gridLayer
.doCommand(new SelectCellCommand(gridLayer, 1, 4, false, true));
gridLayer
.doCommand(new SelectCellCommand(gridLayer, 2, 2, false, true));
gridLayer
.doCommand(new SelectCellCommand(gridLayer, 3, 4, false, true));
gridLayer
.doCommand(new SelectCellCommand(gridLayer, 5, 4, false, true));
gridLayer
.doCommand(new SelectCellCommand(gridLayer, 6, 2, false, true));
Assert.assertEquals(5, gridLayer.getBodyLayer().getSelectionLayer()
.getSelectedCellPositions().length);
final SelectionLayer selectionLayer = gridLayer.getBodyLayer()
.getSelectionLayer();
SelectionSearchStrategy selectionStrategy = new SelectionSearchStrategy(
configRegistry);
selectionStrategy
.setComparator(new CellValueAsStringComparator<Comparable<String>>());
selectionStrategy.setContextLayer(selectionLayer);
Assert.assertNull(selectionStrategy.executeSearch("[0,1]"));
Assert.assertNotNull(selectionStrategy.executeSearch(CELL_VALUE));
Assert.assertNotNull(selectionStrategy.executeSearch(CELL_VALUE));
Assert.assertNull(selectionStrategy.executeSearch("[5,0]"));
}
@Test
public void shouldSearchSelectionBackwards() {
// Select entire grid
gridLayer
.doCommand(new SelectCellCommand(gridLayer, 1, 1, false, false));
gridLayer.doCommand(new SelectAllCommand());
// Should find the first cell in grid
final SelectionLayer selectionLayer = gridLayer.getBodyLayer()
.getSelectionLayer();
SelectionSearchStrategy selectionStrategy = new SelectionSearchStrategy(
configRegistry);
selectionStrategy
.setComparator(new CellValueAsStringComparator<Comparable<String>>());
selectionStrategy.setContextLayer(selectionLayer);
PositionCoordinate positionCoordinate = selectionStrategy
.executeSearch(CELL_VALUE);
Assert.assertEquals(0, positionCoordinate.columnPosition);
Assert.assertEquals(0, positionCoordinate.rowPosition);
// Should find last cell
selectionStrategy = new SelectionSearchStrategy(configRegistry,
ISearchDirection.SEARCH_BACKWARDS, true);
selectionStrategy
.setComparator(new CellValueAsStringComparator<Comparable<String>>());
selectionStrategy.setContextLayer(selectionLayer);
positionCoordinate = selectionStrategy.executeSearch(CELL_VALUE);
Assert.assertEquals(9, positionCoordinate.columnPosition);
Assert.assertEquals(4, positionCoordinate.rowPosition);
}
}