HttpSession Junit 测试

Question

我无法对 HttpSession 进行模拟。测试方法如下所示：

@GetMapping
    @RequestMapping("/feed")
    public String feed(HttpSession session, Model model) throws UnauthorizedException {
        if (session.getAttribute("loginStatus") == null) throw new UnauthorizedException("You have to login first");
        Long userId = (Long) session.getAttribute("userId");
        model.addAttribute("posts", postService.feed(userId));
        return "posts/feed";
    }

测试看起来像这样：

 @Mock
    private PostService postService;

    private MockMvc mockMvc;

    private PostViewController controller;

    @Mock
    private HttpSession session;


    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        controller = new PostViewController(postService);
        mockMvc = MockMvcBuilders.standaloneSetup(controller).build();
    }

    @Test
    public void feed() throws Exception {
        when(session.getAttribute("loginStatus")).thenReturn(true);

        mockMvc.perform(get("/feed"))
                .andExpect(status().isOk())
                .andExpect(view().name("posts/feed"))
                .andExpect(model().attributeExists("posts"));
    }

我总是得到 UnauthorizedException，但我需要避免它。如何为会话添加一些参数来模拟工作？

Answer 1

您应该在配置 MockHttpServlet 期间使用相关会话方法来配置会话状态。在内部，它将为您正在构建的 MockHttpServlet 创建一个 MockHttpSession。

 mockMvc.perform(get("/feed")
           .sessionAttr("loginStatus", true)
           .sessionAttr("userId", 1234l))
                .andExpect(status().isOk())
                .andExpect(view().name("posts/feed"))
                .andExpect(model().attributeExists("posts"));

HttpSession Junit 测试

HttpSession Junit Testing

java

mocking

junit4

mockito