@Override
  public void checkDone() throws IllegalStateException {
    boolean done =
        shouldStop || streamProgress.getCloseStream() != null || streamProgress.isFailToLock();
    Preconditions.checkState(done, "There's more work to be done");
  }
@Test
  public void testDoneOnFailToLockTrue() {
    StreamProgress streamProgress = new StreamProgress();
    streamProgress.setFailToLock(true);
    ReadChangeStreamPartitionProgressTracker tracker =
        new ReadChangeStreamPartitionProgressTracker(streamProgress);
    tracker.checkDone();
  }