CNDB-15447: set CQL timeout equal to server timeout in AggregationQueriesTest (#2073)

### What is the issue

AggregationQueriesTest is flaky

### What does this PR fix and why was it fixed

Sets the CQL timeout equal to server timeout so the client doesn't
timeout before the server is done

 (Rebase of commit 9b3b3a3)

Author: driftx

test/unit/org/apache/cassandra/cql3/validation/operations/AggregationQueriesTest.java

@@ -29,6 +29,7 @@
 import org.junit.runner.RunWith;
 
 import com.datastax.driver.core.Row;
+import com.datastax.driver.core.SimpleStatement;
 import com.datastax.driver.core.exceptions.ReadTimeoutException;
 import org.apache.cassandra.config.DatabaseDescriptor;
 import org.apache.cassandra.config.DataStorageSpec;
@@ -156,7 +157,11 @@ public void testAggregationQueryShouldNotTimeoutWhenItExceedesReadTimeout() thro
             logger.info("Running aggregate, multi-page query");
 
             long queryStartTime = System.nanoTime();
-            List<Row> result = executeNet("SELECT a, count(c) FROM %s group by a").all();
+            // Use a custom read timeout that matches the aggregation timeout
+            // This ensures the driver won't timeout before the server completes the aggregation
+            SimpleStatement statement = new SimpleStatement(formatQuery("SELECT a, count(c) FROM %s group by a"));
+            statement.setReadTimeoutMillis(120000); // Match the server-side aggregation timeout
+            List<Row> result = sessionNet().execute(statement).all();
             long queryDuration = System.nanoTime() - queryStartTime;
 
             assertEquals("Should return 4 groups", 4, result.size());
@@ -268,4 +273,4 @@ public static void delayPageRead()
         Uninterruptibles.sleepUninterruptibly(delay, MILLISECONDS);
         logger.info("Resuming page read");
     }
-}
+}