add tests in scalatest