Selenium 设置浏览器下载 Firefox 和Chrome

2017-10-20 17:33 虫师阅读(13492) 评论(2) 编辑收藏举报

当我们在使用Selenium运行自动化测试时，偶尔需要用到下载功能，但浏览器的下载可能会弹出下载窗口，或者下载路径不是我们想要保存的位置，所以在通过Selenium启动浏览器时需要做相关的设置，将使这些设置在启动的浏览器中生效果。

下图为Firefox的下载弹窗：

Firefox 设置浏览器下载

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.firefox.FirefoxProfile;
import org.openqa.selenium.By;


public class FirefoxDown {

    public static void main(String[] args) {

        FirefoxProfile profile = new FirefoxProfile();
        profile.setPreference("browser.download.folderList", 2);
        profile.setPreference("browser.download.dir", "d:\\java");
        profile.setPreference("browser.helperApps.neverAsk.saveToDisk", "binary/octet-stream");
        WebDriver driver =new FirefoxDriver(profile);
        driver.get("https://pypi.Python.org/pypi/selenium");
        driver.findElement(By.partialLinkText("selenium-3.6.0.tar.gz")).click();
    }
}

先 new 一个FirefoxProfile（）类，通过setPreference 设置浏览器下载类型、路径等。

browser.download.folderList
设置成 0 代表下载到浏览器默认下载路径，设置成 2 则可以保存到指定目录。

browser.download.dir
用于指定所下载文件的目录。 os.getcwd() 函数不需要传递参数，用于返回当前的目录。

browser.helperApps.neverAsk.saveToDisk
指定要下载页面的 Content-type 值， “binary/octet-stream” 为文件的类型。下载的文件不同，这里的类型也会有所不一样。如果不清楚你下载的文件什么类型，请用Fiddler抓包。

Chrome 设置浏览器下载

import org.openqa.selenium.By;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.remote.CapabilityType;
import org.openqa.selenium.remote.DesiredCapabilities;

import java.util.HashMap;


public class ChromeDown {

    public static void main(String[] args) throws InterruptedException {

        String downloadFilepath = "D:\\java";
        HashMap<String, Object> chromePrefs = new HashMap<String, Object>();
        chromePrefs.put("profile.default_content_settings.popups", 0);
        chromePrefs.put("download.default_directory", downloadFilepath);
        ChromeOptions options = new ChromeOptions();
        HashMap<String, Object> chromeOptionsMap = new HashMap<String, Object>();
        options.setExperimentalOption("prefs",chromePrefs);
        options.addArguments("--test-type");
        DesiredCapabilities cap = DesiredCapabilities.chrome();
        cap.setCapability(ChromeOptions.CAPABILITY, chromeOptionsMap);
        cap.setCapability(CapabilityType.ACCEPT_SSL_CERTS, true);
        cap.setCapability(ChromeOptions.CAPABILITY, options);
        
        WebDriver driver = new ChromeDriver(cap);

        driver.get("https://www.baidu.com");
        driver.findElement(By.id("kw")).sendKeys("chrome");
        driver.findElement(By.id("su")).click();
        Thread.sleep(2000);
        driver.findElement(By.linkText("普通下载")).click();
    }
}