This is an automated email from the ASF dual-hosted git repository. ankitsultana pushed a commit to branch master in repository https://gitbox.apache.org/repos/asf/pinot.git
The following commit(s) were added to refs/heads/master by this push: new 246b98c7e1 [flaky-test] Increase Timeouts for TPCHQueryIntegrationTest and MultiStageEngineIntegrationTest (#14653) 246b98c7e1 is described below commit 246b98c7e13de7de6efaca5694fae4bb994f005b Author: Ankit Sultana <ankitsult...@uber.com> AuthorDate: Tue Dec 17 12:58:07 2024 -0600 [flaky-test] Increase Timeouts for TPCHQueryIntegrationTest and MultiStageEngineIntegrationTest (#14653) --- .../integration/tests/MultiStageEngineIntegrationTest.java | 11 +++++++++++ .../integration/tests/tpch/TPCHQueryIntegrationTest.java | 12 ++++++++++++ 2 files changed, 23 insertions(+) diff --git a/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/MultiStageEngineIntegrationTest.java b/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/MultiStageEngineIntegrationTest.java index bc19bace53..0b15d38dce 100644 --- a/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/MultiStageEngineIntegrationTest.java +++ b/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/MultiStageEngineIntegrationTest.java @@ -27,6 +27,7 @@ import java.time.ZoneId; import java.time.format.DateTimeFormatter; import java.util.ArrayList; import java.util.Collections; +import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.concurrent.TimeUnit; @@ -109,6 +110,16 @@ public class MultiStageEngineIntegrationTest extends BaseClusterIntegrationTestS setupTableWithNonDefaultDatabase(avroFiles); } + @Override + protected Map<String, String> getExtraQueryProperties() { + // Increase timeout for this test since it keeps failing in CI. + Map<String, String> timeoutProperties = new HashMap<>(); + timeoutProperties.put("brokerReadTimeoutMs", "120000"); + timeoutProperties.put("brokerConnectTimeoutMs", "60000"); + timeoutProperties.put("brokerHandshakeTimeoutMs", "60000"); + return timeoutProperties; + } + private void setupTableWithNonDefaultDatabase(List<File> avroFiles) throws Exception { _tableName = TABLE_NAME_WITH_DATABASE; diff --git a/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/tpch/TPCHQueryIntegrationTest.java b/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/tpch/TPCHQueryIntegrationTest.java index ee82a96931..0288e61693 100644 --- a/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/tpch/TPCHQueryIntegrationTest.java +++ b/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/tpch/TPCHQueryIntegrationTest.java @@ -27,6 +27,8 @@ import java.nio.charset.Charset; import java.sql.ResultSet; import java.sql.Statement; import java.util.Collections; +import java.util.HashMap; +import java.util.Map; import java.util.Objects; import java.util.Set; import org.apache.commons.collections4.CollectionUtils; @@ -165,6 +167,16 @@ public class TPCHQueryIntegrationTest extends BaseClusterIntegrationTest { return true; } + @Override + protected Map<String, String> getExtraQueryProperties() { + // Increase timeout for this test since it keeps failing in CI. + Map<String, String> timeoutProperties = new HashMap<>(); + timeoutProperties.put("brokerReadTimeoutMs", "120000"); + timeoutProperties.put("brokerConnectTimeoutMs", "60000"); + timeoutProperties.put("brokerHandshakeTimeoutMs", "60000"); + return timeoutProperties; + } + @AfterClass public void tearDown() throws Exception { --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org For additional commands, e-mail: commits-h...@pinot.apache.org