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

package com.vaadin.tests.components.grid;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

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

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

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.number.IsCloseTo.closeTo;
import static org.junit.Assert.fail;

/**
 * Tests that Grid gets correct height based on height mode, and resizes
 * properly with details row if height is undefined.
 *
 * @author Vaadin Ltd
 */
@TestCategory("grid")
public class GridHeightTest extends MultiBrowserTest {

    @Override
    public void setup() throws Exception {
        super.setup();
        openTestURL();
        waitForElementPresent(By.className("v-grid"));
    }

    @Test
    @Ignore
    public void testGridHeightAndResizingUndefined()
            throws InterruptedException {
        assertNoErrors(testGridHeightAndResizing(GridHeight.UNDEFINED));
    }

    @Test
    @Ignore
    public void testGridHeightAndResizingRow() throws InterruptedException {
        assertNoErrors(testGridHeightAndResizing(GridHeight.ROW3));
    }

    @Test
    public void testGridHeightAndResizingFull() throws InterruptedException {
        assertNoErrors(testGridHeightAndResizing(GridHeight.FULL));
    }

    private Map<AssertionError, Object[]> testGridHeightAndResizing(
            Object gridHeight) throws InterruptedException {
        Map<AssertionError, Object[]> errors = new HashMap<>();
        String caption;
        if (GridHeight.ROW3.equals(gridHeight)) {
            caption = gridHeight + " rows";
        } else {
            caption = (String) gridHeight;
        }
        $(RadioButtonGroupElement.class).id("gridHeightSelector")
                .selectByText(caption);
        for (String gridWidth : GridHeight.gridWidths) {
            $(RadioButtonGroupElement.class).id("gridWidthSelector")
                    .selectByText(gridWidth);
            for (String detailsRowHeight : GridHeight.detailsRowHeights) {
                $(RadioButtonGroupElement.class).id("detailsHeightSelector")
                        .selectByText(detailsRowHeight);
                sleep(500);

                GridElement grid = $(GridElement.class).first();
                int initialHeight = grid.getSize().getHeight();
                try {
                    // check default height
                    assertGridHeight(getExpectedInitialHeight(gridHeight),
                            initialHeight);
                } catch (AssertionError e) {
                    errors.put(e, new Object[] { gridHeight, gridWidth,
                            detailsRowHeight, "initial" });
                    fail();
                }

                GridRowElement row = grid.getRow(2);
                row.click(getXOffset(row, 5), getYOffset(row, 5));
                waitForElementPresent(By.id("lbl1"));

                int openHeight = grid.getSize().getHeight();
                try {
                    // check height with details row opened
                    assertGridHeight(getExpectedOpenedHeight(gridHeight,
                            detailsRowHeight), openHeight);
                } catch (AssertionError e) {
                    errors.put(e, new Object[] { gridHeight, gridWidth,
                            detailsRowHeight, "opened" });
                }

                row.click(5, 5);
                waitForElementNotPresent(By.id("lbl1"));

                int afterHeight = grid.getSize().getHeight();
                try {
                    // check height with details row closed again
                    assertThat("Unexpected Grid Height", afterHeight,
                            is(initialHeight));
                } catch (AssertionError e) {
                    errors.put(e, new Object[] { gridHeight, gridWidth,
                            detailsRowHeight, "closed" });
                }
            }
        }
        return errors;
    }

    private void assertNoErrors(Map<AssertionError, Object[]> errors) {
        if (!errors.isEmpty()) {
            StringBuilder sb = new StringBuilder("Exceptions: ");
            for (Entry<AssertionError, Object[]> entry : errors.entrySet()) {
                sb.append("\n");
                for (Object value : entry.getValue()) {
                    sb.append(value);
                    sb.append(" - ");
                }
                sb.append(entry.getKey().getMessage());
            }
            fail(sb.toString());
        }
    }

    private int getExpectedInitialHeight(Object gridHeight) {
        int result = 0;
        if (GridHeight.UNDEFINED.equals(gridHeight)
                || GridHeight.ROW3.equals(gridHeight)) {
            result = 81;
        } else if (GridHeight.FULL.equals(gridHeight)) {
            // pre-existing issue
            result = 400;
        }
        return result;
    }

    private int getExpectedOpenedHeight(Object gridHeight,
            Object detailsRowHeight) {
        int result = 0;
        if (GridHeight.UNDEFINED.equals(gridHeight)) {
            if (GridHeight.PX100.equals(detailsRowHeight)) {
                result = 182;
            } else if (GridHeight.FULL.equals(detailsRowHeight)) {
                result = 131;
            } else if (GridHeight.UNDEFINED.equals(detailsRowHeight)) {
                result = 100;
            }
        } else if (GridHeight.ROW3.equals(gridHeight)
                || GridHeight.FULL.equals(gridHeight)) {
            result = getExpectedInitialHeight(gridHeight);
        }
        return result;
    }

    private void assertGridHeight(int expected, int actual) {
        assertThat("Unexpected Grid Height", (double) actual,
                closeTo(expected, 1));
    }
}