Added tests for directive
More tests