refactoring: rename

This commit is contained in:
Matthias Bisping 2022-05-05 11:22:17 +02:00
parent 456fb1db06
commit 7a1db32c3b
7 changed files with 9 additions and 9 deletions

View File

@ -12,7 +12,7 @@ def has_next(peekable_iter):
return peekable_iter.peek(Nothing) != Nothing
class Sender:
class Dispatcher:
def __call__(self, packages: Iterable[dict]):
packages = peekable(packages)

View File

@ -1,9 +1,9 @@
import requests
from pyinfra.server.sender.sender import Sender
from pyinfra.server.dispatcher.dispatcher import Dispatcher
class RestSender(Sender):
class RestDispatcher(Dispatcher):
def __init__(self, endpoint):
self.endpoint = endpoint

View File

@ -6,7 +6,7 @@ from funcy import identity, rcompose, flatten
from pyinfra.server.packer.packers.rest import RestPacker
from pyinfra.server.receiver.receivers.rest import RestReceiver
from pyinfra.server.sender.senders.rest import RestSender
from pyinfra.server.dispatcher.senders.rest import RestDispatcher
from pyinfra.server.utils import stream_response_payloads
from pyinfra.utils.func import lift
@ -55,7 +55,7 @@ def head(endpoint):
"""Sends packages of data and metadata to endpoint and returns response."""
return rcompose(
RestPacker(),
RestSender(endpoint),
RestDispatcher(endpoint),
RestReceiver()
)(data, metadata)

View File

@ -1,12 +1,12 @@
import pytest
from pyinfra.server.receiver.receivers.rest import RestReceiver
from pyinfra.server.sender.senders.rest import RestSender
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 = RestSender(f"{url}/process")
sender = RestDispatcher(f"{url}/process")
receiver = RestReceiver()
assert all((isinstance(r, list) for r in receiver(sender(packages))))

View File

@ -1,10 +1,10 @@
import pytest
from pyinfra.server.sender.senders.rest import RestSender
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_sender(url, packages, server_process):
sender = RestSender(f"{url}/process")
sender = RestDispatcher(f"{url}/process")
assert all((r.status_code == 200 for r in sender(packages)))