import pytest

from pyinfra.server.receiver.receivers.rest import RestReceiver
from pyinfra.server.sender.senders.rest import RestSender


@pytest.mark.parametrize("batched", [True, False])
@pytest.mark.parametrize("item_type", ["string", "image", "pdf"])
def test_rest_receiver(url, packages, server_process):
    sender = RestSender(f"{url}/process")
    receiver = RestReceiver()
    assert all((isinstance(r, list) for r in receiver(sender(packages))))