把Extension的测试放单独文件，修改了mock的错误使用。