Add integration test for image generator