import pytest

from pyinfra.rest import post_partial


@pytest.mark.parametrize("batched", [True, False])
@pytest.mark.parametrize("item_type", ["pdf", "string", "image"])
def test_sending_partial_request(url, input_data_items, metadata, operation, target_data_items, server_process):
    output = list(post_partial(f"{url}/process", input_data_items, metadata))
    assert output == target_data_items