(spark) branch master updated: [MINOR][TESTS] Add a helper function for `spark.table` in dsl

[ruifengz]

This is an automated email from the ASF dual-hosted git repository.

ruifengz pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/spark.git


The following commit(s) were added to refs/heads/master by this push:
     new 4a471cceebed [MINOR][TESTS] Add a helper function for `spark.table` in 
dsl
4a471cceebed is described below

commit 4a471cceebedd938f781eb385162d33058124092
Author: Ruifeng Zheng <ruife...@apache.org>
AuthorDate: Thu May 23 19:46:46 2024 +0800

    [MINOR][TESTS] Add a helper function for `spark.table` in dsl
    
    ### What changes were proposed in this pull request?
    Add a helper function for `spark.table` in dsl
    
    ### Why are the changes needed?
    to be used in tests
    
    ### Does this PR introduce _any_ user-facing change?
    no, test only
    
    ### How was this patch tested?
    ci
    
    ### Was this patch authored or co-authored using generative AI tooling?
    no
    
    Closes #46717 from zhengruifeng/dsl_read.
    
    Authored-by: Ruifeng Zheng <ruife...@apache.org>
    Signed-off-by: Ruifeng Zheng <ruife...@apache.org>
---
 .../scala/org/apache/spark/sql/connect/dsl/package.scala  | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

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 a94bbf9c8f24..3edb63ee8e81 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
@@ -332,6 +332,21 @@ package object dsl {
       def sql(sqlText: String): Relation = {
         
Relation.newBuilder().setSql(SQL.newBuilder().setQuery(sqlText)).build()
       }
+
+      def table(name: String): Relation = {
+        proto.Relation
+          .newBuilder()
+          .setRead(
+            proto.Read
+              .newBuilder()
+              .setNamedTable(
+                proto.Read.NamedTable
+                  .newBuilder()
+                  .setUnparsedIdentifier(name)
+                  .build())
+              .build())
+          .build()
+      }
     }
 
     implicit class DslNAFunctions(val logicalPlan: Relation) {


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@spark.apache.org
For additional commands, e-mail: commits-h...@spark.apache.org