pyinfra/test/unit_tests/rest/receiver_test.py
2022-05-05 10:17:38 +02:00

13 lines
467 B
Python

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