5、pytest -- 猴子补丁

有时候，测试用例需要调用某些依赖于全局配置的功能，或者这些功能本身又调用了某些不容易测试的代码（例如：网络接入）。fixture monkeypatch可以帮助你安全的设置/删除一个属性、字典项或者环境变量，甚至改变导入模块时的sys.path路径。

monkeypatch提供了以下方法：

monkeypatch.setattr(obj, name, value, raising=True)
monkeypatch.delattr(obj, name, raising=True)
monkeypatch.setitem(mapping, name, value)
monkeypatch.delitem(obj, name, raising=True)
monkeypatch.setenv(name, value, prepend=False)
monkeypatch.delenv(name, raising=True)
monkeypatch.syspath_prepend(path)
monkeypatch.chdir(path)

所有的修改将在测试用例或者fixture执行完成后撤销。raising参数表明：当设置/删除操作的目标不存在时，是否上报KeyError和AttributeError异常。

1. 修改函数功能或者类属性

使用monkeypatch.setattr()可以将函数或者属性修改为你希望的行为，使用monkeypatch.delattr()可以删除测试用例使用的函数或者属性；

参考以下三个例子：

在这个例子中，使用monkeypatch.setattr()修改Path.home方法，在测试运行期间，它一直返回的是固定的Path("/abc")，这样就移除了它在不同平台上的依赖；测试运行完成后，对Path.home的修改会被撤销；

# src/chapter-5/test_module.py

from pathlib import Path


def getssh():
    return Path.home() / ".ssh"


def test_getssh(monkeypatch):
    def mockreturn():
        return Path("/abc")

    # 替换 Path.home
    # 需要在真正的调用之前执行
    monkeypatch.setattr(Path, "home", mockreturn)
    
    # 将会使用 mockreturn 代替 Path.home
    x = getssh()
    assert x == Path("/abc/.ssh")

在这个例子中，使用monkeypatch.setattr()结合类，模拟函数的返回对象；

假设我们有一个简单的功能，访问一个url返回网页内容：

# src/chapter-5/app.py

from urllib import request


def get(url):
    r = request.urlopen(url)
    return r.read().decode('utf-8')

我们现在要去模拟r，它需要一个.read()方法返回的是bytes的数据类型；我们可以在测试模块中定义一个类来代替r：

# src/chapter-5/test_app.py

from urllib import request

from app import get


# 自定义的类模拟 urlopen 的返回值
class MockResponse:

    # 永远返回一个固定的 bytes 类型的数据
    @staticmethod
    def read():
        return b'luizyao.com'


def test_get(monkeypatch):
    def mock_urlopen(*args, **kwargs):
        return MockResponse()

    # 使用 request.mock_urlopen 代替 request.urlopen
    monkeypatch.setattr(request, 'urlopen', mock_urlopen)

    data = get('https://luizyao.com')
    assert data == 'luizyao.com'

你可以继续为实际的场景构建更具有复杂度的MockResponse；例如，你可以包含一个总是返回True的ok属性，或者根据输入的字符串为read()返回不同的值；

我们也可以通过fixture跨用例共享：

# src/chapter-5/test_app.py

import pytest


# monkeypatch 是 function 级别作用域的，所以 mock_response 也只能是 function 级别，
# 否则会报 ScopeMismatch 
@pytest.fixture
def mock_response(monkeypatch):
    def mock_urlopen(*args, **kwargs):
        return MockResponse()

    # 使用 request.mock_urlopen 代替 request.urlopen
    monkeypatch.setattr(request, 'urlopen', mock_urlopen)


# 使用 mock_response 代替原先的 monkeypatch
def test_get_fixture1(mock_response):
    data = get('https://luizyao.com')
    assert data == 'luizyao.com'


# 使用 mock_response 代替原先的 monkeypatch
def test_get_fixture2(mock_response):
    data = get('https://bing.com')
    assert data == 'luizyao.com'

注意：

测试用例使用的fixture由原先的mock_response替换为monkeypatch；

因为monkeypatch是function级别作用域的，所以mock_response也只能是function级别，否则会报ScopeMismatch: You tried to access the 'function' scoped fixture 'monkeypatch' with a 'module' scoped request object 错误；

如果你想让mock_response应用于所有的测试用例，可以考虑将它移到conftest.py里面，并标记autouse=True；

