Adding support for async tests (taking callback parameter).