vaadin-framework/uitest/src/test/java/com/vaadin/tests/components/gridlayout/GridLayoutExpandWithManyRowsTest.java

package com.vaadin.tests.components.gridlayout;

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

import java.util.List;

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

import com.vaadin.testbench.elements.GridLayoutElement;
import com.vaadin.tests.tb3.SingleBrowserTest;

public class GridLayoutExpandWithManyRowsTest extends SingleBrowserTest {

    @Test
    public void equalRowHeights() {
        openTestURL();
        GridLayoutElement gridlayout = $(GridLayoutElement.class).first();

        // Rows are expanded using integer pixels and leftover pixels are added
        // to the first N rows.
        // The tests uses rowspan=2 so one row in the DOM should be max 2 pixels
        // lower than the first row
        List<WebElement> slots = gridlayout
                .findElements(By.className("v-gridlayout-slot"));
        assertEquals(GridLayoutExpandWithManyRows.POPULATED_ROWS, slots.size());

        int firstRowHeight = slots.get(0).getSize().height;
        int lastRowHeight = firstRowHeight;
        for (int i = 1; i < GridLayoutExpandWithManyRows.POPULATED_ROWS; i++) {
            int rowHeight = slots.get(i).getSize().height;
            assertTrue(rowHeight <= firstRowHeight);
            assertTrue(rowHeight >= firstRowHeight - 2);
            assertTrue(rowHeight <= lastRowHeight);

            lastRowHeight = rowHeight;
        }
    }
}