From 15fb4787354a2d5dc97afb31010beb1f3cc3b73d Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Fri, 17 May 2024 18:06:25 +0800 Subject: [PATCH] [SPARK-48321][CONNECT][TESTS] Avoid using deprecated methods in dsl ### What changes were proposed in this pull request? Avoid using deprecated methods in dsl ### Why are the changes needed? `putAllRenameColumnsMap` was deprecated ### Does this PR introduce _any_ user-facing change? no ### How was this patch tested? ci ### Was this patch authored or co-authored using generative AI tooling? no Closes #46635 from zhengruifeng/with_col_rename_dsl. Authored-by: Ruifeng Zheng Signed-off-by: Ruifeng Zheng --- .../scala/org/apache/spark/sql/connect/dsl/package.scala | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/connector/connect/server/src/test/scala/org/apache/spark/sql/connect/dsl/package.scala b/connector/connect/server/src/test/scala/org/apache/spark/sql/connect/dsl/package.scala index da9a0865b8ca6..b50c04e7540f5 100644 --- a/connector/connect/server/src/test/scala/org/apache/spark/sql/connect/dsl/package.scala +++ b/connector/connect/server/src/test/scala/org/apache/spark/sql/connect/dsl/package.scala @@ -1019,7 +1019,13 @@ package object dsl { WithColumnsRenamed .newBuilder() .setInput(logicalPlan) - .putAllRenameColumnsMap(renameColumnsMap.asJava)) + .addAllRenames(renameColumnsMap.toSeq.map { case (k, v) => + WithColumnsRenamed.Rename + .newBuilder() + .setColName(k) + .setNewColName(v) + .build() + }.asJava)) .build() }