优化测试代码，重构部分测试