无法从控制器中的请求获取令牌

时间:2015-04-15 18:21:36

标签: java spring token

我正在尝试在Spring中测试loginController,但我无法从请求中获取令牌。 在我的测试中

   private CsrfToken token; 
   HttpServletRequest request = new MockHttpServletRequest();

     @Before
     public void setUp() throws Exception {        
         MockitoAnnotations.initMocks(this);
         this.mockMvc = MockMvcBuilders.webAppContextSetup(wac).build();        
         token = new DefaultCsrfToken("1", "a", "b"); 
         request.setAttribute(CsrfToken.class.getName(), token); 
     }

    @Test
    public void testCtrlCorrecltyHandlesPassword() throws Exception {         
        CsrfToken token2 = (CsrfToken) request.getAttribute(CsrfToken.class.getName());

        MvcResult result = this.mockMvc.perform(get("/rest/login")
                .param("name", name)
                .param("password", password)
                .param("request", request.toString())
                //.sessionAttr("request", request)
                .accept(MediaType.APPLICATION_JSON))
                .andExpect(status().isOk())
                .andReturn();
    }

当我调试时,我看到token2已成功创建,我从请求获取它的值,但是当我以完全相同的方式尝试时在控制器中

CsrfToken token = (CsrfToken) request.getAttribute(CsrfToken.class.getName());

为空。

我还尝试.sessionAttr("request", request)@ModelAttribute("request")HttpServletRequest request作为控制器中的参数,但它不起作用。

1 个答案:

答案 0 :(得分:0)

...最终

    MockHttpSession session = new MockHttpSession();
    session.setAttribute("token", token);

    MvcResult result = this.mockMvc.perform(get("/rest/login").session(session)
            .header("X-Csrf-Token", "")
            .requestAttr(CsrfToken.class.getName(), token)
            .param("name", name)
            .param("password", password)
            .accept(MediaType.APPLICATION_JSON))
            .andExpect(status().isOk())
            .andReturn();