Add support for tests which return 'async Task'.