Selenium验证文件是否导出成功

一：前言

在自动化测试中，使用Selenium和pytest来进行测试页面的导出功能。
本文是对文件是否能导出成功的判断。
因为项目性质，所以不贴图，仅提供代码，通过这个思路，你可以掌握如何判断的方法。

二：操作

①设置文件下载路径。
在config.py中导入os包。
import os

设置下载路径。
DOWNLOAD_PATH = os.path.join(BASE_DIR, 'resources\\file')

设置模拟器中谷歌浏览器的驱动的下载路径。

CHROME_OPTIONS.add_experimental_option('prefs', {
    'download.default_directory': DOWNLOAD_PATH,     # 设置下载路径，路径不存在会自动创建
    'download.prompt_for_download': False,           # 是否弹窗询问
    'safebrowsing.enabled': False,                   # 是否提示安全警告
    # Boolean that records if the download directory was changed by an
    # upgrade a unsafe location to a safe location.
    'download.directory_upgrade': False              # 记录下载目录是否被更改？
})

②写检验方法
在common_utils.py中写
检验文件是否下载的：（f为文件名，load为等待时间）

def check_download_file(f, load=1):
	try:
		if os.path.exists(DOWNLOAD_PATH):
			time.sleep(int(load))
			for file in os.listdir(DOWNLOAD_PATH):
				if f in file:
					return True
				return False
	except Exception:
		logging.info("The file is not exist！Please check the code！")

每次都先删除一下文件：

def clear_download_file():
	for file in os.listdir(DOWNLOAD_PATH):
		if len(file) > 0:
			os.remove(DOWNLOAD_PATH  "\\"   file)

③最后在主方法中调用。

clear_download()
assert check_download(f="人员信息列表", load='3')

三：尾言

总结一下，思路为，设置下载路径，写try/catch方法，最后在方法中调用。

这篇好文章是转载于：学新通技术网