We need ```batch_size``` to more than actual records count to test this behaviour, because before this path exception occurs only on second iteration. Old version of test with ```batch_size``` make more that 1 iteration and gives false positive result.

In my fixed version with ```batch_size``` more that actual records it fails even if only 1 iteration is done.