Update unit test.

* 少量改进了测试代码，用 `after()` 来实现收尾动作，而不是用一个空测试来做。