修改mock参数使用，优化代码