Allow awaitClick to return the clicked element

Hello :)
It would be great if we could write
var button := awaitClick(...)
directly, instead of calling awaitClick(...) and then getElement(...) in two separate steps.

Start by locating the awaitClick implementation and the getElement usage it currently requires. Confirm how the clicked element is represented, then verify that assigning awaitClick(...) directly provides that element while the existing call pattern remains usable.

Written by the indexing model from the issue text.