Simplify mocking in attr_aug_* tests