Add unit test for generator template