python 模拟 assert_called_with
python mock assert_called_with
我试图理解 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
在这里做什么?
根据 python 文档 https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.assert_called_with
'This method is a convenient way of asserting that calls are made in a
particular way'
所以它测试参数是否以正确的方式使用。
关于您收到的错误,我认为您传递的参数有误。它必须是这样的:
mock_twitter.PostUpdate.assert_called_with(message='msg')
我试图理解 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
在这里做什么?
根据 python 文档 https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock.assert_called_with
'This method is a convenient way of asserting that calls are made in a particular way'
所以它测试参数是否以正确的方式使用。
关于您收到的错误,我认为您传递的参数有误。它必须是这样的:
mock_twitter.PostUpdate.assert_called_with(message='msg')