JUnit 테스트 - 세션, 포워딩 확인

merci·2023년 2월 25일
0
import static org.assertj.core.api.Assertions.assertThat;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;

import javax.servlet.http.HttpSession;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
import org.springframework.http.MediaType;
import org.springframework.mock.web.MockHttpSession;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.ResultActions;
import org.springframework.transaction.annotation.Transactional;

import com.example.loginexample.model.User;

@AutoConfigureMockMvc
@SpringBootTest(webEnvironment = WebEnvironment.MOCK)
public class UserControllerTest {
    
    @Autowired
    private MockMvc mvc;

    private MockHttpSession session;

    @BeforeEach
    public void setUp(){
        User mockUser = new User();
        mockUser.setId(2);
        mockUser.setUsername("ssar");
        mockUser.setPassword("1234");
        mockUser.setEmail("ssar@nate.com");

        session = new MockHttpSession();
        session.setAttribute("principal", mockUser);
    }
    
    @Transactional
    @Test
    public void userLogin_test() throws Exception{
        //given
        String resp = "username=ssar&password=1234";

        //when
        ResultActions rs = mvc.perform(post("/login").content(resp)
        		.contentType(MediaType.APPLICATION_FORM_URLENCODED_VALUE));
    
        //then
        MvcResult mvcResult = rs.andExpect(status().is3xxRedirection()).andReturn();
        HttpSession session =  mvcResult.getRequest().getSession();
        User userPS = (User) session.getAttribute("principal");
        assertThat(userPS.getUsername()).isEqualTo("ssar");
        assertThat(userPS.getPassword()).isEqualTo("1234");
    }

    @Transactional
    @Test
    public void main_test() throws Exception {
        ResultActions rs = mvc.perform(get("/").session(session));
        MvcResult mvc =  rs.andExpect(status().isOk()).andReturn();
        String result = mvc.getResponse().getForwardedUrl();
        assertThat(result).isEqualTo("/WEB-INF/view/user/main.jsp");
    }
}
profile
작은것부터

0개의 댓글