首页 > 解决方案 > 如何使用 MockMvc 和 Mockito 测试无状态

问题描述

我最近继承了一些 java 代码,需要在 HttpSecurity 配置中切换到无状态:

http.csrf().disable()
    .sessionManagement().sessionCreationPolicy(SessionCreationPolicy.STATELESS);

之后,一些使用 MockMvc 的测试失败了。我知道@WithMockUser 不适用于无状态,那么我应该改变什么才能让它通过?这是一个示例测试:

    @Before
    public void setUp() throws Exception {

        this.mockMvc = MockMvcBuilders.standaloneSetup(apAdminController)
            .apply(SecurityMockMvcConfigurers.springSecurity(springSecurityFilterChain))
            .setCustomArgumentResolvers(new PageableHandlerMethodArgumentResolver())
            .build();

        storeUser = new StoreUser();
        storeUser.setUsername("smith");
        storeUser.setEnabled(true);
    }

    @Test
    @WithMockUser(roles = "ADMIN")
    public void testViewStoreUserWithAdmin() {
        try {
            mockMvc.perform(
                get("{resource}/1", RESOURCE).with(testSecurityContext()))
                .andDo(print())
                .andExpect(status().isOk());

        } catch (Exception e) {
            fail("Test failed: " + e.getMessage());
        }
    }

标签: javamockitostatelessmockmvc

解决方案


我花了很长时间才找到答案,这就是我最终得到的结果,以防万一有人在苦苦挣扎:

在测试中将 SessionCreationPolicy 从“If_REQUIRED”或“ALWAYS”更改为“STATELESS”后:将导入替换.SecurityMockMvcRequestPostProcessors.testSecurityContext;为导入.SecurityMockMvcRequestPostProcessors.user;

并在设置中使用它:

this.mockMvc = MockMvcBuilders.webAppContextSetup(context)
            .addFilters(springSecurityFilterChain)
            .build();

删除所有注释@WithMockUser ,最后更改

get("{resource}", RESOURCE).with(testSecurityContext()))

get("{resource}", RESOURCE).with(user("admin").roles("ADMIN")))


推荐阅读