Mockito тестировние void метода

Тестируемый метод

public boolean delete(Long id) {
    Message message = messageRepository.findOne(id);
    if (message == null) {
        throw  new DataNotFoundException("Data with id=" + id + " not found.");
    }
    messageRepository.delete(id);
    return true;
}

Тест:

@Test
public void delete() throws Exception {
    //prepare
    MessageServiceImp spy = spy(new MessageServiceImp(messageRepository));
    messageServiceImp.save(MESSAGE); //создаю объект, пробовал и не создавать результат одинаков
    doReturn(true).when(spy).delete(ID);
    //testing
    boolean testMessage = messageServiceImp.delete(ID);
    //validate
    verify(messageRepository).delete(ID);
}

Тест не проходит получаю:

.exception.DataNotFoundException: Data with id=1 not found.

Что делаю не так?