Fitten Code可一键生成覆盖成功、失败、超时场景的Mock测试代码:先通过「Analyze Dependency」解析接口签名,再用「Generate Mock Test」创建隔离式unittest骨架,支持JSON Schema定制返回逻辑并实时验证调用覆盖率。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

你需要为尚未开发完成或难以调用的外部依赖接口快速生成可运行的Mock测试代码,避免手动编写繁琐的mock.Mock(return_value=...)语句,同时确保模拟逻辑能覆盖成功、失败、超时等关键场景。
确认目标接口签名与依赖路径
打开待测模块源码,定位到调用外部接口的方法,例如payment.auth(card_num, amount)。右键该方法名 → 选择「Fitten Code → Analyze Dependency」,工具将自动解析其所在类、参数类型、返回值类型及调用栈深度。
这一步必须做,否则后续生成的Mock无法匹配真实调用签名。若分析失败,【请先确保光标停在方法定义行内,且项目已正确配置Python解释器】。
一键生成多场景Mock测试骨架
选中目标方法(如auth)→ 右键 → 「Fitten Code → Generate Mock Test」→ 在弹出面板中勾选「Success / Failure / Timeout」三类基础场景 → 点击「Generate」。
Fitten Code会自动在当前文件同级目录下创建test_payment_mock.py,内含完整unittest.TestCase结构,每个测试方法均带注释说明触发条件,例如# 模拟requests.post抛出requests.Timeout异常。
生成过程不修改原业务代码,所有mock.patch装饰器均作用于测试方法内部,隔离性严格。
定制化注入模拟返回逻辑
方法一:直接编辑生成的test文件,在test_auth_success方法内找到mock_response = mock.Mock()行下方插入:
Fitten Code 1.0.3是一款由清华博士团队打造的AI编程助手,基于国产计图(Jittor)深度学习框架开发。它支持VS Code、JetBrains系列等主流IDE及80多种编程语言。核心功能包括智能代码补全、注释生成、代码解释、Bug检测、单元测试生成等,旨在全方位提升开发效率。该工具对个人用户免费开放。
mock_response.status_code = 200 → mock_response.json.return_value = {"result": "approved", "tx_id": "TX123"}
方法二:在Fitten Code侧边栏「Mock Rules」面板中,为auth方法添加JSON Schema规则,输入:
{"status_code": 500, "json": {"error": "insufficient_balance", "code": "BAL_001"}} → 点击「Apply to test_auth_failure」,对应测试方法中的return_value将自动更新为该结构。
注意:Schema中字段名必须与真实响应一致,否则断言会因KeyError失败。
运行并验证Mock行为
第一步:在PyCharm中右键点击test_payment_mock.py → 「Run Unittests in test_payment_mock」。
第二步:观察控制台输出,确认三个测试用例全部绿色通过,且每条日志含「MOCKED: auth called with card_num=6666, amount=100000」字样。
第三步:打开「Fitten Code → Mock Dashboard」,查看实时统计:已拦截外部HTTP请求0次,本地Mock调用37次,覆盖率100%。

















