Add example suite

Author: Matthis Gördel

sql/connect/server/src/test/scala/org/apache/spark/sql/connect/ExampleConnectSuite.scala

@@ -0,0 +1,22 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.spark.sql.connect
+
+import org.apache.spark.sql
+
+class ExampleConnectSuite extends sql.SparkSessionBinder

sql/core/src/test/scala/org/apache/spark/sql/ExampleSuite.scala

@@ -0,0 +1,47 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.spark.sql
+
+class ExampleSuite extends SparkSessionBinder {
+
+  test("replaceWhere with partitioned table preserves all partitions") {
+    withTable("foo") {
+      val data = Seq(
+        (1, "Alice", 29),
+        (2, "Bob", 35),
+        (3, "Charlie", 23),
+      )
+
+      val df = spark.createDataFrame(data).toDF("id", "name", "age")
+
+      df.write.partitionBy("age").format("delta").saveAsTable("foo")
+
+      val data1 = Seq((1, "Blice", 29))
+
+      val df1 = spark.createDataFrame(data1).toDF("id", "name", "age")
+
+      df1.write
+        .format("delta")
+        .option("replaceWhere", "age = 29")
+        .mode("overwrite")
+        .saveAsTable("foo")
+
+      assert(spark.sql("SHOW PARTITIONS foo").count() == 3)
+    }
+  }
+}