From 11581fbc725e5f4ce9c5f5314206605a005eaab5 Mon Sep 17 00:00:00 2001 From: Karol Sobczak Date: Wed, 20 Apr 2022 16:33:13 +0200 Subject: [PATCH] Fix running TestTableScanNodePartitioning test on single CPU nodes --- .../io/trino/sql/planner/TestTableScanNodePartitioning.java | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/core/trino-main/src/test/java/io/trino/sql/planner/TestTableScanNodePartitioning.java b/core/trino-main/src/test/java/io/trino/sql/planner/TestTableScanNodePartitioning.java index 0f588f202f38..4a1f065e7778 100644 --- a/core/trino-main/src/test/java/io/trino/sql/planner/TestTableScanNodePartitioning.java +++ b/core/trino-main/src/test/java/io/trino/sql/planner/TestTableScanNodePartitioning.java @@ -47,6 +47,7 @@ import java.util.Optional; import java.util.function.ToIntFunction; +import static io.trino.SystemSessionProperties.TASK_CONCURRENCY; import static io.trino.SystemSessionProperties.USE_TABLE_SCAN_NODE_PARTITIONING; import static io.trino.spi.connector.ConnectorBucketNodeMap.createBucketNodeMap; import static io.trino.spi.type.BigintType.BIGINT; @@ -78,11 +79,13 @@ public class TestTableScanNodePartitioning .setCatalog(MOCK_CATALOG) .setSchema(TEST_SCHEMA) .setSystemProperty(USE_TABLE_SCAN_NODE_PARTITIONING, "true") + .setSystemProperty(TASK_CONCURRENCY, "2") // force parallel plan even on test nodes with single CPU .build(); public static final Session DISABLE_PLAN_WITH_TABLE_NODE_PARTITIONING = testSessionBuilder() .setCatalog(MOCK_CATALOG) .setSchema(TEST_SCHEMA) .setSystemProperty(USE_TABLE_SCAN_NODE_PARTITIONING, "false") + .setSystemProperty(TASK_CONCURRENCY, "2") // force parallel plan even on test nodes with single CPU .build(); public static final int BUCKET_COUNT = 10; @@ -121,7 +124,8 @@ protected LocalQueryRunner createLocalQueryRunner() { Session.SessionBuilder sessionBuilder = testSessionBuilder() .setCatalog(MOCK_CATALOG) - .setSchema(TEST_SCHEMA); + .setSchema(TEST_SCHEMA) + .setSystemProperty(TASK_CONCURRENCY, "2"); // force parallel plan even on test nodes with single CPU LocalQueryRunner queryRunner = LocalQueryRunner.builder(sessionBuilder.build()) .withNodeCountForStats(10)