ISPN-15738 Test max results with distributed iterators

In particular if the state transfer chunk size is larger and smaller than the max results

query/src/main/java/org/infinispan/query/clustered/ClusteredQueryInvoker.java

@@ -65,12 +65,12 @@ List<QueryResponse> broadcast(ClusteredQueryOperation operation) {
                BitSet segments = e.getValue();
                SegmentsClusteredQueryCommand cmd = new SegmentsClusteredQueryCommand(cache.getName(), operation, segments);
                return rpcManager.invokeCommand(address, cmd, SingleResponseCollector.validOnly(),
-                                               rpcManager.getSyncRpcOptions()).toCompletableFuture();
+                     rpcManager.getSyncRpcOptions()).toCompletableFuture();
             }).map(a -> a.thenApply(r -> (QueryResponse) r.getResponseValue())).collect(Collectors.toList());
 
       // then, invoke on own node
       CompletionStage<QueryResponse> localResponse = localInvoke(localCommand);
-      
+
       List<QueryResponse> results = new ArrayList<>();
       try {
          results.add(await(localResponse.toCompletableFuture()));

query/src/test/java/org/infinispan/query/maxresult/DistributedMaxResultTest.java

@@ -0,0 +1,58 @@
+package org.infinispan.query.maxresult;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.infinispan.configuration.cache.IndexStorage.LOCAL_HEAP;
+
+import org.infinispan.Cache;
+import org.infinispan.configuration.cache.CacheMode;
+import org.infinispan.configuration.cache.ConfigurationBuilder;
+import org.infinispan.query.Search;
+import org.infinispan.query.dsl.Query;
+import org.infinispan.query.dsl.QueryFactory;
+import org.infinispan.query.dsl.QueryResult;
+import org.infinispan.query.model.Game;
+import org.infinispan.test.MultipleCacheManagersTest;
+import org.testng.annotations.Test;
+
+@Test(groups = "functional", testName = "query.maxresult.DistributedMaxResultTest")
+public class DistributedMaxResultTest extends MultipleCacheManagersTest {
+
+   private Cache<Integer, Game> node1;
+
+   @Override
+   protected void createCacheManagers() throws Throwable {
+      ConfigurationBuilder config = getDefaultClusteredCacheConfig(CacheMode.DIST_SYNC, false);
+      config
+            .clustering()
+               .hash().numOwners(2)
+               .stateTransfer().chunkSize(100)
+            .indexing().enable()
+               .storage(LOCAL_HEAP)
+               .addIndexedEntity("org.infinispan.query.model.Game")
+            .query().defaultMaxResults(50);
+
+      createClusteredCaches(2, config);
+      node1 = cache(0);
+      cache(1);
+   }
+
+   public void executeSmokeTest() {
+      for (int i = 1; i <= 110; i++) {
+         node1.put(i, new Game("Game " + i, "This is the game " + i + "# of a series"));
+      }
+
+      QueryFactory factory = Search.getQueryFactory(node1);
+      Query<Game> query = factory.create("from org.infinispan.query.model.Game");
+      QueryResult<Game> result = query.execute();
+
+      assertThat(result.hitCount()).hasValue(110);
+      assertThat(result.list()).hasSize(50); // use custom default
+
+      query = factory.create("from org.infinispan.query.model.Game");
+      query.maxResults(200); // raise it
+      result = query.execute();
+
+      assertThat(result.hitCount()).hasValue(110);
+      assertThat(result.list()).hasSize(110);
+   }
+}