Add test for directive