Extract JUnit integration code from Layout Dialect