import pytest
from funcy import lmap

from pyinfra.rest import unpack, submit_and_pickup


@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, submit_and_pickup(url, input_data_items, metadata))
    assert output == lmap(unpack, target_data_items)