Avoid to extend arguments fully

apache · Jan 22, 2019 · 4dc4a2a · 4dc4a2a
1 parent c6df730
commit 4dc4a2a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/jdbc/JDBCSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/jdbc/JDBCSuite.scala
@@ -1544,9 +1544,10 @@ class JDBCSuite extends QueryTest
             .load()
 
           df.logicalPlan match {
-            case LogicalRelation(JDBCRelation(_, parts, _), _, _, _) =>
-              val whereClauses = parts.map(_.asInstanceOf[JDBCPartition].whereClause).toSet
-              assert(whereClauses === Set(
+            case lr: LogicalRelation if lr.relation.isInstanceOf[JDBCRelation] =>
+              val jdbcRelation = lr.relation.asInstanceOf[JDBCRelation]
+              val whereClauses = jdbcRelation.parts.map(_.asInstanceOf[JDBCPartition].whereClause)
+              assert(whereClauses.toSet === Set(
                 s""""T" < '$middle' or "T" is null""",
                 s""""T" >= '$middle'"""))
           }