It would be good to add unit tests that test the new logic you've added around these PutResults. If you look in `BlockManagerSuite` there are fairly extensive tests for the Block Manager. You should change the existing tests that do puts and cause things to flush out of memory to verify that the `droppedBlocks` gets updated. And also see if you can add new tests if those don't sufficiently cover the new code.