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

13 lines
469 B
Python

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