Add tests for cli generate command, fix incorrect comment