update DataSourceV2UtilsSuite

apache · Dec 15, 2017 · f7d5a4d · f7d5a4d
1 parent d964158
commit f7d5a4d
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/sources/v2/DataSourceV2UtilsSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/sources/v2/DataSourceV2UtilsSuite.scala
@@ -17,15 +17,13 @@
 
 package org.apache.spark.sql.sources.v2
 
-import org.apache.spark.sql.{QueryTest, SparkSession}
+import org.apache.spark.SparkFunSuite
 import org.apache.spark.sql.execution.datasources.v2.DataSourceV2Utils
 import org.apache.spark.sql.internal.SQLConf
 
-class DataSourceV2UtilsSuite extends QueryTest {
+class DataSourceV2UtilsSuite extends SparkFunSuite {
 
-  protected var spark: SparkSession = null
-
-  private val keyPrefix = "userDefinedDataSource"
+  private val keyPrefix = new DataSourceV2WithSessionConfig().keyPrefix
 
   test("method withSessionConfig() should propagate session configs correctly") {
     // Only match configs with keys start with "spark.datasource.${keyPrefix}".