Update testing code.

* 把 action 模块的单元测试加回来
* 少量代码整理