H3C HCL与WSL2共存

private String idmsA3Token = "";
private String questionAnswerPairId = "";
private final String SagebatchId = "QA-Batch_1" + LocalDate.now().format(DateTimeFormatter.ofPattern("dd-MM-YY"));

@BeforeMethod
public void getIDMSA3Token() {

    // Create request body for token generation
    JSONObject body = new JSONObject();
    body.put("appId", Common.HEADER_APP_ID);
    body.put("appPassword", Common.HEADER_APP_PASSWORD);
    body.put("otherApp", "135942");
    body.put("context", Common.HEADER_CONTEXT_FEEDBACKENGINE);
    body.put("contextVersion", "1");
    body.put("oneTimeToken", false);

    // Logging the request body for debugging
    System.out.println("Request Body for Token Generation: " + body.toJSONString());

    // Make POST request to get token
    JsonPath jsonPath = given()
        .headers("Content-Type", "application/json")
        .contentType("application/json")
        .accept(ContentType.JSON)
        .body(body.toJSONString())
        .post("https://idmsac-uat.corp.apple.com/auth/apptoapp/token/generate")
        .then()
        .log().all() // Log the full response for debugging
        .statusCode(200) // Expecting 200 status
        .body(notNullValue())
        .extract()
        .jsonPath();

    // Check if the token is generated
    idmsA3Token = jsonPath.get("token").toString();
    System.out.println("Generated IDMSA3 Token: " + idmsA3Token);
    Assert.assertNotNull(idmsA3Token, "Token was not generated successfully!");
}

@Test
public void verify1CreateSage() {

    // Log token to ensure it's being passed correctly
    System.out.println("Token used for Sage creation: " + idmsA3Token);

    // Get initial Sage count
    JsonPath jsonPath = getAllSages();
    int sageCount = jsonPath.getList("$").size();

    // Set headers for the request
    JSONObject headers = new JSONObject();
    headers.put("Content-Type", "application/json");
    headers.put("X_APPID", Common.HEADER_APP_ID);
    headers.put("X-Apple-Client-App-ID", Common.HEADER_APP_ID);
    headers.put("X-Apple-Client-App-Name", Common.HEADER_CONTEXT_SUPPORTENGINE);
    headers.put("X-Apple-IDMS-A3-Token", idmsA3Token); // Pass token in headers

    // Prepare the body of the request
    JSONObject body = new JSONObject();
    JSONObject generation = new JSONObject();
    JSONObject tags = new JSONObject();
    generation.put("model", "gemini");
    tags.put("generation", generation);
    body.put("tags", tags);
    body.put("batchId", SagebatchId); // Ensure this matches the generated format
    body.put("question", "How to improve battery life of Iphone");
    body.put("answer", "Battery Life Improvement Answer");
    body.put("articleId", "QA_ArticleID");
    body.put("articleAccess", "Public");

    // Logging the request body and headers for debugging
    System.out.println("Request Body for Sage Creation: " + body.toJSONString());
    System.out.println("Request Headers: " + headers.toJSONString());

    // Send the POST request to create a Sage entry
    jsonPath = given()
        .headers(headers)
        .contentType("application/json")
        .accept(ContentType.JSON)
        .body(body.toJSONString())
        .post(Common.APP_URL + "/api/v1/service/topanswer/pregenerated/summaries")
        .then()
        .log().all() // Log the full response for debugging
        .statusCode(200) // Expecting 200 status
        .body(notNullValue())
        .extract()
        .jsonPath();

    // Debugging: Print the response
    System.out.println(jsonPath.prettify());

    // Retrieve and verify the newly created Sage entry
    jsonPath = getAllSages();
    questionAnswerPairId = jsonPath.get("find { it.batchId == '" + SagebatchId + "' }.questionAnswerPairId");

    System.out.println("questionAnswerPairId: " + questionAnswerPairId);

    // Assert that the number of sages has increased by 1
    Assert.assertEquals(jsonPath.getList("$").size(), sageCount + 1);

    // Check if the batchId is found and not null
    String batchId = jsonPath.getString("find { it.batchId == '" + SagebatchId + "' }.batchId");

    assert batchId != null : "displayName not found.";
    System.out.println("Assert statements passed.");
}

// Mock method for getting all sages (you need to implement the real method)
private JsonPath getAllSages() {
    // This should call the actual API endpoint to retrieve all sages
    return given()
        .headers("Content-Type", "application/json")
        .accept(ContentType.JSON)
        .get(Common.APP_URL + "/api/v1/service/topanswer/pregenerated/summaries")
        .then()
        .statusCode(200)
        .extract()
        .jsonPath();
}