block4.scala

class ExecutionListenerManagerSuite extends SparkFunSuite with LocalSparkSession {  
  
  import CountingQueryExecutionListener._  
  
  test("register query execution listeners using configuration") {  
    val conf = new SparkConf(false)  
      .set(QUERY_EXECUTION_LISTENERS, Seq(classOf[CountingQueryExecutionListener].getName()))  
    spark = SparkSession.builder().master("local").appName("test").config(conf).getOrCreate()  
  
    spark.sql("select 1").collect()  
    spark.sparkContext.listenerBus.waitUntilEmpty()  
    assert(INSTANCE_COUNT.get() === 1)  
    assert(CALLBACK_COUNT.get() === 1)  
  
    val cloned = spark.cloneSession()  
    cloned.sql("select 1").collect()  
    spark.sparkContext.listenerBus.waitUntilEmpty()  
    assert(INSTANCE_COUNT.get() === 1)  
    assert(CALLBACK_COUNT.get() === 2)  
  }