Replace `wait_for_aae` with `wait_until`

Use the built-in riak-tset `wait_until` functionality instead of using
my own.

Replace sleep with `wait_for_index`.