Creating unit tests for a class based Scala approach