如何在 Python 单元测试中模拟一个类？

Question

我有一堂课：

class A:
    __init__(self):
        self.b = B()

   def is_authorized(self)
      name = self.b.get_name()

      if name == 'admin':
          return True
      else:
          return False

我想编写一个单元测试来测试该is_authorized方法。问题是它需要一个B类的实例，因为它需要网络连接和其他东西，所以孤立地构造它是非常复杂的。我如何模拟这个类并提供一些只有该get_name方法的东西。这样我就可以创建A类并测试方法。

Answer 1

通过使用mock库，您可以修补B类并将其替换为MagicMock()对象。mock库正是为了做这些工作而设计的，并打破了有问题的对象或真实资源的硬依赖。

在您的简单示例中，完整的测试将是：

模块_a.py

class B():
    def __init__(self):
        print("The awful B class!!!")

    def get_name(self):
        print("The awful B.get_name() method!!!")


class A():
    def __init__(self):
        self.b = B()

    def is_authorized(self):
        name = self.b.get_name()
        if name == 'admin':
            return True
        else:
            return False

module_a_test.py

import unittest
from unittest.mock import patch
from module_a import A


class MyTestCase(unittest.TestCase):

    # patch B class in a_module by a MagicMock instance
    # mock_b_constructor passed to test method
    @patch("module_a.B")
    def test_a(self, mock_b_constructor):
        # B() return value will be the B() instance assigned to a.b property
        mock_b = mock_b_constructor.return_value
        # Now start test:
        a = A()
        # Ok! b is our mock...
        self.assertIs(a.b, mock_b)
        # Not authorized
        self.assertFalse(a.is_authorized())
        mock_b.get_name.return_value = 'admin'
        # Yeah!!! we are admin
        self.assertTrue(a.is_authorized())
        # Sanity check
        mock_b.get_name.return_value = 'guest'
        self.assertFalse(a.is_authorized())

补丁将仅适用于您的测试方法上下文。这是一个关于如何使用模拟和补丁的简单而直接的示例，mock但实际情况可能会稍微复杂一些。