23 lines
492 B
Python
23 lines
492 B
Python
from _operator import itemgetter
|
|
|
|
import pytest
|
|
|
|
from pyinfra.queue.consumer import Consumer
|
|
from test.utils.input import pair_data_with_queue_message
|
|
|
|
|
|
@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 pair_data_with_queue_message(numbers)
|