org.jspringbot.keyword.selenium.web.DownloadShouldExists Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jspringbot-selenium Show documentation

There is a newer version: 1.9

package org.jspringbot.keyword.selenium.web;

import java.io.File;

import org.jspringbot.KeywordInfo;
import org.jspringbot.keyword.selenium.AbstractSeleniumKeyword;
import org.jspringbot.keyword.selenium.FirefoxProfileBean;
import org.jspringbot.keyword.selenium.WebDriverWaitBean;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

@Component
@KeywordInfo(
        name = "Download Should Exists",
        parameters = {"filename"},
        description = "classpath:desc/DownloadShouldExists.txt"
)
public class DownloadShouldExists extends AbstractSeleniumKeyword implements ApplicationContextAware {

    private FirefoxProfileBean profileBean;

    private WebDriver driver;

    private WebDriverWaitBean waitBean;

    @Override
    public Object execute(Object[] params) throws Exception {
        if(profileBean == null) {
            throw new IllegalStateException("Keyword not supported, FirefoxProfileBean not defined.");
        }

        String filename = String.valueOf(params[0]);
        final File downloadDir = profileBean.getDownloadDir();
        final File download = new File(downloadDir, filename);

        (new WebDriverWait(driver, waitBean.getDownloadTimeoutInSeconds())).until(new ExpectedCondition() {
            public Boolean apply(WebDriver d) {
                return download.isFile();
            }
        });

        if(!download.isFile()) {
            throw new IllegalArgumentException(String.format("Download file '%s' not found.", filename));
        }

        return null;
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        try {
            profileBean = applicationContext.getBean(FirefoxProfileBean.class);
            driver = applicationContext.getBean(WebDriver.class);
            waitBean = applicationContext.getBean(WebDriverWaitBean.class);
        } catch(BeansException ignore) {
        }
    }
}