public void awaitRecords(long timeoutMillis) throws InterruptedException {
        awaitRecords(timeoutMillis, TimeUnit.MILLISECONDS);
    }
Wait up to the specified number of milliseconds for this task to meet the expected number of
records as defined by {@code expectedRecords}.

@param timeoutMillis number of milliseconds to wait for records
@throws InterruptedException if another threads interrupts this one while waiting for records
connect/runtime/src/test/java/org/apache/kafka/connect/integration/TaskHandle.java