Mockito и ожидаемое исключение

Question

Я создаю тест из своего контроллера и должен вернуть перенаправляющий URL-адрес в «/ register/duplicate». Но когда я выполняю это, я получаю исключения, такие как DataIntegrityViolationException. Контроллер должен поймать это исключение, но не поймает

Как это работает? А что я могу сделать, чтобы контроллер исключение произошло и выполняет код в «поймать» причину, будет перенаправление/регистрации/дублировать

Тест:

@Test 
public void testRegisterPageLogic_Duplicate() throws Exception { 
    expectedUser = new User("Jorg", "Brush"); 
    UserService mockService = mock(UserService.class); 

    userService.add(expectedUser); 

    when(mockService.add(expectedUser)).thenCallRealMethod(); 

    registerController = new RegisterController(mockService); 
    mockMvc = standaloneSetup(registerController).build(); 

    mockMvc.perform(post("/register") 
      .param("hide", "up") 
      .param("username", expectedUser.getUsername()) 
      .param("password", expectedUser.getPassword())) 
      .andExpect(redirectedUrl("/register/duplicate")); 

    Mockito.verify(mockService, atLeastOnce()).add(expectedUser); 

    userService.remove(userService.findUser("Jorg").getUserId()); 
} 

Контроллера метод:

@RequestMapping(method = POST) 
public String register(@RequestParam("hide") String hide, 
         @RequestParam("username") String username, 
         @RequestParam("password") String password) { 

    if (hide.equals("up")) { 
     try { 
      userService.add(new User(username, password)); 
     } catch (DataIntegrityViolationException e) { 
      return "redirect:/register/duplicate"; 
     } 
    } 
    User user = userService.findUser(username); 
    if (user == null) { 
     return "redirect:/register/notExists"; 
    } 
    UserSession.setUser(user); 
    UserSession.setId(user.getUserId()); 
    return "redirect:/notes/" + UserSession.getUser().getUsername(); 
} 

Answer 1

Давайте попробуем упростить, если код:

try { 
    userService.add(new User(username, password)); 
} catch (DataIntegrityViolationException e) { 
    return "redirect:/register/NOT-OK"; 
} 
return "redirect:/register/OK"; 

И вы хотите, чтобы проверить как ветви, затем

@RunWith(MockitoJUnitRunner.class) // for @Mock 
class TheTest { 

    @Mock 
    UserService mockService; 

    MockMvc mockMvc;  

    @Before 
    public void setup() {   
     RegisterController registerController = new RegisterController(mockService); 
     mockMvc = standaloneSetup(registerController).build(); 
    } 

    @Test 
    public void testOK() throw Exception { 
     // mockService does nothing by default 

     mockMvc.perform(post("/register")) 
       .andExpect(view().name("/register/OK")); 

     Mockito.verify(mockService).add(Matchers.any(User.class)); 
    } 

    @Test 
    public void testNotOk() { 
     Mockito.when(mockService.add(Matchers.any(User.class)) 
       .thenThrow(new DataIntegrityViolationException()); // adjust DataIntegrityViolationException constructor args if required. 

     // now mockService throw a DataIntegrityViolationException when mockService#add(User) is called 

     mockMvc.perform(post("/register")) 
       .andExpect(view().name("/register/NOT-OK")); 

     Mockito.verify(mockService).add(Matchers.any(User.class)); 
    } 
} 

Answer 2

Чтобы повысить исключение в Mockito, вам не нужно вызывать реальный метод. Вы можете использовать этот код:

when(mockService.add(expectedUser)).thenThrow(DataIntegrityViolationException.class)