python - 如何模拟类方法的 cls 参数？

Question

我有一个使用类方法作为自定义构造函数的类。

# my_module.py
class MyClass:
    def __init__(self, my_arg):
        self.my_arg = my_arg

    @classmethod
    def my_constructor(cls):
        return cls("my val")

我想在测试中断言已使用正确的值调用构造函数并返回实例。

我尝试了几种打补丁的方法，但无法正常工作。

我尝试修补整个类，使用 wraps 参数来执行真正的方法。

# my_test.py
from unittest import TestCase
from unittest.mock import patch

from my_module import MyClass


class MyClassTestCase(TestCase):
    @patch("my_module.MyClass", wraps=MyClass)
    def test_my_constructor(self, my_class):
        result = my_class.my_constructor()
        my_class.assert_called_once_with("my val")
        self.assertEqual(result, my_class.return_value)

这里的问题是 _my_module.MyClass.my_class_ 方法获取的 cls 参数不是我的 Mock，而是真正的 MyClass类。

我的另一个想法是使用 patch.object 来修补真正的 __init__ 构造函数，但这也不起作用。这次是因为 __init__ 不应该返回任何东西，但是 MagicMock 有一个返回值。

# my_test.py
from unittest import TestCase
from unittest.mock import patch

from my_module import MyClass


class MyClassTestCase(TestCase):
    @patch.object(MyClass, "__init__")
    def test_my_constructor(self, init):
        result = MyClass.my_constructor()
        init.assert_called_once_with("my val")
        self.assertEqual(result, init.return_value)

是否有实际的方法来断言 MyClass 类已由 _my_constructor_ 使用给定参数实例化并返回？

Answer 1

模拟 __new__ 而不是 __init__并断言将是init.assert_called_once_with(MyClass, "my val")