Mockito 并期望异常

Question

我从我的控制器创建测试，它必须将重定向 url 返回到“/register/duplicate”。但是当我执行该操作时，我会遇到异常，例如 DataIntegrityViolationException。控制器应该捕捉到这个异常，但没有捕捉到

它是如何工作的？我能做的，对控制器来说，发生了一个异常并以“catch”原因执行代码，将重定向到/register/duplicate

测试：

@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)