vaadin-framework/uitest/src/test/java/com/vaadin/tests/components/ui/PollListeningTest.java

package com.vaadin.tests.components.ui;

import org.junit.Test;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.support.ui.ExpectedCondition;

import com.vaadin.tests.tb3.MultiBrowserTest;

public class PollListeningTest extends MultiBrowserTest {

    @Test
    public void testReceivePollEvent() {
        openTestURL();
        waitUntilPollEventReceived();
    }

    private void waitUntilPollEventReceived() {
        waitUntil(new ExpectedCondition<Boolean>() {
            private String expected = "PollEvent received";

            @Override
            public Boolean apply(WebDriver arg0) {
                return driver.getPageSource().contains(expected);
            }

            @Override
            public String toString() {
                return String.format("page to contain text '%s'", expected);
            }
        });
    }
}