Added test unit for Controller generation