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

/*
 * Copyright 2000-2016 Vaadin Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.vaadin.tests.components.grid;

import static org.junit.Assert.assertFalse;

import java.util.List;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import com.vaadin.testbench.By;
import com.vaadin.testbench.elements.ButtonElement;
import com.vaadin.testbench.elements.GridElement;
import com.vaadin.testbench.elements.GridElement.GridCellElement;
import com.vaadin.testbench.elements.TextFieldElement;
import com.vaadin.tests.tb3.SingleBrowserTest;

public class GridHeaderFooterComponentsTest extends SingleBrowserTest {

    @Before
    public void setUp() {
        setDebug(true);

        openTestURL();
    }

    @Test
    public void hideAndShowComponentsInHeader() {
        GridElement grid = $(LegacyGridElement.class).first();

        int filterRow = 2;
        Assert.assertNull(getHeaderElement(grid, filterRow, 1));
        Assert.assertNotNull(getHeaderElement(grid, filterRow, 2));
        Assert.assertNotNull(getHeaderElement(grid, filterRow, 3));

        // Show (1,2)
        grid.getHeaderCell(1, 1).$(ButtonElement.class).first().click();

        TextFieldElement textfield = getHeaderElement(grid, filterRow, 1);
        Assert.assertNotNull(textfield);
        Assert.assertEquals("Filter: string", textfield.getValue());

        textfield.setValue("foo");
        Assert.assertEquals("1. value change for field in string to foo",
                getLogRow(0));

        assertNoErrorNotifications();
    }

    private TextFieldElement getHeaderElement(GridElement grid, int row,
            int col) {
        GridCellElement cell = grid.getHeaderCell(row, col);
        List<TextFieldElement> all = cell.$(TextFieldElement.class).all();
        if (all.size() == 0) {
            return null;
        } else if (all.size() == 1) {
            return all.get(0);
        } else {
            throw new RuntimeException(
                    "Multiple elements found in the header cell at " + row + ","
                            + col);
        }
    }

    @Test
    public void hideAndShowComponentsInFooter() {
        GridElement grid = $(LegacyGridElement.class).first();

        int filterRow = 0;
        Assert.assertNull(getFooterElement(grid, filterRow, 1));
        Assert.assertNotNull(getFooterElement(grid, filterRow, 2));
        Assert.assertNotNull(getFooterElement(grid, filterRow, 3));

        // Show (1,2)
        grid.getFooterCell(1, 1).$(ButtonElement.class).first().click();

        TextFieldElement textfield = getFooterElement(grid, filterRow, 1);
        Assert.assertNotNull(textfield);
        Assert.assertEquals("Filter: string", textfield.getValue());

        textfield.setValue("foo");
        Assert.assertEquals("1. value change for field in string to foo",
                getLogRow(0));

        assertNoErrorNotifications();
    }

    private TextFieldElement getFooterElement(GridElement grid, int row,
            int col) {
        GridCellElement cell = grid.getFooterCell(row, col);
        List<TextFieldElement> all = cell.$(TextFieldElement.class).all();
        if (all.size() == 0) {
            return null;
        } else if (all.size() == 1) {
            return all.get(0);
        } else {
            throw new RuntimeException(
                    "Multiple elements found in the footer cell at " + row + ","
                            + col);
        }
    }

    @Test
    public void testRemoveAllHeadersAndFooters() {
        openTestURL();

        for (int i = 2; i >= 0; --i) {
            // Remove Header
            $(LegacyGridElement.class).first().getHeaderCell(i, 0)
                    .$(ButtonElement.class).first().click();
            assertFalse("Header " + i + " should not be present.",
                    $(LegacyGridElement.class).first()
                            .isElementPresent(By.vaadin("#header[" + i + "]")));

            // Remove Footer
            $(LegacyGridElement.class).first().getFooterCell(i, 0)
                    .$(ButtonElement.class).first().click();
            assertFalse("Footer " + i + " should not be present.",
                    $(LegacyGridElement.class).first()
                            .isElementPresent(By.vaadin("#footer[" + i + "]")));
        }

        assertNoErrorNotifications();
    }
}