selenium之文件下载

Selenium 自动化测试：轻松搞定文件下载

在 Web 自动化测试中，经常会遇到需要验证文件下载功能的场景。例如，测试报告的导出、用户上传文件的下载、PDF 文档的生成与下载等等。Selenium 本身并没有直接处理文件下载的内置方法，但我们可以通过一些巧妙的配置和技巧来轻松实现文件下载的自动化测试。

本文将深入探讨在使用 Selenium 进行自动化测试时，如何处理文件下载，让你能够有效地验证 Web 应用程序的文件下载功能。

为什么 Selenium 本身不直接处理文件下载？

Selenium 的主要职责是驱动浏览器进行交互，模拟用户的行为。而文件的下载通常是由浏览器自身处理的，涉及到操作系统的文件系统。Selenium 无法直接干预浏览器底层的下载过程和文件保存行为。

处理文件下载的常见策略

虽然 Selenium 本身不直接处理，但我们可以通过配置浏览器行为或借助其他工具来间接实现文件下载的自动化测试。

配置浏览器设置，实现自动下载到指定目录

这是最常用且推荐的方法。通过修改浏览器的配置，我们可以指示浏览器在下载文件时，自动保存到我们预先指定的目录，而无需弹出“另存为”对话框。

可以看到，执行下面代码的时候；当前的目录会创建一个downloads文件夹；并且里面存放下载的文件

import time, os
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.service import Service


class browser_download:
    def __init__(self, path, options):
        self.path = path
        self.service = Service(path)
        self.options = options
        self.driver = webdriver.Chrome(service=self.service, options=self.options)

    def get(self, url):
        self.driver.get(url)
        self.driver.implicitly_wait(5)

    def download_file(self):
        xpath = f'//*[@id="content"]/div/a[13]'
        self.driver.find_element(By.XPATH, xpath).click()

    def close(self):
        time.sleep(5)
        self.driver.quit()


if __name__ == '__main__':
    serve_path = r'D:\Code_Study\driver\chromedriver-win64\chromedriver.exe'
    url = "https://the-internet.herokuapp.com/download"
    options = Options()
    # 设置下载路径
    download_dir = os.path.abspath("downloads")  # 指定下载目录
    perfs = {
        "download.default_directory": download_dir,
    }
    options.add_experimental_option("prefs", perfs)
    browser = browser_download(serve_path, options)
    browser.get(url)
    browser.download_file()
    browser.close()

自动化测试中文件下载的验证

一旦文件被下载到指定的目录，下一步就是验证下载是否成功以及文件的内容是否符合预期。常见的验证方法包括：

• 检查文件是否存在： 使用 os.path.exists() 方法判断文件是否已成功下载到指定目录。
• 检查文件大小： 使用 os.path.getsize() 方法获取文件大小，并与预期大小进行比较。
• 检查文件类型： 使用 os.path.splitext() 获取文件扩展名，判断文件类型是否正确。
• 读取文件内容并进行断言：
  • 对于文本文件（如 CSV、TXT），可以使用 Python 的文件操作读取内容并进行断言。
  • 对于 JSON 文件，可以使用 json 库解析并进行断言。
  • 对于 PDF 文件，可以使用第三方库（如 PyPDF2）读取文本内容或元数据进行断言。
  • 对于其他二进制文件，可能需要根据文件格式进行特定的解析和验证。

最佳实践和注意事项

• 指定稳定的下载目录： 确保下载目录在测试执行过程中是可写且不会被清理的。可以使用绝对路径，方便管理。
• 处理不同文件类型： 针对不同的文件类型，配置浏览器以正确处理，并在验证时使用相应的库进行解析。
• 等待文件下载完成： 在触发下载后，需要等待一段时间才能进行文件验证。可以使用显式等待，例如轮询检查文件是否存在或文件大小是否稳定。
• 清理下载目录： 在每个测试用例执行前后，清理下载目录，避免相互影响。
• 考虑并发测试： 如果需要进行并发测试，确保每个测试用例使用独立的下载目录，避免文件冲突。
• 处理下载失败的情况： 考虑网络问题、服务器错误等可能导致下载失败的情况，并在测试中进行适当的处理和断言。

总结

虽然 Selenium 本身不直接处理文件下载，但通过配置浏览器设置，我们可以实现将下载文件自动保存到指定目录，从而方便我们进行后续的验证。结合 Python 的文件操作和相关的第三方库，我们可以编写出可靠的自动化测试用例，有效地验证 Web 应用程序的文件下载功能。掌握这些技巧，将使你的自动化测试能力更上一层楼！