Write tests for ClassGeneratorUtils

- Fix return and docBlock