Add tests files in generator