Use done idiom in unit tests.