// remove a split and verify
assertEquals(hiveSplitSource.getNextBatch(1).size(), 1);
assertEquals(hiveSplitSource.getOutstandingSplitCount(), 4);
// fail source
hiveSplitSource.fail(new RuntimeException("test"));
assertEquals(hiveSplitSource.getOutstandingSplitCount(), 4);
// try to remove a split and verify we got the expected exception
try {
hiveSplitSource.getNextBatch(1);