Add test for intergation and unit test generator