断言 __init__ 是用正确的参数调用的

我使用 python 模拟来断言特定对象是使用正确的参数创建的。这就是我的代码的样子：

class Installer:
    def __init__(foo, bar, version):
        # Init stuff
        pass
    def __enter__(self):
        return self

    def __exit__(self, type, value, tb):
        # cleanup
        pass

    def install(self):
        # Install stuff
        pass

class Deployer:
    def deploy(self):
        with Installer('foo', 'bar', 1) as installer:
            installer.install()

现在，我想断言installer是用正确的参数创建的。这是我到目前为止的代码：

class DeployerTest(unittest.TestCase):
    @patch('Installer', autospec=True)
    def testInstaller(self, mock_installer):
        deployer = Deployer()
        deployer.deploy()

        # Can't do this :-(
        mock_installer.__init__.assert_called_once_with('foo', 'bar', 1)

这是我得到的错误：

  File "test_deployment.py", line .., in testInstaller
    mock_installer.__init__.assert_called_once_with('foo', 'bar', 1)
AttributeError: 'function' object has no attribute 'assert_called_once_with'

这是固定代码（称之为test.py）。谢谢大家！

import unittest
from mock import patch

class Installer:
    def __init__(self, foo, bar, version):
        # Init stuff
        pass

    def __enter__(self):
        return self

    def __exit__(self, type, value, tb):
        # cleanup
        pass

    def install(self):
        # Install stuff
        pass

class Deployer:
    def deploy(self):
        with Installer('foo', 'bar', 1) as installer:
            installer.install()

class DeployerTest(unittest.TestCase):
    @patch('tests.test.Installer', autospec=True)
    def testInstaller(self, mock_installer):
        deployer = Deployer()
        deployer.deploy()

        # Can't do this :-(
        # mock_installer.__init__.assert_called_once_with('foo', 'bar', 1)

        # Try this instead
        mock_installer.assert_called_once_with('foo', 'bar', 1)

因此，您收到的错误消息实际上是因为您没有正确检查您的模拟。这里你必须明白的是，在你的装饰器中你最终会说，调用Installer将返回一个Mock对象代替。

因此，对于任何调用Installer()关于您要修补的位置，其返回值将调用Mock()反而。

因此，您真正想要检查的断言只是在mock_installer，而不是mock_installer.__init__.:

mock_installer.assert_called_once_with('foo', 'bar', 1)

这是对您的代码所做的修改：

class DeployerTest(unittest.TestCase):

    @patch('Installer', autospec=True)
    def testInstaller(self, mock_installer):
        deployer = Deployer()
        deployer.deploy()

        mock_installer.assert_called_once_with('foo', 'bar', 1)

一些额外的信息来提供更多解释，如果您现在正在测试是否在上下文管理器中调用了 install，您必须在这里意识到您实际上必须检查您的内部__enter__，所以结构会是这样的：

为了清楚起见，在您的测试方法中创建一个mock_obj并：

mock_obj = mock_installer.return_value

现在，在您的上下文管理器中，您将need查看调用内部__enter__()。如果您不知道这是为什么，请阅读上下文管理器。

因此，考虑到这一点，您只需执行检查：

mock_obj.__enter__().install.assert_called_once_with()