修改为junit测试方法