Adding unit tests for directives