Модульное тестирование InterruptedException для ThreadPoolTaskExecutor с Mockito Spy

Question

Я использую весну ThreadPoolTaskExecutor для выполнения задач в несколько потоков. Класс выглядит как ниже

    @Component
    public class LoadData {
    //... ...    

        @Inject
        private ThreadPoolTaskExecutor taskExecutor;

        public SomeData getData(Long id) {
            Future<SomeData> loadData = taskExecutor.submit(() -> {
                    //return methodToGetDataSynchronously(id);
                    return new SomeData();
            });
            try {
                SomeData data = loadData.get();
            } catch (InterruptedException | ExecutionException e) {
                logger.error("error");
                //some more processing for the error here
            }
            return data;
       }
  }

Чтобы иметь возможность провести модульное тестирование этого класса и охватить InterruptedException а также ExecutionException Я попробовал несколько подходов (используя MockitoSpy), но не смогли успешно проверить это.

Класс юнит-теста выглядит так:

@RunWith(MockitoJUnitRunner.class)
public class LoadDataTest {

    @InjectMocks
    private LoadData loadData;

    @Spy
    private ThreadPoolTaskExecutor spyTaskExecutor = new ThreadPoolTaskExecutor();

    @Before
    public void init() {
        MockitoAnnotations.initMocks(this);
        spyTaskExecutor.setCorePoolSize(1);
        spyTaskExecutor.setWaitForTasksToCompleteOnShutdown(true);
        spyTaskExecutor.initialize();
    }

    @Test
    public void testGetData_shouldThrowInterruptedException () {
        Mockito.when(spyTaskExecutor.submit(Matchers.<Callable<SomeData>>any())).thenAnswer(new Answer<Future<SomeData>>() {
            public Future<SomeData> answer(InvocationOnMock invocation) throws Throwable {
                Future<SomeData> future = Mockito.mock(FutureTask.class);
                when(future.isDone()).thenReturn(false, false, true);
                when(future.get()).thenThrow(new InterruptedException ());
                return future;
            }
        });

        SomeData result = null;
        result = loadData.getData(101L);
        //verify here that InterruptedException processing was performed
    }
}

я использую Spy за ThreadPoolTaskExecutor как у меня есть другие методы в тесте, который проверяет фактическое поведение многопоточного выполнения. Когда я пытаюсь запустить тестовый метод, он бросает NullPointerException на шпионаже

Mockito.when(spyTaskExecutor.submit(Matchers.<Callable<SomeData>>any()))...

Исключение:

java.lang.NullPointerException
    at java.util.concurrent.AbstractExecutorService.submit(AbstractExecutorService.java:132)
    at org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor.submit(ThreadPoolTaskExecutor.java:320)
at com.company.LoadDataTest.testGetData_shouldThrowInterruptedException(LoadDataTest.java:20)

Я думаю, проблема может быть в том, как я заглушаю шпиона Мокито. Тем не менее, я не уверен, что нахожусь на правильном пути здесь, чтобы заглушить InterruptedException, У кого-нибудь есть образец, как этого добиться?

Версии:

  jdk-1.8.0_72
  junit-4.12
  mockito-core-1.10.19
  spring-context-4.3.4.RELEASE

Примечание: это фрагмент действующего кода. Фактический код выполняет много других вещей и имеет много других bean-компонентов, некоторые из которых Mock в Test, некоторые Spy - в зависимости от того, что я тестирую. Я упомянул только ту часть, с которой я сталкиваюсь здесь.

Answer 1

Наконец-то разобрался с вопросом. Я был очень близок, просто использование Mockito Spy было немного неприемлемым. Вот рабочее решение:

@RunWith(MockitoJUnitRunner.class)
public class LoadDataTest {

    @InjectMocks
    private LoadData loadData;

    @Spy
    private ThreadPoolTaskExecutor spyTaskExecutor = new ThreadPoolTaskExecutor();

    @Before
    public void init() {
        MockitoAnnotations.initMocks(this);
        spyTaskExecutor.setCorePoolSize(1);
        spyTaskExecutor.setWaitForTasksToCompleteOnShutdown(true);
        spyTaskExecutor.initialize();
    }

    @Test
    public void testGetData_shouldThrowInterruptedException () {
        setupSpyTaskExecutorForException(new InterruptedException("junit"));

        SomeData result = loadData.getData(101L);
        //verify here that InterruptedException processing was performed
    }

    private void setupSpyTaskExecutorForException(Exception e) {
        Mockito.doAnswer(new Answer<Future<?>>() {
            public Future<?> answer(InvocationOnMock invocation) throws Throwable {
                Future<?> future = Mockito.mock(FutureTask.class);
                when(future.get()).thenThrow(e);
                return future;
            }
        }).when(spyTaskExecutor).submit(Matchers.<Callable<?>>any());
    }
}