import pytest
from funcy import lmap

from pyinfra.server.rest import process_lazily
from pyinfra.server.utils import unpack


@pytest.mark.parametrize("batched", [True, False])
@pytest.mark.parametrize("item_type", ["string", "image", "pdf"])
def test_pickup_endpoint(url, input_data_items, metadata, operation, target_data_items, server_process):
    output = lmap(unpack, process_lazily(url, input_data_items, metadata))
    assert output == target_data_items