@Override
        public void onRestoreEnd(final TopicPartition topicPartition,
                                 final String storeName,
                                 final long totalRestored) {
            if (restored != null) {
                restored.addAndGet(totalRestored);
            }
        }
Tracks the offsets and number of restored records on a per-partition basis.
Currently assumes only one store in the topology; you will need to update this
if it's important to track across multiple stores in a topology
streams/src/test/java/org/apache/kafka/streams/integration/utils/IntegrationTestUtils.java