Spring测试-Mock@Model属性

Question

我有一个RestController，我正试图通过SpringMVC测试来测试它。它包含以下ModelAttribute：

    @ModelAttribute("authUser")
    public User authUser(@AuthenticationPrincipal SpringAuthUser springAuthUser) throws Exception {
        User user = ConstantsHome.userprofileMgr.getUserByUserId(springAuthUser.getUsername(), true, true);
        user.updateRights(null);
        request.getSession().setAttribute(ConstantsHome.USEROBJECT_KEY, user);
        return user;
    }

当我对这个RestController运行测试时，我在这个AuthUser方法中得到一个NullPointerExc0019。

有没有一种方法可以模拟这个方法，使得模拟的方法被用来代替这个方法进行测试？我已经阅读了关于这个的其他帖子，并认为我可以在测试中传递一个“AuthUser”参数，但这不起作用。最终试图让这个“用户”不要扔NPE...这是我的测试...

    @Test
    public void testGetAllUsers() throws Exception {
        String userJson = AvadaObjectMapper.getMapper().writeValueAsString(new User());
        System.out.println("userJson=" + userJson);
        MvcResult result = this.mockMvc.perform(get("/").param("authUser", userJson).accept(MediaType.parseMediaType("application/json;charset=UTF-8")))
            .andExpect(status().isOk())
            .andExpect(content().contentType("application/json"))
            .andDo(print())
            .andReturn();

        String content = result.getResponse().getContentAsString();

        assertTrue(content.contains("Hello"));
    }

Answer 1

我能够通过下面的测试配置类让我的AuthUser工作。请特别注意. defaultRequest...行，因为这是建立授权用户的地方。我在所有测试中使用下面的类作为@ContextConfiguration类。

import com.avada.rest.api.users.UserService;
import com.avada.rest.api.users.TestUserService;
import com.avada.rest.security.SpringAuthUser;

import java.util.HashSet;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.FilterType;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.authority.SimpleGrantedAuthority;
import org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors;
import org.springframework.security.test.web.servlet.setup.SecurityMockMvcConfigurers;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;

@Configuration
@ComponentScan(excludeFilters={
                @ComponentScan.Filter(type=FilterType.ASSIGNABLE_TYPE, value=UserService.class)
              })
public class TestSpringRestConfig {

    public static final SpringAuthUser AUTH_USER =
        new SpringAuthUser("test", "test",
            new HashSet<GrantedAuthority>() {{
                add(new SimpleGrantedAuthority("ROLE_ADMIN"));
            }}
        );

    @Bean
    public UserService userService() {
        return new TestUserService();
    }

    @Bean
    public MockMvc mockMvc(WebApplicationContext wac) {
          return MockMvcBuilders
                 .webAppContextSetup(wac)
                 .defaultRequest(MockMvcRequestBuilders.get("/").with(SecurityMockMvcRequestPostProcessors.user(AUTH_USER)))
                 .apply(SecurityMockMvcConfigurers.springSecurity())
                 .build();
    }
}