使用 Selenium WebDriver for Firefox 下载 pdf

Question

我正在尝试将 .pdf 下载到我的本地，以便我可以使用 Apache PDFBox 从中读取文本并将其验证为我的测试套件的一部分。我已经找到了一些通过点击 URL 从 Firefox 下载 pdf 的代码。这对我不起作用，因为我正在使用的 pdf 是机密文档，因此它不会通过 URL 公开，而是作为弹出窗口加载到 PDF 查看器中。在浏览器中加载 PDF 查看器后，有谁知道如何点击 Firefox PDF 查看器中的下载按钮？

我尝试通过元素的 id which = "download":

(new WebDriverWait(driver, 10)).until(ExpectedConditions.presenceOfElementLocated(By.id("download")));
driver.findElement(By.id("download")).click(); 

不幸的是，这不起作用，因为它说它找不到元素。有人知道解决方法吗？

更新：我描述的弹出窗口是一个 iframe 元素。这导致无法找到“下载”元素。使用@4M01 的 switchTo() 答案修复。

Answer 1

正如你所说，

而是作为弹出窗口加载到 PDF 查看器中

你需要使用驱动对象的switchTo()方法来处理不同窗口之间的切换。

下面的代码对我来说工作正常，没有问题，我可以点击下载图标。

public class FirefoxPDFTest {
      WebDriver driver;

    @BeforeClass
    void Setup(){
        System.setProperty("webdriver.gecko.driver", "C:\\Automation\\Selenium\\drivers\\geckodriver.exe");
        driver = new FirefoxDriver();
        driver.manage().window().maximize();
    }

    @Test
    void downloadPDF(){
        driver.get("http://www.pdf995.com/samples/pdf.pdf");
        waitTillPageLoad();
        driver.findElement(By.id("download")).click();
    }



    private void waitTillPageLoad(){
        new WebDriverWait(driver, 30).until(driver -> ((JavascriptExecutor) driver).executeScript("return document.readyState").equals("complete"));
    }


    @AfterClass
    void tearDown(){
        driver.close();
        driver.quit();
    }

}

Answer 2

只需使用以下代码点击下载按钮：

    driver.findElement(By.xpath("//button[@id='download']")).click();

    Thread.sleep(8000);

    Robot robot = new Robot();

    robot.keyPress(KeyEvent.VK_ENTER);
    robot.keyRelease(KeyEvent.VK_ENTER);

Answer 3

我们可以使用 Firefox 浏览器设置和使用 WebDriver 的 Firefox Profile 设置来处理 Firefox 浏览器中的下载弹出窗口。

第 1 步：在 Firefox 浏览器中更新设置。

打开 Firefox 浏览器并导航到工具 -> 选项 导航到应用程序。 将 PDF 的操作类型设置为“保存文件”。

第 2 步：使用 FirefoxProfile 初始化 FireFoxDriver

File downloadsDir = new File("");

// Set Preferences for FirefoxProfile.
FirefoxProfile profile = new FirefoxProfile();
profile.setPreference("browser.download.folderList", 2);
profile.setPreference("browser.download.dir", downloadsDir.getAbsolutePath());
profile.setPreference("browser.download.manager.alertOnEXEOpen", false);
profile.setPreference("browser.helperApps.neverAsk.saveToDisk",
      "application/msword, application/csv, application/ris, text/csv, image/png, application/pdf, text/html, text/plain, application/zip, application/x-zip, application/x-zip-compressed, application/download, application/octet-stream");
profile.setPreference("browser.download.manager.showWhenStarting", false);
profile.setPreference("browser.download.manager.focusWhenStarting", false);
profile.setPreference("browser.download.useDownloadDir", true);
profile.setPreference("browser.helperApps.alwaysAsk.force", false);
profile.setPreference("browser.download.manager.alertOnEXEOpen", false);
profile.setPreference("browser.download.manager.closeWhenDone", true);
profile.setPreference("browser.download.manager.showAlertOnComplete", false);
profile.setPreference("browser.download.manager.useWindow", false);
profile.setPreference("services.sync.prefs.sync.browser.download.manager.showWhenStarting", false);
profile.setPreference("pdfjs.disabled", true);

// Initialize the FireFoxDriver instance.
FirefoxDriver webDriver = new FirefoxDriver(profile);

第 3 步：执行脚本

执行点击下载 PDF 图标的脚本。

结果：将下载 PDF 文件，并且不会显示“下载”弹出窗口。

Answer 4

您可以使用以下 (C#) 代码处理下载图标（使用 Firefox）：

IWebElement element = Driver.FindElement(By.Id("download"));
IJavaScriptExecutor executor = (IJavaScriptExecutor)Driver;
executor.ExecuteScript("arguments[0].click();", element);