使用奇怪的语法用 pytest 测试装饰器

Question

所以我想测试一个装饰器，它可以为您提供一个已启动的协程（所以我不必这样做 coroutine.send(None) ）。

详情请见here。大卫比兹利的例子）

所以在 conftest.py 中，我创建了一个函数包装器，它是一个 pytest 固定装置，如果有意义的话，它会为我提供一个原始协程。

我这样做的原因是因为 pytest 会将 yield 关键字之后的任何内容作为拆卸代码，而我不希望这样。所以我自己封装了一个原始协程。

@pytest.fixture(scope="module")
def create_coroutine_func():

    def grep(pattern):
        print "Looking for %s" % pattern
        while True:
            line = (yield)
            if pattern in line:
                print line
    return grep

现在在我的测试文件中，我有一个相当简单的测试：

    from something.decorators import *
    import pytest

1    def test_coroutine(create_coroutine_func):
2        coro_func = create_coroutine_func()
3        coro_wrapper = coroutine(coro_func)
4    
5        assert callable(coro_wrapper)

好的，就是这样。 pytest 抱怨：

create_coroutine() 缺少 1 个必需的位置参数：'pattern'

然后我继续编辑第 2 行：

coro_func = create_coroutine_func("python") # match anything has 'python' in it.

现在测试通过了。

现在我对装饰器有点困惑。

所以在我的例子中 grep 接受一个参数；当我调用我的 grep 函数包装器时，我还需要给函数包装器一个参数，尽管它本身不接受任何参数？这就是装饰器的工作原理吗？

Answer 1

UPD：正如@jacques-kvam 在 cmets 中指出的那样，当您将夹具作为参数传递给测试函数时，夹具的返回值就是参数的值。所以在你的情况下，第 2 行 create_coroutine_func 是 grep。

这种对pytest fixtures的误解可能是test函数本身的值被误用的主要原因。

一旦清楚，这是我的原始答案：

---

在这里，您在create_coroutine_func() 的第 2 行调用您的函数。这就是它期望参数被传递的原因。

如果你想将函数包装到装饰器中，你真正需要的是对函数对象本身进行操作。它不调用它，而是将其作为值传递给装饰器：

def test_coroutine(create_coroutine_func):
    coro_wrapper = coroutine(create_coroutine_func)
    assert callable(coro_wrapper)
    coro_wrapper("python")

这就是装饰器的工作方式：它们将一个函数作为参数并返回另一个（或相同）函数作为结果。

---

PS：同样的测试可以写成更好的语法：

from something.decorators import *
import pytest

def test_coroutine(create_coroutine_func):

    @coroutine
    def grep(pattern):
        print "Looking for %s" % pattern
        while True:
            line = (yield)
            if pattern in line:
                print line       

    assert callable(grep)
    grep("python")

请注意，装饰器的使用在测试内部。因此，如果出现任何问题，测试将正常失败，而不是 pytest 的模块导入（如果装饰函数在模块级别声明，就会发生这种情况，您可能已经注意到了）。