package com.vaadin.tests.application;

import static org.junit.Assert.assertEquals;

import org.junit.Test;

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

public class PreserveWithExpiredHeartbeatTest extends SingleBrowserTest {
    @Test
    public void testNavigateBackAfterMissingHeartbeats()
            throws InterruptedException {
        final int heartbeatInterval = 5000;

        openTestURL();
        String originalId = getUiIdentification();

        long startTime = System.currentTimeMillis();

        while (System.currentTimeMillis() - startTime < heartbeatInterval
                * 3.1) {
            // "Close" the tab
            driver.get("about:blank");

            sleep(heartbeatInterval / 2);

            // "Reopen" tab
            openTestURL();

            // Verify that that we still get the same UI
            assertEquals("Original UI has been closed", originalId,
                    getUiIdentification());
        }
    }

    private String getUiIdentification() {
        return $(LabelElement.class).id("idLabel").getText();
    }
}