添加测试用例
1. 添加测试用例
2. 修改代码结构