Add test for ButtonBehavior