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)