Fix Augeas mocking for attr_aug_* functions