Apifox 自动化测试实战：如何优雅实现 Token 自动获取与刷新

1次阅读

没有评论

共计 2434 个字符，预计需要花费 7 分钟才能阅读完成。

在接口测试中，Token 的管理常常让人头疼。手动管理 Token 会带来几个明显的问题：

Token 过期问题 ：测试过程中 Token 突然失效，需要中断测试重新获取
多环境切换困难 ：不同环境（开发、测试、生产）需要不同的 Token，手动切换容易出错
维护成本高 ：当 Token 变更时，需要修改所有相关测试用例
安全性风险 ：Token 直接写死在脚本或配置文件中存在泄露风险

这些痛点使得我们需要一个自动化的 Token 管理方案。

Apifox 提供了强大的脚本能力和环境变量管理功能，可以完美解决上述问题。与传统工具（如 Postman）相比，Apifox 的优势在于：

内置 JavaScript 执行环境 ：可以直接编写脚本处理 Token
完善的环境变量系统 ：支持全局、环境级和临时变量
脚本执行时机灵活 ：可以在请求前、请求后、集合运行前后执行脚本
变量作用域清晰 ：不同层级的变量互不干扰

在 Apifox 中，我们可以利用 ” 前置脚本 ” 功能，在发送请求前自动获取 Token。以下是实现步骤：

创建一个专门用于获取 Token 的接口
在该接口的 ” 前置脚本 ” 中添加 Token 获取逻辑
将获取到的 Token 存储到环境变量中

Apifox 的环境变量分为多个层级：

全局变量：对所有项目可用
环境变量：只在特定环境下可用
临时变量：只在当前请求或集合运行期间有效

对于 Token 这种敏感信息，建议使用环境级变量。

为了防止 Token 过期，我们需要实现自动刷新机制。核心思路是：

在每次请求前检查 Token 是否有效
如果 Token 即将过期或已过期，自动获取新 Token
更新环境变量中的 Token 值

以下是一个完整的 JavaScript 实现示例，包含错误处理和日志输出：

// 前置脚本：自动获取并更新 Token
const getAuthToken = async () => {
    try {
        // 1. 构造获取 Token 的请求
        const tokenResponse = await pm.sendRequest({url: pm.environment.get("auth_url") + "/oauth/token",
            method: "POST",
            header: {"Content-Type": "application/json"},
            body: {
                mode: "raw",
                raw: JSON.stringify({client_id: pm.environment.get("client_id"),
                    client_secret: pm.environment.get("client_secret"),
                    grant_type: "client_credentials"
                })
            }
        });

        // 2. 解析响应并存储 Token
        const responseJson = tokenResponse.json();
        if (responseJson.access_token) {pm.environment.set("access_token", responseJson.access_token);
            pm.environment.set("token_expires_in", Date.now() + (responseJson.expires_in * 1000));
            console.log("Token 获取成功", responseJson.access_token);
        } else {throw new Error("获取 Token 失败：" + JSON.stringify(responseJson));
        }
    } catch (error) {console.error("获取 Token 出错：", error);
        throw error; // 抛出错误让测试用例失败
    }
};

// 检查 Token 是否有效
const checkTokenValid = () => {const tokenExpires = pm.environment.get("token_expires_in");
    const currentTime = Date.now();

    // Token 不存在或将在 30 秒内过期，则刷新
    if (!tokenExpires || currentTime > (tokenExpires - 30000)) {console.log("Token 即将过期，自动刷新");
        return false;
    }
    return true;
};

// 主执行逻辑
if (!checkTokenValid()) {await getAuthToken();
}

// 设置当前请求的 Authorization 头部
pm.request.headers.add({
    key: "Authorization",
    value: `Bearer ${pm.environment.get("access_token")}`
});