Start writing unit tests for directive