Adds tests to cover button states