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 Serviceclass browser_download:def __init__(self, path, options):self.path = pathself.service = Service(path)self.options = optionsself.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 應用程序的文件下載功能。掌握這些技巧，將使你的自動化測試能力更上一層樓！