File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
sql/core/src/test/scala/org/apache/spark/sql/jdbc Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -223,7 +223,9 @@ class JDBCSuite extends SparkFunSuite
223223 val parentPlan = df.queryExecution.executedPlan
224224 // Check if SparkPlan Filter is not removed in a physical plan because JDBCRDD
225225 // cannot compile given predicates.
226- assert(parentPlan.isInstanceOf [org.apache.spark.sql.execution.Filter ])
226+ assert(parentPlan.isInstanceOf [org.apache.spark.sql.execution.WholeStageCodegen ])
227+ val node = parentPlan.asInstanceOf [org.apache.spark.sql.execution.WholeStageCodegen ]
228+ assert(node.plan.isInstanceOf [org.apache.spark.sql.execution.Filter ])
227229 df
228230 }
229231 assert(checkNotPushdown(sql(" SELECT * FROM foobar WHERE (THEID + 1) < 2" )).collect().size == 0 )
You can’t perform that action at this time.
0 commit comments