python mock 的一些小例子
1.假如我想要mock库中date对象，但是我只想mock today（）这个方法，就是today（）返回我想要的结果，但是其他的date的初始化方法不能有所改变。
from datatime import date
with patch('mymodule.date') as mock_date:
mock_date.taday.return_value = date(2010,10,8)
mock_date.side_effect = lambda *args, **kw:date(*args, **kw)
assert mymodule.date.today() == date(2010,10,8)
assert mymodule.date(2009,6,8) == date(2009,6,8)2.配置mock
mock = Mock()
attrs = {'method.return_value':3,'other.side_effect':KeyError}
mock.configure_mock(**attrs)
mock.method() # return 3
mock.other() # return KeyError3.mock其他module中使用的函数
假如test_fuction_uu.py中要测试fuction_uu，但是fuction_uu调用了myfuction，我想要mock这个myfuction。终于用如下的patch的方式搞定了
test_module.py
def myfuction():
return 2
def fuction_uu():
return myfuction()test_mock_fuction.py
@patch('test_module.myfuction',MagicMock(return_value = 13))
def test_faction_uu():
print fuction_uu()
if __name__ == '__main__':
test_faction_uu()4.对于类的patch，注意setup和teardown，在mock的官方文档中说如果setup中抛出异常，那么teardown就不会被执行，这可能会造成一个坑。文档上说用unittest2中的cleanup函数可以避免这个问题。
class MyTest():
def setUp(self):
patcher = patch('package.module.Class')
self.MockClass = patcher.start()
self.addCleanup(patcher.stop)
def test_something(self):
assert package.Module.Class is self.MockClass 5.patch.object可以有“替换”参数的作用，下面这个
@patch.object(SomeClass,'class_mothod')
def test(mock_method):
SomeClass.class_method(3)
mock_method.assert_called_with(3)
test() # attention: no parameter!6.mock数据库操作
mock = Mock()
cursor = mock.connection.cursor.return_value
cursor.execute.return_value = ['foo']
mock.conection.cursor().execute("select 1")
# return ['foo']
expected = call.connection.cursor().execute("select 1")
mock.mock_calls
#return [call.connection.cursor(),call.connection.cursor().ececute('select 1')]
mock.mock_calls == expected
#return True本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2019-09-03 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读