优化测试用例，使用基于状态的 MOCK，同时增加断言判断