Question

我需要运行一个JUnit vs Spring MVC测试用例，其中前置条件包括HTTP Session中存在某些数据。最重要的是我无法连接session - 范围的bean：我必须访问httpServletContext.getSession()。

在显示代码之前，让我解释一下。我需要测试的控制器假定某个数据存储在会话中，否则抛出异常。这就是现在正确的行为，因为在没有会话的情况下永远不会调用该控制器，并且会话总是在登录时使用应用程序数据进行初始化。显然，控制器处于安全状态。

在我的测试中，我只需要测试该控制器是否根据请求参数返回重定向或404未找到。

我想构建我的测试用例，例如

@Autowired
private HttpServletRequest httpServletRequest;

@Autowired
private ModuleManager moduleManager;

@Autowired
private WebApplicationContext webApplicationContext;

private MenuItem rootMenu;

private MockMvc mockMvc;


@Before
public void setUp() throws Exception
{

    mockMvc = MockMvcBuilders.webAppContextSetup(webApplicationContext)
                             // No asserzioni
                             .build();

    rootMenu = moduleManager.getRootMenu()
                            .clone();
    httpServletRequest.getSession()
                      .setAttribute(MenuItem.SESSION_KEY, rootMenu);

    assertNotNull(rootMenu.getDescendant(existingSelectedMenu));
    assertNull(rootMenu.getDescendant(notExistingMenu));

}

@Test
public void testNavigate() throws Exception
{

    mockMvc.perform(get("/common/navigate?target=" + existingSelectedMenu))
           .andExpect(status().is3xxRedirection());

    assertNotSelected(rootMenu, existingSelectedMenu);

    mockMvc.perform(get("/common/navigate?target=" + notExistingMenu))
           .andExpect(status().is4xxClientError());

}

部分代码真正自我解释。无论如何，我希望/common/navigate使用我在会话中存储的值。喜欢这个

@RequestMapping(value = "/common/navigate",
        method = RequestMethod.GET)
public String navigate(@RequestParam("target") String target) throws NotFoundException
{

    MenuItem rootMenu = (MenuItem) httpServletRequest.getSession()
                                               .getAttribute(MenuItem.SESSION_KEY);
    if (rootMenu == null)
        throw new RuntimeException("Menu not found in session"); //Never happens

    MenuItem menuItem = rootMenu.getAndSelect(target);
    if (menuItem == null)
        throw new NotFoundException(MenuItem.class, target); //Expected

    return "redirect:" + menuItem.getUrl();
}

现在猜。运行代码时会发生什么？

在我评论的行中抛出了RuntimeException，因为在会话中找不到菜单对象

显然这个问题现在是隐含的，但我仍然会写它：如何将数据注入Session对象，以便被测控制器将它们作为前置条件提供？

Answer 1

现在自己找到解决方案。

问题是会话本身也必须被嘲笑。 Spring提供了一个MockHttpSession类来完成这个技巧。它可以预先填充所有前提条件，但必须将传递给每个MockMvc请求，以便模拟会将会话连接到（模拟的）servlet上下文。

以下代码初始化会话

    mockHttpSession = new MockHttpSession(webApplicationContext.getServletContext());

    mockHttpSession.setAttribute(MenuItem.SESSION_KEY, rootMenu);

以下通过连接到它的模拟会话执行请求

mockMvc.perform(get("/common/navigate?target=" + existingSelectedMenu).session(mockHttpSession))
           .andExpect(status().is3xxRedirection());

从JUnit测试将数据注入会话

现在猜。运行代码时会发生什么？

1 个答案: