Contain get_attribute in the tests within a single method.

This needs to be renamed to 'wait_for_attribute' or something.