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

package com.vaadin.tests.components.ui;

import static org.junit.Assert.assertEquals;

import java.net.URI;

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

import com.vaadin.testbench.By;
import com.vaadin.tests.tb3.MultiBrowserTest;

public class PushStateAndReplaceStateTest extends MultiBrowserTest {

    @Test
    public void testUriFragment() throws Exception {
        driver.get(getTestUrl());
        assertUri(getTestUrl());

        hitButton("test");

        assertUri(getTestUrl() + "/test");

        driver.navigate().back();

        driver.findElement(By.className("v-Notification")).getText()
                .contains("Popstate event");

        assertUri(getTestUrl());

        hitButton("test");
        URI base = new URI(getTestUrl() + "/test");
        hitButton("X");
        URI current = base.resolve("X");
        driver.findElement(By.xpath("//*[@id = 'replace']/input")).click();
        hitButton("root_X");
        current = current.resolve("/X");

        assertUri(current.toString());

        // Now that last change was with replace state, two back calls should go
        // to initial
        driver.navigate().back();
        driver.navigate().back();

        assertUri(getTestUrl());

    }

    private void assertUri(String uri) {
        final String expectedText = "Current Location: " + uri;
        waitUntil(new ExpectedCondition<Boolean>() {

            @Override
            public Boolean apply(WebDriver input) {
                return expectedText.equals(getLocationLabelValue());
            }
        });

        assertEquals(uri, driver.getCurrentUrl());
    }

    private String getLocationLabelValue() {
        String text = vaadinElementById("locationLabel").getText();
        return text;
    }

}