diff --git a/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy b/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy index 89bf5a4eba041b..b04cb117e01a57 100644 --- a/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy +++ b/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy @@ -1138,6 +1138,8 @@ class Suite implements GroovyInterceptable { tupleResult = JdbcUtils.executeToStringList(context.getArrowFlightSqlConnection(), (PreparedStatement) arg) } else if (tag.contains("target_sql")) { tupleResult = JdbcUtils.executeToStringList(context.getTargetConnection(this), (PreparedStatement) arg) + } else if (tag.contains("master_sql")) { + tupleResult = JdbcUtils.executeToStringList(context.getMasterConnection(), (PreparedStatement) arg) } else { tupleResult = JdbcUtils.executeToStringList(context.getConnection(), (PreparedStatement) arg) } @@ -1151,6 +1153,8 @@ class Suite implements GroovyInterceptable { (String) ("USE ${context.dbName};" + (String) arg)) } else if (tag.contains("target_sql")) { tupleResult = JdbcUtils.executeToStringList(context.getTargetConnection(this), (String) arg) + } else if (tag.contains("master_sql")) { + tupleResult = JdbcUtils.executeToStringList(context.getMasterConnection(), (String) arg) } else { tupleResult = JdbcUtils.executeToStringList(context.getConnection(), (String) arg) }