Protractor端到端测试用例中媒体文件下载实现方案问询

嘿，你的实现思路方向是对的，但还有几个关键细节可以完善，让媒体文件下载更稳定可靠。咱们一步步拆解来看：

一、当前Chrome配置的合理性 #

你设置的capabilities配置是完全合理的：

• prompt_for_download: false 避免了下载弹窗打断自动化流程
• 指定绝对路径的default_directory能确保文件下载到预期位置（注意：这个目录必须提前手动创建，Chrome不会自动生成目录，否则下载会失败）
• --no-sandbox参数在Linux环境或者CI/CD中很实用，能避免Chrome启动权限问题

小建议：新版本Chrome中--test-type=browser参数已经逐渐废弃，可以换成--disable-infobars来屏蔽浏览器的自动化提示条，让测试更稳定。

二、自定义download函数的完善 #

你写的函数框架没问题，但需要补全核心逻辑才能正常工作。完整的媒体文件下载函数应该包含以下步骤：

function downloadMedia(mediaUrl, saveFilename) {
    // 创建a标签元素
    const downloadLink = document.createElement('a');
    // 设置要下载的媒体文件URL
    downloadLink.href = mediaUrl;
    // 指定下载后的文件名（可选，不设置则用原文件名）
    downloadLink.download = saveFilename || 'media-file';
    // 把a标签加入DOM（部分浏览器需要元素在DOM中才能触发点击）
    document.body.appendChild(downloadLink);
    // 模拟用户点击下载
    downloadLink.click();
    // 移除临时创建的a标签
    document.body.removeChild(downloadLink);
}

调用时记得传入实际参数：

await browser.executeScript('downloadMedia(arguments[0], arguments[1])', 'https://example.com/test-video.mp4', 'my-test-video.mp4');

三、关键补充：等待下载完成 #

Protractor是异步执行的，调用完下载函数后不会自动等待文件下载完成，必须手动添加等待逻辑，否则后续的文件断言会失败。可以写一个辅助函数来检查文件是否存在且完全写入：

const fs = require('fs');
const path = require('path');

async function waitForDownloadComplete(downloadDir, targetFilename, timeout = 30000) {
    const startTime = Date.now();
    while (Date.now() - startTime < timeout) {
        try {
            const filePath = path.join(downloadDir, targetFilename);
            // 先检查文件是否存在
            const fileStats = fs.statSync(filePath);
            // 等待500ms后再次检查文件大小，确认文件已完全写入（避免文件还在传输中）
            await browser.sleep(500);
            const updatedStats = fs.statSync(filePath);
            if (fileStats.size === updatedStats.size) {
                return true;
            }
        } catch (err) {
            // 文件还没生成，继续等待
            await browser.sleep(1000);
        }
    }
    throw new Error(`下载超时：未找到文件 ${targetFilename}`);
}

在测试中这样使用：

// 触发下载
await browser.executeScript('downloadMedia(arguments[0], arguments[1])', mediaUrl, 'my-test-video.mp4');
// 等待下载完成
await waitForDownloadComplete('C:/absolute/path/to/tmp', 'my-test-video.mp4');
// 后续可以做文件断言，比如检查文件大小、是否存在等

四、特殊场景处理：跨域媒体文件 #

如果要下载的媒体文件和当前测试页面不同源，a标签的download属性可能会失效（浏览器会直接打开媒体文件而不是下载）。这种情况可以换一种实现方式：
直接让浏览器访问媒体文件URL，结合之前的Chrome配置，浏览器会自动下载到指定目录：

await browser.driver.get(mediaUrl);
await waitForDownloadComplete('C:/absolute/path/to/tmp', 'expected-filename.mp4');

内容的提问来源于stack exchange，提问作者AnumShk