如何使用pytest在“設置”方法中使用Monkeypatch進行單元測試？

Question

我正在嘗試在單元測試中模擬一個實用程序類（在本例中為python logger實用程序）。

雖然我知道如何在每個測試級別上使用Monkeypatch做到這一點，但我希望我可以以某種方式簡單地將其作為安裝程序/全局的一部分。

這是我希望可以做的（但出現錯誤）：

import logging

...

def setup(self, monkeypatch):

    class fake_logger(level):
        def __init__(self, val):
            pass

        def setLevel(self, level):
            # Do something

    def mock_logger(level):
        return fake_logger(level)
    monkeypatch.setattr(logging, 'getLogger', mock_logger)

什么是正確的方法？

編輯：示例錯誤

name = 'setup'

def call_optional(obj, name):
    method = getattr(obj, name, None)
    isfixture = hasattr(method, "_pytestfixturefunction")
    if method is not None and not isfixture and py.builtin.callable(method):
        # If there's any problems allow the exception to raise rather than
        # silently ignoring them
>           method()
E           TypeError: setup() missing 1 required positional argument: 'monkeypatch'

Answer 1

monkeypatch可以用作普通的pytest固定裝置。 如果要使用它，則還需要將您的方法用作固定裝置。

import logging

import pytest


@pytest.fixture
def setup(monkeypatch):

    class fake_logger(object):
        def __init__(self, val):
            pass

        def setLevel(self, level):
            # Do something
            pass

    def mock_logger(level):
        return fake_logger(level)
    monkeypatch.setattr(logging, 'getLogger', mock_logger)

def test_fake_logger(setup):
    # test steps

並且如果您在測試中檢查logging.getLogger('any level')的類型，則它將是您定義的fake_logger 。