UI自动化测试判断文件是否下载成功

先设置一下Chrome 的下载路径

CHROME_OPTIONS = webdriver.ChromeOptions()
CHROME_OPTIONS.add_experimental_option('prefs', {
    'download.default_directory': download_dir,  # 设置下载路径，如果指定的路径不存在，Chrome 会自动创建该路径
    'download.prompt_for_download': False,  # 控制是否弹出下载提示框进行确认，默认设置为 False，即不弹出下载提示框。
    'safebrowsing.enabled': False,  # 控制是否提示安全警告，默认设置为 False，即不提示安全警告。
    'download.directory_upgrade': False  # 记录下载目录是否被更改的状态，默认设置为 False。
})

再把设置好的路径当做参数传入进去

driver = webdriver.Chrome(options=CHROME_OPTIONS)

写一个检查检查文件是否上传成功的方法

def check_download_file(f, load=1):
    """
    检查文件是否下载成功
    f: 文件名称
    load: 等待时间
    """
    try:
        # 判断下载目录是否存在
        if os.path.exists(download_dir):
            # 等待文件下载的时间
            time.sleep(int(load))
            # 如果下载目录中存在文件，返回True否则返回False
            return any(f in file for file in os.listdir(download_dir))
    except Exception as e:
        logger.info("The file is not exist! Please check the code!")
        logger.error(str(e))
    return False

最后再断言中调用改方法即可

assert check_download_file("测试任务列表", load=5)

# 自动化测试笔记