Refactor button helper tests