Add test for PDS-59284

apache · Dec 7, 2017 · 2535b21 · 2535b21
1 parent 8ae004b
commit 2535b21
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameSuite.scala
@@ -2228,4 +2228,12 @@ class DataFrameSuite extends QueryTest with SharedSQLContext {
     checkAnswer(df, Row(0, 10) :: Nil)
     assert(df.queryExecution.executedPlan.isInstanceOf[WholeStageCodegenExec])
   }
+
+  test("SPARK-ABC123: support select with a splatted stream") {
+    val df = spark.createDataFrame(sparkContext.emptyRDD[Row], StructType(List("bar", "foo").map {
+      StructField(_, StringType, false)
+    }))
+    val allColumns = Stream(df.col("bar"), col("foo"))
+    val result = df.select(allColumns : _*)
+  }
 }