重构
增加测试类