pyinfra/test/fixtures/consumer.py

23 lines
492 B
Python

from _operator import itemgetter
import pytest
from pyinfra.queue.consumer import Consumer
from test.utils.input import adorn_data_with_storage_info
@pytest.fixture(scope="session")
def consumer(queue_manager, callback):
return Consumer(callback, queue_manager)
@pytest.fixture(scope="session")
def access_callback():
return itemgetter("fileId")
@pytest.fixture()
def items():
numbers = [f"{i}".encode() for i in range(3)]
return adorn_data_with_storage_info(numbers)