flash-sales-check.scala

ArbSingleDsCheck("less than 5 flash sales per customer") { ds =>
  val tooManyFlashSaleCustomerCount = ds
    .filter(col("orderType") === "Flash Sale")
    .groupBy("customerId").agg(count("orderId").as("flashSaleCount"))
    .filter(col("flashSaleCount") > 5)
    .count

  if (tooManyFlashSaleCustomerCount > 0)
    RawCheckResult(CheckStatus.Error, s"$tooManyFlashSaleCustomerCount customers had too many flash sales")
  else
    RawCheckResult(CheckStatus.Success, "No customers had more than 5 flash sales :)")
}