时间戳

首页 > 解决方案 > 如何在 MockMVC 帖子中将对象传递给 ModelAttrbiute？

java - 如何在 MockMVC 帖子中将对象传递给 ModelAttrbiute？

问题描述

用户型号： 

@Entity
@Table(name="user")
public class User {

@Id @GeneratedValue(strategy=GenerationType.IDENTITY)
private Integer id;

@NotBlank
@Column(name="username")
private String username;

@NotEmpty
@ManyToMany(fetch = FetchType.EAGER)
@JoinTable(name="user_role", joinColumns = {@JoinColumn(name="user_id")},
                             inverseJoinColumns = {@JoinColumn(name="role_id")})
private Set<Role> roles;
}

控制器：

@RequestMapping(value = {"/users/edit/{id}"}, method = RequestMethod.POST)
public String editUser(ModelMap model, @Valid @ModelAttribute("user") User user, BindingResult result) {

    if(result.hasErrors()) {
        return "AddUserView";
    }

    return "redirect:/users";
}

使用 MockMVC 进行测试：

@Test
public void performUpdateUserTest() throws Throwable {
            mockMvc.perform(post("/users/edit/{id}", user.getId())
                   .param("username", "User"));
}

好吧，我可以像往常一样使用param()传递参数用户名。但是我应该如何处理ROLES？该字段是一个单独的对象。我无法使用param()传递它。那怎么可能在测试中通过呢？

我发现的唯一方法是创建一个实体并使用.flashAttr()传递它：

    @Test
    public void performUpdateUserTest() throws Throwable {
          User user = new User("User", new HashSet<Role>(Arrays.asList(new Role("USER"))));
          mockMvc.perform(post("/users/edit/{id}", user.getId())
                 .flashAttr("user", user));
    }

但是，如果我需要测试由于ROLES字段中的绑定错误而无法更新该用户（ROLES 不能为空，并且假设它被设置为空），该怎么办？因此，我无法创建带有绑定错误的用户（并将其与 .flashAttr 一起使用），因为将引发异常。而且我还是要单独通过。

标签： javaspringspring-mvcspring-testmockmvc

解决方案

嗯，找了半天，发现应该在MockMVC里面加一个转换器。例如，您可以在此处阅读什么转换器。我的项目中已经有了它，但没有意识到它不适用于 MockMVC。

因此，您可以像这样将转换器添加到 MockMVC：

    @Autowired
    private StringToRoleConverter stringToRoleConverter;

    @Before
    public void init() {
        FormattingConversionService cs = new FormattingConversionService();
        cs.addConverter(stringToRoleConverter);

        mockMvc = MockMvcBuilders.standaloneSetup(userController)
                .setConversionService(cs)
                .build();
    }

转换器本身：

@Component
public class StringToRoleConverter implements Converter<String, Role> {

    @Autowired
    private RoleService roleService;

    @Override
    public Role convert(String id) {
        Role role = roleService.findById(Integer.valueOf(id));

        return role;
    }

}

然后我可以像这样添加参数：

mockMvc.perform(post("/users/edit/{id}", user.getId())
        .param("roles", "2"))

虽然我在那里传递了一个字符串，但它将在 Spring 转换器的帮助下转换为 Role。

推荐阅读