python - 在类方法中传递的 Python unittest 测试被调用

Question

我有一个方法，它需要两个输入一个电子邮件类和一个数据字典。

def send_email(data, email_client):
     **** various data checks and formatting *****
     response_code = email_client.create_email(recipient=receipient
                                               sender=sender
                                               etc...)

我正在尝试编写一个单元测试，它将断言 email_client.create_email 是根据输入数据使用正确的值调用的。

在我的测试文件中，我有

from emailer.email import send_email

class TestEmails(unittest.TestCase):

    def test_send_email(self):
        email.send_email(get_transactional_email_data, MagicMock())

我通常会通过类似于以下内容的方法来测试调用的方法：

mock.assert_called_with(recipient=receipient
                       sender=sender
                       etc..)

但是，由于这次我正在测试使用（和 MagicMock）调用传入的类的内容，所以我不知道应该如何完成。

Answer 1

我认为您不需要 MagicMock。只需预先创建模拟

from emailer.email import send_email

class TestEmails(unittest.TestCase):

def test_send_email(self):
    myclient = Mock()
    mycreate = Mock()
    myclient.create_email = mycreate
    email.send_email(get_transactional_email_data, myclient)
    self.assertTrue(
        mycreate.called_with(sender='...')
    )