fix tests

apache · Jan 3, 2018 · 0059ff3 · 0059ff3
1 parent 971e7a4
commit 0059ff3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/streaming/DataStreamReader.scala b/sql/core/src/main/scala/org/apache/spark/sql/streaming/DataStreamReader.scala
@@ -167,8 +167,8 @@ final class DataStreamReader private[sql](sparkSession: SparkSession) extends Lo
       userSpecifiedSchema = userSpecifiedSchema,
       className = source,
       options = extraOptions.toMap)
-    val v1Relation = v1DataSource match {
-      case ds: StreamSourceProvider => Some(StreamingRelation(ds))
+    val v1Relation = ds match {
+      case _: StreamSourceProvider => Some(StreamingRelation(v1DataSource))
       case _ => None
     }
     ds match {

diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/streaming/RateSourceV2Suite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/streaming/RateSourceV2Suite.scala
@@ -59,6 +59,7 @@ class RateSourceV2Suite extends StreamTest {
   test("basic microbatch execution") {
     val input = spark.readStream
       .format("rateV2")
+      .option("numPartitions", "1")
       .option("rowsPerSecond", "10")
       .option("useManualClock", "true")
       .load()