MainGenerator unit test