ScalaTest unit test examples for spark and cassandra

CalculateCurrentBalanceSpec.scala

feature("Calculating current balances") {
     scenario("Positive transactions") {
      Given("a cassandra table with only positive transactions")
      val cassandraRDD = buildMockCassandraRDD(positiveTransactionsFilePath)

      When("I calculate users' balance")
      val results = CalculateCurrentBalance.calculateCustomerBalances(cassandraRDD).collect()

      Then("all balances should be positive")
      every(results.map { case ((_, _), amount) => amount }) should be > 0
    }
}