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

1次阅读

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

在 API 测试过程中，Token 的维护一直是个让人头疼的问题。每次手动复制粘贴 Token，不仅效率低下，还容易出错。特别是在 Token 过期后，整个测试流程就中断了，需要重新获取再继续。今天我就来分享如何在 Apifox 中实现 Token 的自动获取与刷新，解放我们的双手。

在实际开发中，我们经常会遇到以下问题：

Token 过期导致测试中断，需要重新手动获取
不同环境（开发、测试、生产）使用不同 Token，切换麻烦
团队成员之间 Token 不同步，协作困难
敏感 Token 信息可能被意外泄露

这些问题不仅降低了测试效率，还增加了维护成本。如果能实现 Token 的自动获取和刷新，就能大幅提升我们的工作效率。

Apifox 提供了强大的脚本功能，可以帮助我们解决这些问题。主要通过以下两个机制：

环境变量管理：可以存储 Token 等敏感信息
脚本执行：支持在请求前后执行 JavaScript 代码

具体来说，我们可以利用 Pre-request Script 在发送请求前自动获取 Token，并将其存储在环境变量中。后续请求直接从环境变量读取，无需重复获取。

下面是一个完整的实现方案，包含 Token 获取、存储和刷新逻辑：

// 检查 Token 是否有效
function isTokenValid(token) {if (!token) return false;
    // 这里可以添加更复杂的校验逻辑，比如解析 JWT 判断过期时间
    return true;
}

// 从环境变量获取 Token
const currentToken = pm.environment.get('API_TOKEN');

// 如果 Token 不存在或无效，则重新获取
if (!isTokenValid(currentToken)) {
    // 获取新 Token 的请求配置
    const getTokenRequest = {url: pm.environment.get('AUTH_URL') || 'https://api.example.com/auth',
        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'
            })
        }
    };

    // 发送获取 Token 的请求
    pm.sendRequest(getTokenRequest, (err, response) => {if (err) {console.error('获取 Token 失败:', err);
            return;
        }

        // 解析响应数据
        const jsonData = response.json();
        if (jsonData && jsonData.access_token) {
            // 将新 Token 存入环境变量
            pm.environment.set('API_TOKEN', jsonData.access_token);
            console.log('Token 更新成功');

            // 如果需要，可以同时存储过期时间
            if (jsonData.expires_in) {const expiresAt = new Date();
                expiresAt.setSeconds(expiresAt.getSeconds() + jsonData.expires_in);
                pm.environment.set('TOKEN_EXPIRES_AT', expiresAt.toISOString());
            }
        } else {console.error('无效的 Token 响应:', jsonData);
        }
    });
}

这段代码主要实现了以下功能：

isTokenValid函数检查 Token 是否有效
从环境变量读取当前 Token
如果 Token 无效，构造获取 Token 的请求
发送请求并处理响应
将新 Token 存入环境变量

关键点说明：

使用 pm.environment 管理环境变量
pm.sendRequest发送异步请求
处理各种可能的错误情况
可以扩展 Token 过期时间的检查

在实际使用中，我们需要考虑以下异常情况：

网络问题导致获取 Token 失败
认证服务器返回错误
Token 格式不正确

完善后的异常处理代码如下：

pm.sendRequest(getTokenRequest, (err, response) => {if (err) {console.error('网络请求失败:', err);
        // 可以设置重试逻辑
        return;
    }

    try {const jsonData = response.json();
        if (!jsonData) {throw new Error('无效的 JSON 响应');
        }

        if (response.code !== 200) {throw new Error(` 认证失败: ${jsonData.error || '未知错误'}`);
        }

        if (!jsonData.access_token) {throw new Error('响应中缺少 access_token');
        }

        // 存储 Token...
    } catch (e) {console.error('处理 Token 响应时出错:', e);
    }
});

在生产环境中使用时，建议注意以下几点：