Selenium 设置浏览器下载 Firefox 和Chrome

当我们在使用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();

    }

}

相比较Firefox来说，Chrome的下载默认不会弹出下载窗口的，我们主要是想修改默认的默认下载路径。

Chrome的设置看上去要比Firefox复杂一次，不过，你需要关注两个设置。

profile.default_content_settings.popups 0 设置为禁止弹出下载窗口

download.default_directory 设置为文件下载路径