时间戳

首页 > 解决方案 > pytest - monkeypatch 关键字参数默认值

python - pytest - monkeypatch 关键字参数默认值

问题描述

我想测试一个函数的默认行为。我有以下内容：

# app/foo.py
DEFAULT_VALUE = 'hello'

def bar(text=DEFAULT_VALUE):
    print(text)

# test/test_app.py
import app

def test_app(monkeypatch):
    monkeypatch.setattr('app.foo.DEFAULT_VALUE', 'patched')
    app.foo.bar()
    assert 0

输出是hello；不是我想要的。

一种解决方案是显式传递默认值：app.foo.bar(text=app.foo.DEFAULT_VALUE).

但我发现有趣的是，当默认为全局范围时，这似乎不是问题：

# app/foo.py
DEFAULT_VALUE = 'hello'

def bar():
    print(DEFAULT_VALUE)

输出是patched。

为什么会这样？还有比明确传递默认值更好的解决方案吗？

标签： pythonpytestmonkeypatching

解决方案

函数默认值在函数定义时绑定。

当您在测试代码中时，定义函数的模块已经被导入，并且通过在模块级别常量上的猴子补丁替换默认值已经太晚了。那个名字已经被解析了。

一种解决方法是像这样定义函数：

def bar(text=None):
    if text is None:
        text = DEFAULT_VALUE
    print(text)

现在在函数调用时查找默认值，这意味着模块级别默认的猴子补丁仍然有效。

如果您不喜欢修改函数定义，那么您可以对函数对象本身进行猴子补丁：

monkeypatch.setattr("app.foo.bar.__defaults__", ("test_hello",))

推荐阅读