adding tests around the generation controller