python mock assert_called_with

Question

我试图理解mock中的assert_called_with，但我写的代码会抛出一些错误。

import os
import twitter

URL = "http://test.com"

def tweet(api, message):
    if len(message) > 40:
        message = message.strip("?.,.,.")

    status = api.PostUpdate(message)
    return status

def main():
    api = twitter.Api(consumer_key=''
                    ,consumer_secret='')
    msg = 'This is test message'
    tweet(api, msg)

if __name__ == '__main__':
    main()

单元测试

import unittest
from mock import Mock
import test

class TweetTest(unittest.TestCase):
    def test_example(self):
        mock_twitter = Mock()        
        test.tweet(mock_twitter,'msg')        
        mock_twitter.PostUpdate.assert_called_with('message')        

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

我想了解这里的assert_called_with是干什么用的？

- user1050619

---

那个测试对你没有任何价值。它的作用只是在api.PostUpdate没有使用'message'时失败。根据代码，它每次都会失败。更好的测试是查找Twitter API测试凭据并实际访问外部API。 - Dan

@Dan 我同意这个代码片段的用途有限，但我不同意查找API测试凭据并调用外部API。引入模拟的整个目的不就是让你可以对代码如何与外部系统交互进行断言，而无需实际调用/改变该外部系统吗？否则我们就会涉及端到端/集成测试的领域了？ - Yu Chen

Answer 1

根据Python文档https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.assert_called_with，该方法是一种方便的断言方式，用于判断调用是否以特定方式进行，因此它测试参数是否以正确的方式使用。

关于您收到的错误，我认为您传递的参数是错误的。应该像这样：

mock_twitter.PostUpdate.assert_called_with(message='msg')

- Nicolas Teodosio