Rest Assured中处理POST请求的307临时重定向：手动追踪策略

import io.restassured.RestAssured;
import io.restassured.http.ContentType;
import io.restassured.response.Response;
import io.restassured.specification.RequestSpecification;
import io.restassured.path.json.JsonPath;
import org.junit.BeforeClass;
import org.junit.Test;

import static io.restassured.RestAssured.*;
import static org.hamcrest.Matchers.*;

public class PostRedirectHandler {

    // 假设这些配置来自ConfigReader或直接定义
    private static String BASE_URI = "http://your-api-base-url.com";
    private static String AUTH_ENDPOINT = "/authenticate";
    private static String USERNAME = "testuser";
    private static String PASSWORD = "testpassword";

    @BeforeClass
    public static void setup() {
        baseURI = BASE_URI;
        // 可以选择性地配置日志，方便调试
        RestAssured.enableLoggingOfRequestAndResponseIfValidationFails();
    }

    @Test
    public void handlePost307RedirectManually() {
        String accessToken = getAccessTokenManually();
        System.out.println("获取到的AccessToken: " + accessToken);

        // 假设后续请求需要这个AccessToken
        // ...
    }

    /**
     * 模拟原始用户代码中的授权和AccessToken获取逻辑，并手动处理重定向
     * @return 获取到的AccessToken
     */
    public String getAccessTokenManually() {
        // 步骤1: 发送初始POST请求，禁用自动重定向
        Response initialResponse = given().log().all() // 记录请求和响应的详细信息，便于调试
                .contentType(ContentType.JSON)
                .body("{\n" +
                        " \"username\": \"" + USERNAME + "\",\n" +
                        "\"password\": \"" + PASSWORD + "\" \n" +
                        "}")
                .redirects().follow(false) // 禁用自动重定向
                .when()
                .post(AUTH_ENDPOINT);

        // 验证初始响应是否为307重定向
        initialResponse.then().log().all()
                .statusCode(307); // 期望收到307状态码

        // 步骤2: 解析重定向响应
        String locationHeader = initialResponse.getHeader("Location");
        // 提取会话Cookie，例如JSESSIONID，如果存在且需要维持会话
        String jsessionid = initialResponse.getCookie("JSESSIONID");

        System.out.println("收到307重定向到: " + locationHeader);
        System.out.println("提取到的JSESSIONID: " + jsessionid);

        // 确保Location头存在
        if (locationHeader == null || locationHeader.isEmpty()) {
            throw new RuntimeException("重定向响应中未找到Location头！");
        }

        // 步骤3: 发起后续请求到重定向目标
        // 这里假设重定向目标是一个获取AccessToken的端点，并且可能需要GET请求
        // 如果重定向要求继续POST，则需要再次构建POST请求体
        RequestSpecification followUpRequest = given().log().all();

        if (jsessionid != null && !jsessionid.isEmpty()) {
            followUpRequest.cookie("JSESSIONID", jsessionid);
        }

        Response finalResponse = followUpRequest
                .when()
                .get(locationHeader); // 假设重定向后是GET请求获取资源

        // 验证最终响应是否成功
        finalResponse.then().log().all()
                .statusCode(200) // 期望最终响应是200 OK
                .body("AccessToken", notNullValue()); // 验证AccessToken是否存在

        String jsonString = finalResponse.asString();
        return JsonPath.from(jsonString).getString("AccessToken");
    }

    /**
     * 原始用户代码中获取AccessToken的方法，未处理重定向
     * 可能会返回307状态码
     */
    public String getAccessTokenOriginal(String endpoint) {
        Response response = given().log().all()
                .header("Content-Type", "application/json")
                .body("{\n" +
                        " \"username\": \"" + USERNAME + "\",\n" +
                        "\"password\": \"" + PASSWORD + "\" \n" +
                        "}")
                .post(endpoint);

        // 此时如果收到307，这里可能无法正确解析AccessToken
        String jsonString = response.asString();
        return JsonPath.from(jsonString).get("AccessToken");
    }
}