Re-add the ability to accept a "done" function for async tests.