为什么Spring MockMvc结果不包含cookie？

Question

我试图在RESTAPI中对登录和安全性进行单元测试，所以我尝试尽可能地模拟现实生活中的请求序列。

我的第一个要求是：

this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).
    addFilters(springSecurityFilterChain).build();
this.mapper = new ObjectMapper();
....
MvcResult result=mockMvc.perform(get("/login/csrf")).andExpect(status().is(200)).andReturn();
Cookie[] cookies = result.getResponse().getCookies();

（请参见pastebin上的完整类）。

我尝试在这里获取cookie，以便以后能够使用收到的CSRF令牌登录，但是cookie数组是空的！

然而，如果我运行我的应用程序并调用

curl -i http://localhost:8080/login/csrf

我确实得到了一个Set Cookie头，可以稍后使用该Cookie（和CSRF令牌）进行身份验证。

所以问题是：如何让MockMvc将cookie返回给我？

Answer 1

根据P.Péter的回答，a生成了此代码片段，该代码片段将自动获取并返回在mockMvc上执行的每个请求（MockHttpServletRequestBuilder）的会话。

@Autowired
private WebApplicationContext webApplicationContext;

@Autowired
private Filter springSecurityFilterChain;

@Before
public void setUp() throws Exception {
    final MockHttpServletRequestBuilder defaultRequestBuilder = get("/dummy-path");
    this.mockMvc = MockMvcBuilders.webAppContextSetup(this.webApplicationContext)
            .defaultRequest(defaultRequestBuilder)
            .alwaysDo(result -> setSessionBackOnRequestBuilder(defaultRequestBuilder, result.getRequest()))
            .apply(springSecurity(springSecurityFilterChain))
            .build();
}

private MockHttpServletRequest setSessionBackOnRequestBuilder(final MockHttpServletRequestBuilder requestBuilder,
                                                             final MockHttpServletRequest request) {
    requestBuilder.session((MockHttpSession) request.getSession());
    return request;
}

详细回答：检查这个解决方案（答案是Spring4）：如何使用Spring3.2新mvc测试登录用户

Answer 2

我的工作与RestTem板测试与饼干。RestTem板cookie处理程序

@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT)
@Import(RestTemplateWithCookies.class)
public class ApplicationTest {

    @LocalServerPort
    private int port;

    @Autowired
    private Environment env;

    @Autowired
    private RestTemplateWithCookies restTemplate;

    @Test
    public void appTest() throws Exception {
        HttpHeaders headers = new HttpHeaders();
        headers.set("Referer", env.getProperty("allowed_referer"));
        HttpEntity<String> entity = new HttpEntity<String>("parameters", headers);
        ResponseEntity<String> response = restTemplate.exchange("http://localhost:" + port + "/scan?email=[e-mail]", HttpMethod.GET, entity, String.class);
        assertTrue(response.getStatusCode() == HttpStatus.FOUND);
        HttpCookie c = restTemplate.getCoookies().stream().filter(x -> env.getProperty("server.session.cookie.name").equals(x.getName())).findAny().orElse(null);
        assertTrue(c != null);

    }

}

Answer 3

我找到了一个解决方法，使用直接从MockHttpServletRequest提取会话对象的功能：

session=(MockHttpSession)result.getRequest().getSession();

然后直接插入会话：

req.session(session);

我对这个解决方案不满意的原因是，如果在这方面，mock http pservlet的行为与真正的servlet不同，我如何确定它在其他情况下是否与真正的servlet的行为相同。所以我没有测试应用程序本身，这可能会在测试中留下漏洞。