[Scala DSL] Test refactoring.
[Scala DSL] Removed deprecated test API.