chore: add tests for the directive