Moved async/await tests into a separate class