vaadin-framework/uitest/src/test/java/com/vaadin/tests/components/grid/GridDefaultSelectionModeTest.java

package com.vaadin.tests.components.grid;

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

import org.junit.Test;

import com.vaadin.testbench.elements.GridElement;
import com.vaadin.testbench.elements.ButtonElement;
import com.vaadin.testbench.elements.GridElement.GridCellElement;
import com.vaadin.tests.tb3.MultiBrowserTest;

public class GridDefaultSelectionModeTest extends MultiBrowserTest {

    @Test
    public void testSelectionFromServer() {
        setDebug(true);
        openTestURL();

        $(ButtonElement.class).caption("Select on server").first().click();

        assertTrue("Row should be selected.",
                $(GridElement.class).first().getRow(0).isSelected());

        $(ButtonElement.class).caption("Deselect on server").first().click();

        assertFalse("Row should not be selected.",
                $(GridElement.class).first().getRow(0).isSelected());

        assertNoErrorNotifications();
    }

    @Test
    public void testSelectionWithSort() {
        setDebug(true);
        openTestURL();

        GridElement grid = $(GridElement.class).first();
        grid.getCell(0, 0).click();

        GridCellElement header = grid.getHeaderCell(0, 1);
        header.click();
        header.click();

        assertTrue("Row should be selected.", grid.getRow(1).isSelected());

        assertNoErrorNotifications();
    }

    @Test
    public void testReselectDeselectedRow() {
        setDebug(true);
        openTestURL();

        $(ButtonElement.class).caption("Select on server").first().click();

        GridElement grid = $(GridElement.class).first();
        assertTrue("Row should be selected.", grid.getRow(0).isSelected());

        $(ButtonElement.class).caption("Deselect on server").first().click();

        assertFalse("Row should not be selected.", grid.getRow(0).isSelected());

        grid.getCell(0, 0).click();
        assertTrue("Row should be selected.", grid.getRow(0).isSelected());

        assertNoErrorNotifications();
    }
}