Unit testing directive some more