如何在 VS Code 中通过 CodeGeex 一键生成符合项目风格的单元测试套件

想要让 CodeGeeX 在 VS Code 里自动生成一套与项目风格完全合拍的单元测试——比如 pytest 风格，带 fixture、参数化，mock 行为也和当前代码库严丝合缝——光靠按个键可不够。三个前提条件必须全部到位：函数上下文得完整，测试依赖要能被识别，项目配置也已经加载。否则生成出来的测试大概率缺 import、搞错命名、漏掉异常路径。

简单来说，这三个前提分别对应着：你选的函数得有完整上下文，测试所需的库（比如 pytest、pytest-mock）能被环境找到，项目根目录下的配置文件中已经写清楚了测试框架和风格。缺一个，生成效果就大打折扣。

确认插件已激活且连接稳定

打开 VS Code 后，先按 Ctrl+Shift+X 进入扩展市场，搜索“CodeGeeX”，确认状态栏右下角出现蓝色鲸鱼图标并且显示“Online”。注意：如果图标是灰色或者提示 offline，那所有生成操作都不会生效。

点击图标，在弹出面板的右上角点用户头像，检查账号状态是不是“Authenticated”。没登录的话，用 GitHub 或者邮箱完成授权就行。

让 CodeGeeX 精准识别项目测试风格

CodeGeeX 不会凭空猜测你用 pytest 还是 unittest——它依赖项目根目录下的实际配置文件来做判断。

首先，打开项目根目录，检查有没有 pyproject.toml 或 setup.cfg。如果有，确认里面包含了 [tool.pytest] 或 [pytest] 区块。

其次，如果项目使用了 conftest.py 自定义了 fixture（比如 @pytest.fixture(scope="session")），确保这个文件放在 tests/ 目录或其父级，并且已经被 VS Code 的 Python 解释器识别到。

最后，在终端跑一下 python -m pytest --version，输出必须包含“pytest X.Y.Z”字样。如果报错“no module named pytest”，那就先在当前环境执行 pip install pytest pytest-mock。

三种触发方式及适用场景

把光标停在目标函数的 def 行任意位置，右键，选择“CodeGeeX: Generate Unit Test”。注意：这个操作只能在 .py 文件里进行，.ipynb 或者未保存的临时文件不支持。生成的测试函数会自动以 test_原函数名 命名，插入位置默认在当前文件末尾；如果当前文件已经有 if __name__ == "__main__" 块，则会插在这个块的上方。

按 Ctrl+Shift+C 调出 CodeGeeX 面板，在输入框里输入 /tests for the function above，回车。模型会解析光标所在函数的全部上下文，包括类型注解、docstring 中的 Example 和 Raises 说明。举个具体例子：如果函数里调用了 requests.get，而且项目已经安装了 responses 库，生成的代码会自动引入 @responses.activate 和对应的 mock 响应；没装的话，就会退化到 patch 方式。

在函数正上方添加一行注释：# @codegeex generate pytest test with fixture="db_session"，光标停在这行，然后按 Ctrl+Enter（Windows/Linux）或 Cmd+Enter（macOS）。这样就会生成一个包含 def test_xxx(db_session): 参数的测试函数，函数体首行还会自动插入类似 assert db_session.is_active 的校验语句。注意：fixture 的名字必须和 conftest.py 中定义的名字完全一致，大小写敏感。