vaadin-framework/uitest/src/com/vaadin/tests/application/ReconnectDialogThemeTest.java

/*
 * Copyright 2000-2014 Vaadin Ltd.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.vaadin.tests.application;

import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.IOException;

import javax.imageio.ImageIO;

import org.junit.Assert;
import org.junit.Test;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.ui.ExpectedCondition;

import com.vaadin.testbench.elements.ButtonElement;
import com.vaadin.testbench.parallel.BrowserUtil;
import com.vaadin.testbench.parallel.TestCategory;
import com.vaadin.tests.tb3.CustomTestBenchCommandExecutor;
import com.vaadin.tests.tb3.MultiBrowserThemeTestWithProxy;

@TestCategory("")
public class ReconnectDialogThemeTest extends MultiBrowserThemeTestWithProxy {

    static By reconnectDialogBy = By.className("v-reconnect-dialog");

    @Test
    public void reconnectDialogTheme() throws IOException {
        openTestURL();
        ButtonElement helloButton = $(ButtonElement.class).caption("Say hello")
                .first();
        helloButton.click();
        Assert.assertEquals("1. Hello from the server", getLogRow(0));
        disconnectProxy();
        helloButton.click();
        testBench().disableWaitForVaadin();
        waitUntil(new ExpectedCondition<Boolean>() {

            @Override
            public Boolean apply(WebDriver input) {
                boolean present = isElementPresent(reconnectDialogBy);
                return present;
            }
        });

        WebElement dialog = findElement(reconnectDialogBy);
        WebElement spinner = dialog.findElement(By.className("spinner"));

        // Hide spinner to make screenshot stable
        executeScript("arguments[0].style.visibility='hidden';", spinner);
        compareScreen("onscreen-without-spinner");

        // Show spinner and make sure it is shown by comparing to the screenshot
        // without a spinner
        executeScript("arguments[0].style.visibility='visible';", spinner);
        BufferedImage fullScreen = ImageIO.read(new ByteArrayInputStream(
                ((TakesScreenshot) getDriver())
                        .getScreenshotAs(OutputType.BYTES)));
        BufferedImage spinnerImage = CustomTestBenchCommandExecutor
                .cropToElement(spinner, fullScreen,
                        BrowserUtil.isIE8(getDesiredCapabilities()));
        assertHasManyColors("Spinner is not shown", spinnerImage);

    }

    private void assertHasManyColors(String message, BufferedImage spinnerImage) {
        int backgroundColor = spinnerImage.getRGB(0, 0);
        for (int x = 0; x < spinnerImage.getWidth(); x++) {
            for (int y = 0; y < spinnerImage.getHeight(); y++) {
                if (Math.abs(spinnerImage.getRGB(x, y) - backgroundColor) > 50) {
                    return;
                }
            }
        }
        Assert.fail(message);

    }

    @Override
    protected Class<?> getUIClass() {
        return ReconnectDialogUI.class;
    }

}