apifox自动登录脚本

发布时间 2023-07-13 15:00:00作者: 天意啊

apifox自动登录脚本编写思路:

1、能够通过环境变量获取账号、密码

2、通过账号密码能够自动调用登录接口

3、登录接口不能每次请求都调用,需要有判断条件

4、要支持更换账号密码

 

脚本:

// 定义发送登录接口请求方法
function sendLoginRequest() {
    // 获取环境里的 前置URL
    const baseUrl = pm.environment.get("BASE_URL");

    // 登录用户名,这里从环境变量 LOGIN_USERNAME 获取,也可以写死(但是不建议)
    const username = pm.environment.get("LOGIN_USERNAME");
    // 登录用户名,这里从环境变量 LOGIN_PASSWORD 获取,也可以写死(但是不建议)
    const password = pm.environment.get("LOGIN_PASSWORD");
    // 构造一个 POST x-www-form-urlencoded 格式请求。这里需要改成你们实际登录接口的请求参数。
    const loginRequest = {
        url: baseUrl + "/api/apifox/login",
        method: "POST",
        // body 为 x-www-form-urlencoded 格式
        // body: {
        //     mode: "urlencoded", // 此处为 urlencoded
        //     // 此处为 urlencoded
        //     urlencoded: [
        //         { key: "account", value: "apifox" },
        //         { key: "password", value: "123456" },
        //     ],
        // },
        /*
        // body 为 form-data 格式
        body: {
          mode: 'formdata', // 此处为 formdata
          // 此处为 formdata
          formdata: [   
            { key: 'account', value: 'apifox' },
            { key: 'password', value: '123456' }
          ]
        }
    */
        // body 为 json 格式
        header: {
          "Content-Type": "application/json", // 注意:header 需要加上 Content-Type
        },
        body: {
          mode: 'raw',// 此处为 raw
          raw: JSON.stringify({ username: username, password: password }), // 序列化后的 json 字符串
        }
    
        // body 为 raw 或 json 格式
        // body: {
        //   mode: 'raw',
        //   raw: '此处为 body 内容',
        // }
      
    };

    // 发送请求。
    // pm.sendrequest 参考文档: https://www.apifox.cn/help/app/scripts/api-references/pm-reference/#pm-sendrequest
    pm.sendRequest(loginRequest, function (err, res) {
        if (err) {
            console.log(err);
        } else {
            // 读取接口返回的 json 数据。
            // 如果你的 token 信息是存放在 cookie 的,可以使用 res.cookies.get('token') 方式获取。
            // cookies 参考文档:https://www.apifox.cn/help/app/scripts/api-references/pm-reference/#pm-cookies
            const jsonData = res.json();
            // 将 accessToken 写入环境变量 ACCESS_TOKEN
            pm.environment.set("ACCESS_TOKEN", jsonData.data.access_token);
            // 将 accessTokenExpires 过期时间写入环境变量 ACCESS_TOKEN_EXPIRES
            // 将 token 过期时间转换为毫秒数
            // console.log(jsonData.data.expires_in)
            // console.log(Date.now())
            pm.environment.set(
                "ACCESS_TOKEN_EXPIRES",
                Date.now() + jsonData.data.expires_in * 60 * 1000
            );
            pm.environment.set(
                "ACCOUNT_FLAG","T"
            );
        }
    });
}

// 获取环境变量里的 ACCESS_TOKEN
const accessToken = pm.environment.get("ACCESS_TOKEN");

// 获取环境变量里的 ACCESS_TOKEN_EXPIRES
const accessTokenExpires = pm.environment.get("ACCESS_TOKEN_EXPIRES");
// 获取环境变量里的 ACCOUNT_FLAG
const accountFlag = pm.environment.get("ACCOUNT_FLAG");
// 获取当前时间的时间戳
 var currentTimestamp = Date.now();
// 如 ACCESS_TOKEN 没有值,或 ACCESS_TOKEN_EXPIRES 已过期,则执行发送登录接口请求

if (
    !accessToken || 
    (accessTokenExpires && currentTimestamp > accessTokenExpires) ||
    accountFlag !== "T" 
) {
    sendLoginRequest();
}