SparkTest

package spark

import org.apache.spark.sql.SparkSession
import org.junit.{AfterClass, BeforeClass}

abstract class SparkTest {

  val spark: SparkSession = SparkTest.spark

}

object SparkTest {
  var spark: SparkSession = _

  @BeforeClass
  def initializeSpark(): Unit = {
    spark = SparkSession
      .builder()
      .appName("TableUpdaterTest")
      .master("local[*]")
      .config("spark.driver.memory", "471859200")
      .config("spark.testing.memory", "471859200")
      .getOrCreate()
  }

  @AfterClass
  def stopSpark(): Unit = {
    spark.stop()
  }
}