Spring MVC điều khiển ngoại lệ Kiểm tra

Question

Tôi có đoạn code sau

@RequestMapping(value = "admin/category/edit/{id}",method = RequestMethod.GET) 
public String editForm(Model model,@PathVariable Long id) throws NotFoundException{ 
    Category category=categoryService.findOne(id); 
    if(category==null){ 
     throw new NotFoundException(); 
    } 

    model.addAttribute("category", category); 
    return "edit"; 
} 

Tôi đang cố gắng để kiểm tra đơn vị khi NotFoundException được ném ra, vì vậy tôi viết mã như thế này

@Test(expected = NotFoundException.class) 
public void editFormNotFoundTest() throws Exception{ 

    Mockito.when(categoryService.findOne(1L)).thenReturn(null); 
    mockMvc.perform(get("/admin/category/edit/{id}",1L)); 
} 

Nhưng thất bại. Bất kỳ đề xuất nào về cách kiểm tra ngoại lệ?

Hoặc tôi nên ném ngoại lệ bên trong CategoryService vì vậy tôi có thể làm một cái gì đó như thế này

Mockito.when(categoryService.findOne(1L)).thenThrow(new NotFoundException("Message")); 

Answer 1

Cuối cùng tôi giải quyết nó. Vì tôi đang sử dụng thiết lập độc lập cho kiểm tra bộ điều khiển lò xo mvc nên tôi cần tạo HandlerExceptionResolver trong mọi kiểm tra đơn vị điều khiển cần thực hiện kiểm tra ngoại lệ.

mockMvc= MockMvcBuilders.standaloneSetup(adminCategoryController).setSingleView(view) 
      .setValidator(validator()).setViewResolvers(viewResolver()) 
      .setHandlerExceptionResolvers(getSimpleMappingExceptionResolver()).build(); 

Sau đó, các mã để kiểm tra

@Test 
public void editFormNotFoundTest() throws Exception{ 

    Mockito.when(categoryService.findOne(1L)).thenReturn(null); 
    mockMvc.perform(get("/admin/category/edit/{id}",1L)) 
      .andExpect(view().name("404s")) 
      .andExpect(forwardedUrl("/WEB-INF/jsp/404s.jsp")); 
}