/** * Restore each store in taskStoresToRestore sequentially */ public void restoreStores() { LOG.debug("Restoring stores for task: {}", taskModel.getTaskName()); for (String storeName : taskStoresToRestore) { SystemConsumer systemConsumer = systemConsumers.get(storeName); SystemStream systemStream = changelogSystemStreams.get(storeName); SystemStreamPartitionIterator systemStreamPartitionIterator = new SystemStreamPartitionIterator(systemConsumer, new SystemStreamPartition(systemStream, taskModel.getChangelogPartition())); taskStores.get(storeName).restore(systemStreamPartitionIterator); } }
storeRestoreCallCount++; return null; }).when(mockStorageEngine).restore(Mockito.any());