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

package com.vaadin.tests.components.grid;

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

import org.junit.Test;
import org.openqa.selenium.By;

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

@TestCategory("grid")
public class GridHeaderFormatChangeTest extends MultiBrowserTest {

    @Test
    public void testHeaderRetainsSelectAllForColumnRemoval() {
        openTestURL();
        GridElement grid = $(GridElement.class).first();

        // Assert that we do not have the select all checkbox
        assertTrue("Found input in header even though none should exist.",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());

        // Set grid into multiselection mode
        toggleSelectionMode();

        // Assert that we now have a select all checkbox in the header
        assertFalse("Expected one input field in header",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());

        // Hide the firstName column from the grid.
        toggleFirstName();

        // Assert that we still have the select all checkbox in the header.
        assertFalse("Header was missing checkbox after hiding column",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());

        // Show the firstName column.
        toggleFirstName();

        // Assert that we still have the select all checkbox in the header.
        assertFalse("Header was missing checkbox after bringing back column",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());
    }

    @Test
    public void testHeaderRetainsSelectAllForJoinColumnAdd() {
        openTestURL();
        GridElement grid = $(GridElement.class).first();

        // Assert that we do not have the select all checkbox
        assertTrue("Found input in header even though none should exist.",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());

        // Set grid into multiselection mode
        toggleSelectionMode();

        // Assert that we now have a select all checkbox in the header
        assertFalse("Expected one input field in header",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());

        // Add Join columns header
        toggleJoin();

        // Assert that we still have the select all checkbox in the header.
        assertFalse("Header was missing checkbox after hiding column",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());

        // remove Join Columns header
        toggleJoin();

        // Assert that we still have the select all checkbox in the header.
        assertFalse("Header was missing checkbox after bringing back column",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());
    }

    @Test
    public void selectAllShouldKeepState() {
        openTestURL();
        GridElement grid = $(GridElement.class).first();

        // Assert that we do not have the select all checkbox
        assertTrue("Found input in header even though none should exist.",
                grid.getHeader().findElements(By.tagName("input")).isEmpty());

        // Set grid into multiselection mode
        toggleSelectionMode();

        // Assert that we now have a select all checkbox in the header
        assertFalse("Should not be selected after adding",
                grid.getHeader().findElement(By.tagName("input")).isSelected());

        grid.getHeader().findElement(By.tagName("input")).click();

        // Assert that checkbox is checked
        assertSelectAllChecked(
                "Not selected even though we just clicked selection", grid);

        // Hide the firstName column from the grid.
        toggleFirstName();

        // Assert that checkbox is still checked
        assertSelectAllChecked("Selection disappeared after removing column",
                grid);

        // Show the firstName column.
        toggleFirstName();

        // Assert that checkbox is still checked
        assertSelectAllChecked("Selection disappeared after adding column",
                grid);

    }

    private void assertSelectAllChecked(String message, GridElement grid) {
        assertTrue(message,
                grid.getHeader().findElement(By.tagName("input")).isSelected());
    }

    private void toggleSelectionMode() {
        $(ButtonElement.class).id("selection_mode").click();
    }

    private void toggleFirstName() {
        $(ButtonElement.class).id("show_hide").click();
    }

    private void toggleJoin() {
        $(ButtonElement.class).id("join").click();
    }
}