Update unit test.

* 少量改进了测试代码，用 `after()` 来实现收尾动作，而不是用一个空测试来做。

***

先看 PR #6 吧，本 PR 是在它之上的修改。