Add test for layout plugin