From 593eb9b8e7d2dfadde18874992f4523f7cfd9fa3 Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Wed, 6 May 2026 14:44:53 +0000 Subject: [PATCH] [MINOR][SQL][TESTS] Inline unnecessary asInstanceOf[classic.SparkSession] cast in QueryTest --- .../src/test/scala/org/apache/spark/sql/QueryTest.scala | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/sql/core/src/test/scala/org/apache/spark/sql/QueryTest.scala b/sql/core/src/test/scala/org/apache/spark/sql/QueryTest.scala index 291aa7cab7256..8df3bb3a687e1 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/QueryTest.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/QueryTest.scala @@ -1073,14 +1073,13 @@ object QueryTest extends Assertions { } } - val classicSession = spark.asInstanceOf[classic.SparkSession] - classicSession.sparkContext.listenerBus.waitUntilEmpty(15000) - classicSession.listenerManager.register(listener) + spark.sparkContext.listenerBus.waitUntilEmpty(15000) + spark.listenerManager.register(listener) try { thunk - classicSession.sparkContext.listenerBus.waitUntilEmpty(15000) + spark.sparkContext.listenerBus.waitUntilEmpty(15000) } finally { - classicSession.listenerManager.unregister(listener) + spark.listenerManager.unregister(listener) } capturedQueryExecutions