Reflect review comment

Author: HeartSaVioR

sql/hive-thriftserver/src/test/scala/org/apache/spark/sql/hive/thriftserver/SparkSQLEnvSuite.scala

@@ -17,19 +17,16 @@
 
 package org.apache.spark.sql.hive.thriftserver
 
-import test.custom.listener.{DummyQueryExecutionListener, DummyStreamingQueryListener}
-
 import org.apache.spark.SparkFunSuite
 import org.apache.spark.launcher.SparkLauncher
 import org.apache.spark.sql.hive.HiveUtils.{HIVE_METASTORE_JARS, HIVE_METASTORE_VERSION}
 import org.apache.spark.sql.hive.test.TestHiveContext
+import org.apache.spark.sql.hive.thriftserver.__root__.test.custom.listener.{DummyQueryExecutionListener, DummyStreamingQueryListener}
 import org.apache.spark.sql.internal.StaticSQLConf.{QUERY_EXECUTION_LISTENERS, STREAMING_QUERY_LISTENERS, WAREHOUSE_PATH}
 
 class SparkSQLEnvSuite extends SparkFunSuite {
   test("SPARK-29604 external listeners should be initialized with Spark classloader") {
     withSystemProperties(
-      // Intentionally place listeners to the out of spark package, because IsolatedClientLoader
-      // leverages Spark classloader for shared classess including spark package.
       QUERY_EXECUTION_LISTENERS.key -> classOf[DummyQueryExecutionListener].getCanonicalName,
       STREAMING_QUERY_LISTENERS.key -> classOf[DummyStreamingQueryListener].getCanonicalName,
       WAREHOUSE_PATH.key -> TestHiveContext.makeWarehouseDir().toURI.getPath,
@@ -68,3 +65,27 @@ class SparkSQLEnvSuite extends SparkFunSuite {
     }
   }
 }
+
+/**
+ * These classes in this package are intentionally placed to the outer package of spark,
+ * because IsolatedClientLoader leverages Spark classloader for shared classess including
+ * spark package, and the test should fail if Spark initializes these listeners with
+ * IsolatedClientLoader.
+ */
+package __root__.test.custom.listener {
+
+  import org.apache.spark.sql.execution.QueryExecution
+  import org.apache.spark.sql.streaming.StreamingQueryListener
+  import org.apache.spark.sql.util.QueryExecutionListener
+
+  class DummyQueryExecutionListener extends QueryExecutionListener {
+    override def onSuccess(funcName: String, qe: QueryExecution, durationNs: Long): Unit = {}
+    override def onFailure(funcName: String, qe: QueryExecution, error: Throwable): Unit = {}
+  }
+
+  class DummyStreamingQueryListener extends StreamingQueryListener {
+    override def onQueryStarted(event: StreamingQueryListener.QueryStartedEvent): Unit = {}
+    override def onQueryProgress(event: StreamingQueryListener.QueryProgressEvent): Unit = {}
+    override def onQueryTerminated(event: StreamingQueryListener.QueryTerminatedEvent): Unit = {}
+  }
+}