增加测试类，修改bug