Scala：テストが終了したらどのようにアクションを実行できますか？

Question

私はscalatest_2.11バージョン2.2.1を使用しています。私はSparkContextで実行されるテストを書くことを試みています。テストの開始時にSparkContextを起動し、このScをすべてのテストに使用して、すべて完了したら停止するにはどうすればよいですか？

（私はそれ自体が停止すると仮定していることを知っているが、I'sdはまだそれを自分でやりたい）

Answer 1

あなたはScalaTestからBeforeAndAfterAllを使用することができます。 SparkContextを開始および停止し、他のテストで使用する基本特性を定義します。

trait SparkTest extends BeforeAndAfterAll { 
    self: Suite => 

    @transient var sc: SparkContext = _ 

    override def beforeAll { 
    val conf = new SparkConf(). 
     setMaster("local[*]"). 
     setAppName("test") 
    sc = new SparkContext(conf) 
    super.beforeAll() 
    } 

    override def afterAll: Unit = { 
    try { 
     sc.stop() 
    } finally { 
     super.afterAll 
    } 
    } 

} 

// Mix-in the trait with your tests like below. 
class MyTest extends FunSuite with SparkTest { 
    test("my test") { 
    // you can access "sc" here. 
    } 
}