응용해볼려고 세션을 이용한 테스트코드 작성중인데 잘모르겠습니다.

스프링 MVC 2편 - 백엔드 웹 개발 활용 기술

서블릿 필터 - 인증 체크

작성

강의를 보다 혼자 응용해봤는데

@Mock
    private HttpServletRequest request;

    @Mock
    private HttpSession session;

@Test
void store_canAccess() throws Exception {
    // given
    given(request.getSession(false)).willReturn(session);
    given(session.getAttribute(SessionConst.LOGIN_PARTNER))
            .willReturn(1L);

    // when // then
    mockMvc.perform(post("/store/somePoint")
                    .contentType(MediaType.APPLICATION_JSON))
            .andExpect(status().isOk());
}

@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
    HttpSession session = request.getSession(false);
    if (session == null || session.getAttribute(SessionConst.LOGIN_PARTNER) == null) {
        throw new CustomException(ErrorCode.PARTNER_ONLY_ACCESS);
    }

    return true;
}

채찍피티의 도움과 함께 이렇게 짰는데 CustomException이 터져요...혹시 잘못된 부분이 있을까요

spring mvc

답변 2

kongminoo

질문자

두시간동안 삽질하다가 잘된거 같은데

@Test
void store_canAccess() throws Exception {
    // given
    MockHttpSession mockSession = new MockHttpSession();
    mockSession.setAttribute(SessionConst.LOGIN_PARTNER, 1L);
    given(request.getSession(false)).willReturn(mockSession);

    // when // then
    mockMvc.perform(post("/store/somePoint")
                    .session(mockSession)
                    .contentType(MediaType.APPLICATION_JSON))
            .andExpect(status().isOk());
}

이런식으로 했습니다.

가짜(?)mockhttpsession을 만들어서 거기에다가 키,밸류 담은 후에 리턴시켯더니 원하는대로 동작한는 거 같아요

y2gcoder

안녕하세요. kongminoo님, 공식 서포터즈 y2gcoder입니다.

스스로 잘 해결하셨습니다 :)

강의 수강 중 강의 관련 궁금한 사항은 언제든 질문 남겨주세요^^

인프런 AI 인턴

안녕하세요, 인프런 AI 인턴입니다.

kongminoo님의 질문에 대해 살펴보니, HttpSession을 제대로 Mocking하지 않아 발생하는 문제일 수 있습니다. CustomException이 발생하는 것은 세션이 정상적으로 설정되지 않았거나, SessionConst.LOGIN_PARTNER 속성이 세션에 존재하지 않기 때문입니다. 아래에서 문제의 주요 원인을 분석하고 해결책을 제안드립니다.

문제 가능성

Mocking 오류: request.getSession(false)가 null을 반환하거나, session.getAttribute(SessionConst.LOGIN_PARTNER)가 기대한 값을 반환하지 않는 상황일 수 있습니다.
세션 설정 오류: Mock 설정 중 HttpSession이 올바르게 설정되지 않았을 가능성이 있습니다. 잘못된 Mock 설정이 세션이나 속성이 없다는 것으로 간주되어 CustomException이 발생할 수 있습니다.

해결 방법

Mock 객체의 올바른 설정 확인:
- given(request.getSession(false)).willReturn(session); 이 코드를 통해 request가 세션 객체를 반환하도록 설정했으므로 문제 없어 보입니다.
- session.getAttribute(SessionConst.LOGIN_PARTNER) 부분의 설정도 확인해보세요. 여기서 코드 상으로는 1L이라는 Long 값이 반환되도록 설정되어 있습니다. 해당 값이 올바른지 확인해 주세요.
테스트 코드에 세션 초기화 추가:
- 테스트 코드의 세션 초기화를 명확히 하여 값이 올바르게 설정되어 있는지 다시 한 번 확인해보세요.
로그 활용:
- 예외가 발생할 때 추가적인 정보를 로그로 남기도록 설정하여 어떤 단계에서 문제가 발생하는지 더 자세히 확인할 수 있습니다.

혹시, 자세한 해결 방안이나 추가적인 코드 수정이 필요하실 경우, 이미 유사한 문제에 대해 논의된 게시글들이 있으니 참고하시기를 권장드립니다:

인증 필터 로직 관련 질문있습니다!: 세션 관리와 관련된 인터셉터 로직의 기본 원리를 이해하는 데 유용할 수 있습니다.

저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다.
현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏
추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.

인프런 커뮤니티 질문&답변