对 HttpSession 的部分模拟

Question

有一个我想测试的 servlet，servlet 有会话并将“loginBean”（包含登录用户相关信息）放入会话中，我已经模拟并且工作正常，现在在 GUI 级别，有 2 个选项卡， DetailSet1 ， detailsS​​et2 ，当你输入 DetailSet1 的数据时，它会保存在 session 中，并且还会执行一些业务逻辑，现在谈到 DetailsS​​et2 ，你已经在 session 中有 DetailSet1 ，所以它得到了它所需要的一切，数据保存在 DB 中。现在很明显我必须模拟 HttpSession 因为我从容器外部运行单元案例，但是存储的数据也在 HttpSession 中，如果我也模拟这些，它就违背了测试的目的。所以我的问题是，我需要 HttpSession 对象来为我模拟它的对象返回模拟数据，并且假设在其他情况下它的行为就像任何普通的 HttpSession 对象一样。就像，如果代码是 session.setAttribute("name","Vivek") ，那么 session.getAttribute("name") 应该在那之后返回“Vivek”，但是如果是模拟对象，它返回“NULL”，为什么？因为我没有嘲笑“getAttribute("name") 的行为。

简而言之，对 HttpSession 或接口进行部分模拟。

Answer 1

HttpSession 是一个接口，因此您需要编写自己的实现或模拟它。我建议使用 Mockito 模拟它，然后将 getAttribute 和 setAttribute 存根以委托给 HashMap 或其他一些合适的结构。

所以在您的测试类中，您将拥有

的字段

• 你嘲笑的HttpSession，
• 真正的HashMap<String,Object>

您将为getAttribute 和setAttribute 中的每一个使用Answer<Object> 对象。每个Answer 只会将调用委托给HashMap。

您可以在@Before 方法或@Test 这样的方法中设置所有这些。

@Mock private HttpSession mockHttpSession;
Map<String,Object> attributes = new HashMap<String,Object>();

@Test
public void theTestMethod() {

    Mockito.doAnswer(new Answer<Object>(){
        @Override
        public Object answer(InvocationOnMock invocation) throws Throwable {
            String key = (String) invocation.getArguments()[0];
            return attributes.get(key);
        }
    }).when(mockHttpSession).getAttribute(anyString());

    Mockito.doAnswer(new Answer<Object>(){
        @Override
        public Object answer(InvocationOnMock invocation) throws Throwable {
            String key = (String) invocation.getArguments()[0];
            Object value = invocation.getArguments()[1];
            attributes.put(key, value);
            return null;
        }
    }).when(mockHttpSession).setAttribute(anyString(), any());

Answer 2

Spring Testing framework 包括 MockHttpSession。它可以通过 Maven 获得：

<dependency>
    <groupId>org.springframework</groupId>
    <artifactId>spring-test</artifactId>
    <version>4.0.1.RELEASE</version>
</dependency>