Factor out Await test support class