Question

我正在使用Mock从具有特定返回值的类替换方法。它工作得非常好，也许有点太好......我这样做（见下文），但在下一个测试类中，我重复使用密码类而不进行模拟，并且放置在该测试中的模拟仍然有效。

from utils import password as pass_helper

class TestPassword(unittest.TestCase):
    def setUp(self):
        self.username = "user"
        self.password = "Test_1234_pass"
        pass_helper._get_password_from_keyboard = Mock(return_value=self.password)

    def test_password(self):
        password = pass_helper._get_password_from_keyboard(self.username)
        self.assertEqual(password, self.password)

我尝试通过这样的方式撤消TearDown方法中的mock，但它不起作用。

pass_helper._get_password_from_keyboard = pass_helper._get_password_from_keyboard

如何恢复类方法的原始功能？

Answer 1

问题，就像你已经收集到的那样，你所做的改变不仅限于测试的范围，而是流入其他测试（这在单元测试时当然是个大问题）。你想要颠倒你的拆解方法的改变是一个很好的想法，但问题是当你这样做时，你正在重新分配方法的模拟版本：

pass_helper._get_password_from_keyboard = pass_helper._get_password_from_keyboard

这样的东西应该可行，在模拟方法之前，你将方法的“真实”版本分配给一个临时变量：

def setUp(self):
    self.username = "user"
    self.password = "Test_1234_pass"
    self.real_get_password_from_keyboard = pass_helper._get_password_from_keyboard
    pass_helper._get_password_from_keyboard = Mock(return_value=self.password)

def tearDown(self):
    pass_helper._get_password_from_keyboard = self.real_get_password_from_keyboard

def test_password(self):
    password = pass_helper._get_password_from_keyboard(self.username)
    self.assertEqual(password, self.password)

希望这有帮助！

Python撤消方法mock

1 个答案: