Add `assertThatAfter` for testing asynchronous code