添加测试使用的库。
根据测试重构代码。