如何在Pytest类方法中访问session作用域的Celery Fixture？

嘿，这个场景我之前写Celery任务测试时也碰到过！默认情况下pytest的fixture是注入到测试实例中的，类方法没法直接拿到session作用域的Celery fixture，不过有几个靠谱的办法可以解决：

方法1：通过request Fixture在setup_class中获取 #

你可以利用pytest的request fixture，在类的setup_class方法中获取session作用域的Celery fixture，然后把它赋值给类属性，这样所有类方法和实例方法都能访问到。

举个实际的代码例子：

首先是你的Celery session fixture：

import pytest
from celery import Celery

@pytest.fixture(scope="session")
def celery():
    # 初始化测试用的Celery实例
    test_celery = Celery("test_tasks")
    test_celery.conf.update(
        broker_url="memory://",
        result_backend="cache+memory://",
        task_always_eager=True  # 可选：让任务同步执行，方便测试
    )
    yield test_celery

然后是测试类：

import pytest

class TestCeleryTasks:
    @classmethod
    def setup_class(cls, request):
        # 从request中获取session作用域的celery fixture
        cls.celery_app = request.config.getfixturevalue("celery")

    def test_instance_level_task(self):
        # 实例方法中通过self访问
        result = self.celery_app.send_task("your_app.tasks.add", args=(1, 2))
        assert result.get() == 3

    @classmethod
    def test_class_level_task(cls):
        # 类方法中直接通过cls访问
        result = cls.celery_app.send_task("your_app.tasks.multiply", args=(2, 3))
        assert result.get() == 6

注意这里setup_class方法需要接收request参数，pytest会自动注入这个fixture。

方法2：使用@pytest.mark.usefixtures结合类属性 #

如果你不想在setup_class里写太多逻辑，可以先用@pytest.mark.usefixtures把Celery fixture注入到测试类的实例中，然后在setup_class里把实例的fixture引用复制到类属性上：

import pytest

@pytest.mark.usefixtures("celery")
class TestCeleryTasks:
    @classmethod
    def setup_class(cls):
        # 这里需要先创建一个临时实例来获取fixture
        temp_instance = cls()
        cls.celery_app = temp_instance.celery

    @classmethod
    def test_class_task(cls):
        result = cls.celery_app.send_task("your_app.tasks.subtract", args=(5, 2))
        assert result.get() == 3

不过这个方法需要注意，临时实例的创建不会触发其他fixture的初始化（如果有的话），所以如果你的Celery fixture依赖其他session级fixture，方法1会更稳妥。

关键要点 #

• Session作用域的fixture在整个测试会话中只会初始化一次，所以不管你用哪种方法获取，拿到的都是同一个Celery实例
• 避免在类方法中直接尝试实例化fixture，一定要通过pytest的request对象或者注入机制来获取，这样才能保证fixture的生命周期被正确管理

内容的提问来源于stack exchange，提问作者Heron Rossi