fix(test): update unit tests for new directives