[DSL] WIP > Improved the generated tests in the Rake task