diff --git a/pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/minion/PinotTaskManager.java b/pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/minion/PinotTaskManager.java index f4460f56e85f..ce986dd5addf 100644 --- a/pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/minion/PinotTaskManager.java +++ b/pinot-controller/src/main/java/org/apache/pinot/controller/helix/core/minion/PinotTaskManager.java @@ -780,8 +780,7 @@ public Map scheduleTasks(TaskSchedulingContext conte * If no specific tables are targeted (i.e., "schedule for every table"), the check iterates the full * table list so that per-table opt-outs are still honored. */ - @VisibleForTesting - boolean shouldUseConcurrentPath(TaskSchedulingContext context) { + protected boolean shouldUseConcurrentPath(TaskSchedulingContext context) { Set targetTables = context.getTablesToSchedule(); Set targetDatabases = context.getDatabasesToSchedule(); Set consolidatedTables = new HashSet<>(); @@ -814,8 +813,7 @@ boolean shouldUseConcurrentPath(TaskSchedulingContext context) { * Resolves the effective concurrent-scheduling flag for a single table: table-level override if * set, otherwise the cluster-level default. */ - @VisibleForTesting - boolean resolveConcurrentScheduling(TableConfig tableConfig) { + protected boolean resolveConcurrentScheduling(TableConfig tableConfig) { TableTaskConfig taskConfig = tableConfig.getTaskConfig(); if (taskConfig != null) { Boolean tableFlag = taskConfig.getConcurrentSchedulingEnabled();