Fixed the CLI unittest

* Uses the new parameters.
* Now cleansup the generated files.