Add test coverage for asynchronous calls