MissionStepTest.java

package roomescape;

import io.restassured.RestAssured;
import io.restassured.http.ContentType;
import io.restassured.response.ExtractableResponse;
import io.restassured.response.Response;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.annotation.DirtiesContext;

import java.util.HashMap;
import java.util.Map;
import roomescape.reservation.ReservationResponse;

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
@DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_EACH_TEST_METHOD)
public class MissionStepTest {

  @Test
  void 일단계() {
    String token = createToken("admin@email.com", "password");
    assertThat(token).isNotBlank();

    ExtractableResponse<Response> checkResponse = RestAssured.given().log().all()
        .contentType(ContentType.JSON)
        .cookie("token", token)
        .when().get("/login/check")
        .then().log().all()
        .statusCode(200)
        .extract();

    assertThat(checkResponse.body().jsonPath().getString("name")).isEqualTo("어드민");
  }

  private static String createToken(final String mail, final String password) {
    Map<String, String> params = new HashMap<>();
    params.put("email", mail);
    params.put("password", password);

    ExtractableResponse<Response> response = RestAssured.given().log().all()
        .contentType(ContentType.JSON)
        .body(params)
        .when().post("/login")
        .then().log().all()
        .statusCode(200)
        .extract();

    return response.headers().get("Set-Cookie").getValue().split(";")[0].split("=")[1];
  }

  @Test
  void 이단계() {
    String token = createToken("admin@email.com",
        "password");  // 일단계에서 토큰을 추출하는 로직을 메서드로 따로 만들어서 활용하세요.

    Map<String, String> params = new HashMap<>();
    params.put("date", "2024-03-01");
    params.put("time", "1");
    params.put("theme", "1");

    ExtractableResponse<Response> response = RestAssured.given().log().all()
        .body(params)
        .cookie("token", token)
        .contentType(ContentType.JSON)
        .post("/reservations")
        .then().log().all()
        .extract();

    assertThat(response.statusCode()).isEqualTo(201);
    assertThat(response.as(ReservationResponse.class).getName()).isEqualTo("어드민");

    params.put("name", "브라운");

    ExtractableResponse<Response> adminResponse = RestAssured.given().log().all()
        .body(params)
        .cookie("token", token)
        .contentType(ContentType.JSON)
        .post("/reservations")
        .then().log().all()
        .extract();

    assertThat(adminResponse.statusCode()).isEqualTo(201);
    assertThat(adminResponse.as(ReservationResponse.class).getName()).isEqualTo("브라운");
  }

  @Test
  void 삼단계() {
    String brownToken = createToken("brown@email.com", "password");

    RestAssured.given().log().all()
        .cookie("token", brownToken)
        .get("/admin")
        .then().log().all()
        .statusCode(401);

    String adminToken = createToken("admin@email.com", "password");

    RestAssured.given().log().all()
        .cookie("token", adminToken)
        .get("/admin")
        .then().log().all()
        .statusCode(200);
  }
}