JUnit + Mockito тестирование исключения

Контроллер

@RequestMapping(
        value = "api/message/{id}",
        method = RequestMethod.DELETE)
public ResponseEntity<?> deleteMessage(@PathVariable Long id) {
    messageService.delete(id);
    return new ResponseEntity<>(HttpStatus.OK);
}

Нужно проверить на исключение если запросить удаление несуществующего id.

Возвращает

{
"timestamp": 1483566196066,
"status": 500,
"error": "Internal Server Error",
"exception": "org.springframework.dao.EmptyResultDataAccessException",
"message": "No class com.sttech.springrest.model.Message entity with id 1000 exists!",
"path": "/api/message/1000"
}

Тест метод

@Test(expected = EmptyResultDataAccessException.class)
    public void deleteMessageIfNotFoundThenException() {
    messageController.deleteMessage(2L);
}

Возвращает ошибку

java.lang.AssertionError: Expected exception: org.springframework.dao.EmptyResultDataAccessException

Если так:

@Test(expected = EmptyResultDataAccessException.class)
public void deleteMessageIfNotFoundThenException1() {
    when(messageController.deleteMessage(2L));
    then(caughtException())
            .isInstanceOf(EmptyResultDataAccessException.class)
            .hasMessageContaining("entity with id 2 exists!");
}

Возвращает ошибку

java.lang.Exception: Unexpected exception, expected<org.springframework.dao.EmptyResultDataAccessException> but was<java.lang.AssertionError>

Вообще это исключение надо тестировать в контроллере?