Python单元测试如何正确mock类方法内的getpass用户输入

抛出EOFError是因为mock没有生效，原生getpass在无交互的测试环境下读取不到终端输入直接报错，具体有3处核心错误：

1. patch路径错误：mock的核心规则是「patch对象被使用的位置，而非定义的位置」。你在被测文件中用from getpass import getpass将函数导入到了当前模块的命名空间，实际调用的是当前模块下的getpass引用。你写的@mock.patch("getpass.getpass")只替换了getpass模块里的原函数，根本没影响到被测模块里的调用，mock完全没生效。
2. 实例化参数缺失：GetInfo类的__init__方法要求传入name参数，你的测试代码写GetInfo()没有传参，本身就会触发参数缺失错误，也无法进入name == 'Bob'的逻辑分支。
3. 原代码逻辑缺陷：当self._name不等于Bob时，new_dict变量从未被定义，直接return会抛出UnboundLocalError。

另外测试代码还有冗余问题：Python3.3+ 标准库已经内置unittest.mock，不需要单独安装第三方mock包，避免版本冲突；最后入口调用的test_main()不存在，要用unittest标准入口。

第一步：补全原代码的边界逻辑 #

先初始化默认返回值，避免非Bob场景下变量未定义：

from getpass import getpass

class GetInfo:
    def __init__(self, name):
        self._name = name

    def get_values(self):
        new_dict = {}
        if self._name == 'Bob':
            phrase = getpass('What is your secret:')
            new_dict = {self._name: phrase}
        return new_dict

第二步：编写正确的单元测试 #

注意把patch路径里的your_module替换成你实际存放GetInfo类的文件名（比如类存在user_info.py里，就patch user_info.getpass）：

import unittest
from unittest.mock import patch
# 替换成你实际的导入路径，比如from user_info import GetInfo
from your_module import GetInfo

class TestInfo(unittest.TestCase):
    @patch("your_module.getpass")
    def test_get_bob_secret(self, mock_getpass):
        # 配置mock返回值
        mock_getpass.return_value = 'Password1'
        # 传入Bob作为name，进入目标测试分支
        info = GetInfo(name='Bob')
        result = info.get_values()
        
        # 断言返回结果符合预期
        self.assertEqual(result, {'Bob': 'Password1'})
        # 断言getpass被正确调用，传入了要求的提示文本
        mock_getpass.assert_called_once_with('What is your secret:')

    def test_get_other_user_value(self):
        info = GetInfo(name='Alice')
        result = info.get_values()
        self.assertEqual(result, {})

if __name__ == '__main__':
    unittest.main()

• 做单元测试mock时，永远记住patch的是被测模块命名空间下的对象，不是对象的原始定义路径，这是90%的mock不生效问题的根源。
• 测试分支逻辑时，必须构造对应分支需要的前置条件，比如测Bob的分支就必须传入name='Bob'，否则逻辑根本不会走到你要测的代码行。
• 除了断言返回结果，建议加mock调用断言，确认依赖函数被正确调用、传入参数符合预期，避免出现逻辑漏判。

内容的提问来源于stack exchange，提问作者thatwayeasy33