Python单元测试中使用unittest.mock.patchstopall()进行修补

在Python中，单元测试是一种用来测试代码中最小可测试单元的方法。在进行单元测试时，有时候我们需要对某些函数或对象进行修补或模拟，以便更好地控制测试环境和测试结果。unittest.mock模块提供了丰富的工具来进行修补和模拟。

其中，unittest.mock.patch.stopall()方法是用来停止全部修补的方法。它将恢复全部已经修补的对象或函数到原始状态，以便在单元测试完成后恢复对其的修补。

下面我们将使用一个例子来说明patch.stopall()方法的使用。

假设我们有一个名为calculator.py的模块，其中包含一个简单的计算器类Calculator，它具有加法和乘法的功能。

# calculator.py

class Calculator:
    def add(self, a, b):
        return a + b
    
    def multiply(self, a, b):
        return a * b

接下来，我们将创建一个名为test_calculator.py的单元测试文件，使用unittest和unittest.mock模块对上述计算器类进行测试，并使用patch.stopall()方法进行修补。

首先，我们导入必要的模块、类和方法。

# test_calculator.py

import unittest
from unittest import TestCase
from unittest.mock import patch
from calculator import Calculator

然后，我们创建一个名为TestCalculator的测试类，并继承自unittest.TestCase。

class TestCalculator(TestCase):
    def setUp(self):
        self.calculator = Calculator()

在TestCalculator类中，我们定义了一个setUp方法，在每个测试方法被执行之前都会被调用。在setUp方法中，我们实例化了Calculator类，并将其赋值给self.calculator属性。

接下来，我们定义了一个测试加法的方法test_add。

    def test_add(self):
        with patch.object(self.calculator, 'add', return_value=5) as mock_add:
            result = self.calculator.add(2, 3)
            self.assertEqual(result, 5)
            mock_add.assert_called_once_with(2, 3)

在test_add方法中，我们使用patch.object方法对self.calculator.add进行修补，将其返回值设为5。然后，我们调用self.calculator.add(2, 3)进行加法计算，并断言其结果为5。最后，我们使用mock_add.assert_called_once_with(2, 3)断言self.calculator.add方法被正确地调用了一次，并传入了参数2和3。

然后，我们定义了一个测试乘法的方法test_multiply。

    def test_multiply(self):
        with patch.object(self.calculator, 'multiply', return_value=6) as mock_multiply:
            result = self.calculator.multiply(2, 3)
            self.assertEqual(result, 6)
            mock_multiply.assert_called_once_with(2, 3)

在test_multiply方法中，我们使用patch.object方法对self.calculator.multiply进行修补，将其返回值设为6。然后，我们调用self.calculator.multiply(2, 3)进行乘法计算，并断言其结果为6。最后，我们使用mock_multiply.assert_called_once_with(2, 3)断言self.calculator.multiply方法被正确地调用了一次，并传入了参数2和3。

最后，我们定义了一个tearDown方法，在每个测试方法执行后都会被调用。

    def tearDown(self):
        patch.stopall()

在tearDown方法中，我们调用patch.stopall()方法来停止全部修补，以便在每个测试方法执行完毕后恢复对已修补的对象的修补。

最后，我们使用unittest.main()方法来运行单元测试。

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

至此，我们完成了对单元测试中使用unittest.mock.patch.stopall()进行修补的例子。通过使用patch.stopall()方法，我们可以更好地控制单元测试的环境和结果，以便更准确地进行测试和评估代码的正确性。