Update tests to start testing directive internals