Pytest - 高级进阶mock
2018-04-05 本文已影响73人
红薯爱帅
1. 概述
Mock测试是在测试过程中对可能不稳定、有副作用、不容易构造或者不容易获取的对象,用一个虚拟的对象来创建以便完成测试的方法。在Python中这种测试是通过第三方的mock库完成的,mock在Python3.3的时候被引入到Python标准库中,改名为unittest.mock。之前的Python版本都需要安装:pip install mock
初学者理解mock比较抽象,可以简单理解为函数指针,通过mock成员方法或变量,可以指定mock对象的返回值,获取是否被调用、被调用次数、调用时的参数等。
2. 使用方法
2.1. 简介
- 在test方法前,添加decorator @mock.patch('pyfile.func', return_value='None'),可以完成对pyfile.func的mock,并指定return_value为‘None’,在test方法的参数中增加mock_func即可
- 单个test方法引用多个mock.patch decorator时,patch从下向上,test方法的参数从左向右,依次对应(可以有剩余的参数,但是必学是已经声明的fixture),样例如下
@mock.patch('os.remove')
@mock.patch('os.listdir')
@mock.patch('shutil.copy')
def test_unix_fs(mocked_copy, mocked_listdir, mocked_remove):
UnixFS.rm('file')
os.remove.assert_called_once_with('file')
assert UnixFS.ls('dir') == expected
# ...
UnixFS.cp('src', 'dst')
# ...
2.2. 示例代码
pytest2.py
# -*- coding:utf-8 -*-
import pytest
import mock
@pytest.fixture(scope='function', params=[1, 2, 3]) # multiple test cases
def mock_data_params(request):
return request.param
def test_not_2(mock_data_params):
print('test_data: %s' % mock_data_params)
assert mock_data_params != 2
##################
def _call(x):
return '{} _call'.format(x)
def func(x):
return '{} func'.format(_call(x))
def test_func_normal():
for x in [11, 22, 33]: # multiple test cases
# print '{} - {}'.format(x, func(x))
assert func(x) == '{} _call func'.format(x)
@mock.patch('pytest2._call', return_value='None') # set mock 'pytest2._call'.return_value is 'None', also could use mock_call.return_value='None'
def test_func_mock(mock_call):
for x in [11, 22, 33]:
# print '{} - {}'.format(x, func(x))
ret = func(x)
assert mock_call.called # True or False
assert mock_call.call_args == ((x,),) # This is either None (if the mock hasn’t been called), or the arguments that the mock was last called with
assert ret == 'None func' # because mock_call.return_value is 'None', so the result here is 'None func'
如何获取mock方法的调用次数和参数数据
-
mock_call.called # True or False
-
mock_call.call_args # This is either None (if the mock hasn’t been called), or the arguments that the mock was last called with
-
mock_call.return_value
image.png -
mock_call.side_effect # This can either be a function to be called when the mock is called, an iterable or an exception (class or instance) to be raised
image.png
详细内容参考官网
- Python3 unittest.mock.Mock
https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock - pytest-mock
https://pypi.python.org/pypi/pytest-mock - pytest-mock github
https://github.com/pytest-dev/pytest-mock/
2.3. 执行结果
通过fixture的param功能,完成多用例测试
$ pytest -v pytest2.py::test_not_2
============================================================================== test session starts ===============================================================================
platform linux2 -- Python 2.7.14, pytest-3.0.0, py-1.5.2, pluggy-0.3.1 -- /home/kevin/soft/anaconda2/bin/python
cachedir: .cache
Using --randomly-seed=1522926920
rootdir: /home/kevin/learn/python-web/tox/case2, inifile:
plugins: randomly-1.0.0, mock-1.2, cov-2.0.0
collected 3 items
pytest2.py::test_not_2[3] PASSED
pytest2.py::test_not_2[2] FAILED
pytest2.py::test_not_2[1] PASSED
==================================================================================== FAILURES ====================================================================================
_________________________________________________________________________________ test_not_2[2] __________________________________________________________________________________
mock_data_params = 2
def test_not_2(mock_data_params):
print('test_data: %s' % mock_data_params)
> assert mock_data_params != 2
E assert 2 != 2
pytest2.py:10: AssertionError
------------------------------------------------------------------------------ Captured stdout call ------------------------------------------------------------------------------
test_data: 2
============================================================================= pytest-warning summary =============================================================================
WC1 None pytest_funcarg__cov: declaring fixtures using "pytest_funcarg__" prefix is deprecated and scheduled to be removed in pytest 4.0. Please remove the prefix and use the @pytest.fixture decorator instead.
============================================================= 1 failed, 2 passed, 1 pytest-warnings in 0.02 seconds ==============================================================
在test方法中通过循环dataset完成多用例测试
$ pytest -vs pytest2.py::test_func_normal
============================================================================== test session starts ===============================================================================
platform linux2 -- Python 2.7.12, pytest-3.0.0, py-1.5.2, pluggy-0.3.1 -- /usr/bin/python
cachedir: .cache
Using --randomly-seed=1522930543
rootdir: /home/kevin/learn/python-web/tox/case2, inifile:
plugins: xdist-1.15.0, randomly-1.0.0, mock-1.2, cov-2.0.0, celery-4.0.1
collected 6 items
pytest2.py::test_func_normal PASSED
============================================================================= pytest-warning summary =============================================================================
WC1 None pytest_funcarg__cov: declaring fixtures using "pytest_funcarg__" prefix is deprecated and scheduled to be removed in pytest 4.0. Please remove the prefix and use the @pytest.fixture decorator instead.
================================================================== 1 passed, 1 pytest-warnings in 0.01 seconds ===================================================================
通过mock,构造虚拟函数的调用
$ pytest -vs pytest2.py::test_func_mock
============================================================================== test session starts ===============================================================================
platform linux2 -- Python 2.7.12, pytest-3.0.0, py-1.5.2, pluggy-0.3.1 -- /usr/bin/python
cachedir: .cache
Using --randomly-seed=1522930553
rootdir: /home/kevin/learn/python-web/tox/case2, inifile:
plugins: xdist-1.15.0, randomly-1.0.0, mock-1.2, cov-2.0.0, celery-4.0.1
collected 6 items
pytest2.py::test_func_mock PASSED
============================================================================= pytest-warning summary =============================================================================
WC1 None pytest_funcarg__cov: declaring fixtures using "pytest_funcarg__" prefix is deprecated and scheduled to be removed in pytest 4.0. Please remove the prefix and use the @pytest.fixture decorator instead.
================================================================== 1 passed, 1 pytest-warnings in 0.00 seconds ===================================================================
3. 参考
- pytest
https://docs.pytest.org/en/latest/example/markers.html - pytest-mock
https://pypi.python.org/pypi/pytest-mock - Python3 unittest.mock.Mock
https://docs.python.org/3/library/unittest.mock.html#unittest.mock.Mock