Add extension methods which help waiting on async tests.