Как в Mockito проверять на исключения?

Есть метод

public ResponseEntity<?> deleteMessage(@PathVariable Long id) {
    if (!messageService.delete(id)) {
        throw  new DataNotFoundException("Data with id=" + id + " not found.");
    }
    return new ResponseEntity<>(HttpStatus.OK);
}

И тест к нему

@Test
public void deleteMessageIfNotFoundThenException() {
    //prepare
    when(messageController.deleteMessage(2L)).thenThrow(DataNotFoundException.class);
    //testing
    ResponseEntity<?> failedDelete = messageController.deleteMessage(2L);
    //validate
    verify(messageController).deleteMessage(2L);
}

Ну пойму как правильно проверять, тест не проходит.

И DataNotFoundException

public class DataNotFoundException extends RuntimeException {
    public DataNotFoundException(String message) {
        super(message);
    }
}