在控制器上测试@ModelAttribute方法

时间:2014-02-13 20:46:01

标签: spring spring-mvc junit mockito spring-test

这是控制器中带注释的方法:

@RequestMapping(method = RequestMethod.GET)
public String getClient(@PathVariable("contractUuid") UUID contractUuid, Model model) {
    ClientDto clientDto = new ClientDto();
    clientDto.setContractUuid(contractUuid);
    model.addAttribute("client", clientDto);
    return "addClient";
}

@ModelAttribute("contract")
public ContractDto getContract(@PathVariable("contractUuid") UUID contractUuid) throws ContractNotFoundException {
    return contractService.fromEntity(contractService.findByUuid(contractUuid));
}

我正在尝试的测试方法如下所示,但属性合同失败。属性客户端已添加到 @RequestMapping 方法中的模型

private MockMvc mockMvc;
@Autowired
private ContractService contractServiceMock;
@Autowired
private ClientService clientServiceMock;
@Autowired
protected WebApplicationContext wac;

@Before
public void setup() {
    Mockito.reset(contractServiceMock);
    Mockito.reset(clientServiceMock);
    this.mockMvc = webAppContextSetup(this.wac).build();
}

@Test
public void test() throws Exception {
    UUID uuid = UUID.randomUUID();
    Contract contract = new Contract(uuid);

    when(contractServiceMock.findByUuid(uuid)).thenReturn(contract);

    mockMvc.perform(get("/addClient/{contractUuid}", uuid))
            .andExpect(status().isOk())
            .andExpect(view().name("addClient"))
            .andExpect(forwardedUrl("/WEB-INF/pages/addClient.jsp"))
            .andExpect(model().attributeExists("client"))
            .andExpect(model().attributeExists("contract"));
}

当我运行应用程序时,契约属性显示在jsp页面中,因为我使用了它的一些属性,但是因为它在测试方法中失败了还有另一种测试方法吗? 它失败并显示以下消息:

  

java.lang.AssertionError:模型属性“合同”不存在

Spring是4.0.1.RELEASE

1 个答案:

答案 0 :(得分:1)

这似乎是我的错。 即使 @ModelAttribute 方法返回 ContractDto 的实例,我只模拟了从服务中使用的一个方法:

when(contractServiceMock.findByUuid(uuid)).thenReturn(contract);

所以 findByUuid 返回了一些内容,但 contractService.fromEntity 保持不变,所以我还要嘲笑它:

UUID uuid = UUID.randomUUID();
    Contract contract = new Contract(uuid);
    ContractDto contractDto = new ContractDto(uuid);

    when(contractServiceMock.findByUuid(uuid)).thenReturn(contract);
    when(contractServiceMock.fromEntity(contract)).thenReturn(contractDto);