Add unit tests for roles directive