vaadin-framework/uitest/src/test/java/com/vaadin/tests/components/table/TableWithPollingTest.java

package com.vaadin.tests.components.table;

import static org.junit.Assert.assertEquals;

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

import com.vaadin.testbench.elements.TableElement;
import com.vaadin.tests.tb3.MultiBrowserTest;

public class TableWithPollingTest extends MultiBrowserTest {

    @Test
    public void testColumnResizing() throws Exception {
        openTestURL();

        int offset = -20;
        int headerCellWidth = getHeaderCell(0).getSize().width;
        int bodyCellWidth = getBodyCell(0).getSize().width;

        resizeColumn(0, offset);

        assertHeaderCellWidth(0, headerCellWidth + offset);
        assertBodyCellWidth(0, bodyCellWidth + offset);

        offset = 50;
        headerCellWidth = getHeaderCell(1).getSize().width;
        bodyCellWidth = getBodyCell(1).getSize().width;

        resizeColumn(1, offset);

        assertHeaderCellWidth(1, headerCellWidth + offset);
        assertBodyCellWidth(1, bodyCellWidth + offset);

    }

    private WebElement getHeaderCell(int column) {
        return $(TableElement.class).get(0).getHeaderCell(column);
    }

    private WebElement getBodyCell(int column) {
        return $(TableElement.class).get(0).getCell(0, column);
    }

    private WebElement getColumnResizer(int column) {
        return getHeaderCell(column)
                .findElement(By.className("v-table-resizer"));
    }

    private void resizeColumn(int column, int by) throws InterruptedException {
        new Actions(driver).clickAndHold(getColumnResizer(column))
                .moveByOffset(by, 0).perform();
        sleep(2000);
        new Actions(driver).release().perform();
    }

    private void assertHeaderCellWidth(int column, int width)
            throws AssertionError {
        assertEquals(width, getHeaderCell(column).getSize().width);
    }

    private void assertBodyCellWidth(int column, int width)
            throws AssertionError {
        assertEquals(width, getBodyCell(column).getSize().width);
    }

}