在这个例子中，使用monkeypatch.delattr()删除urllib.request.urlopen()方法；

# src/chapter-5/test_app.py

@pytest.fixture
def no_request(monkeypatch):
    monkeypatch.delattr('urllib.request.urlopen')


def test_delattr(no_request):
    data = get('https://bing.com')
    assert data == 'luizyao.com'

执行：

λ pipenv run pytest --tb=native --assert=plain --capture=no src/chapter-5/test_app.
py::test_delattr
=============================== test session starts ================================ 
platform win32 -- Python 3.7.3, pytest-5.1.3, py-1.8.0, pluggy-0.13.0
rootdir: D:\Personal Files\Projects\pytest-chinese-doc
collected 1 item

src\chapter-5\test_app.py F

===================================== FAILURES ===================================== 
___________________________________ test_delattr ___________________________________ 
Traceback (most recent call last):
  File "D:\Personal Files\Projects\pytest-chinese-doc\src\chapter-5\test_app.py", line 78, in test_delattr
    data = get('https://bing.com')
  File "D:\Personal Files\Projects\pytest-chinese-doc\src\chapter-5\app.py", line 26, in get
    r = request.urlopen(url)
AttributeError: module 'urllib.request' has no attribute 'urlopen'
================================ 1 failed in 0.04s =================================

注意：
避免删除内置库中的方法，如果一定要这么做，最好加上--tb=native --assert=plain --capture=no；
修改pytest使用到的库，可能会污染pytest本身，建议使用MonkeyPatch.context()，它返回一个MonkeyPatch对象，结合with限制这些修改只发生在包裹的代码中。
def test_stdlib(monkeypatch):
with monkeypatch.context() as m:
    m.setattr(functools, "partial", 3)
    assert functools.partial == 3

2. 修改环境变量

使用monkeypatch的setenv()和delenv()方法，可以在测试中安全的设置/删除环境变量；

# src/chapter-5/test_env.py

import os

import pytest


def get_os_user():
    username = os.getenv('USER')

    if username is None:
        raise IOError('"USER" environment variable is not set.')

    return username


def test_user(monkeypatch):
    monkeypatch.setenv('USER', 'luizyao')
    assert get_os_user() == 'luizyao'


def test_raise_exception(monkeypatch):
    monkeypatch.delenv('USER', raising=False)
    pytest.raises(IOError, get_os_user)

monkeypatch.delenv()的raising要设置为False，否则可能会报KeyError；

你也可以使用fixture，实现跨用例共享：

import pytest


@pytest.fixture
def mock_env_user(monkeypatch):
    monkeypatch.setenv("USER", "TestingUser")


@pytest.fixture
def mock_env_missing(monkeypatch):
    monkeypatch.delenv("USER", raising=False)


# notice the tests reference the fixtures for mocks
def test_upper_to_lower(mock_env_user):
    assert get_os_user_lower() == "testinguser"


def test_raise_exception(mock_env_missing):
    with pytest.raises(OSError):
        _ = get_os_user_lower()

3. 修改字典

使用monkeypatch.setitem()方法可以在测试期间安全的修改字典中特定的值；

DEFAULT_CONFIG = {"user": "user1", "database": "db1"}


def create_connection_string(config=None):
    config = config or DEFAULT_CONFIG
    return f"User Id={config['user']}; Location={config['database']};"

我们可以修改数据库的用户或者使用其它的数据库：

import app


def test_connection(monkeypatch):
    monkeypatch.setitem(app.DEFAULT_CONFIG, "user", "test_user")
    monkeypatch.setitem(app.DEFAULT_CONFIG, "database", "test_db")

    expected = "User Id=test_user; Location=test_db;"

    result = app.create_connection_string()
    assert result == expected

可以使用monkeypatch.delitem删除指定的项：

import pytest

import app


def test_missing_user(monkeypatch):
    monkeypatch.delitem(app.DEFAULT_CONFIG, "user", raising=False)

    with pytest.raises(KeyError):
        _ = app.create_connection_string()

GitHub仓库地址：https://github.com/luizyao/pytest-chinese-doc

posted @ 2019-10-18 15:46 luizyao 阅读(1905) 评论(0) 编辑收藏举报

刷新页面返回顶部

luizyao

戒骄戒躁，砥砺前行

5、pytest -- 猴子补丁

1. 修改函数功能或者类属性

2. 修改环境变量

3. 修改字典

公告