Fix max_iterations test

Author: tomusher

tests/unit/contrib/index/test_query_handler.py

@@ -312,7 +312,7 @@ def test_respects_max_iterations(self):
 
         qs = self.make_qs(documents, overfetch_multiplier=1, max_overfetch_iterations=2)
         qs = qs[0:10]
-        qs._fetch_batch = mock.Mock(return_value=[documents[0]])
+        qs._fetch_batch = mock.Mock(return_value=documents)
 
         list(qs.run_query())