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 return peekable_iter.peek(Nothing) != Nothing
class Sender: class Dispatcher:
def __call__(self, packages: Iterable[dict]): def __call__(self, packages: Iterable[dict]):
packages = peekable(packages) packages = peekable(packages)

View File

@ -1,9 +1,9 @@
import requests 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): def __init__(self, endpoint):
self.endpoint = 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.packer.packers.rest import RestPacker
from pyinfra.server.receiver.receivers.rest import RestReceiver 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.server.utils import stream_response_payloads
from pyinfra.utils.func import lift from pyinfra.utils.func import lift
@ -55,7 +55,7 @@ def head(endpoint):
"""Sends packages of data and metadata to endpoint and returns response.""" """Sends packages of data and metadata to endpoint and returns response."""
return rcompose( return rcompose(
RestPacker(), RestPacker(),
RestSender(endpoint), RestDispatcher(endpoint),
RestReceiver() RestReceiver()
)(data, metadata) )(data, metadata)

View File

@ -1,12 +1,12 @@
import pytest import pytest
from pyinfra.server.receiver.receivers.rest import RestReceiver 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("batched", [True, False])
@pytest.mark.parametrize("item_type", ["string", "image", "pdf"]) @pytest.mark.parametrize("item_type", ["string", "image", "pdf"])
def test_rest_receiver(url, packages, server_process): def test_rest_receiver(url, packages, server_process):
sender = RestSender(f"{url}/process") sender = RestDispatcher(f"{url}/process")
receiver = RestReceiver() receiver = RestReceiver()
assert all((isinstance(r, list) for r in receiver(sender(packages)))) assert all((isinstance(r, list) for r in receiver(sender(packages))))

View File

@ -1,10 +1,10 @@
import pytest 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("batched", [True, False])
@pytest.mark.parametrize("item_type", ["string", "image", "pdf"]) @pytest.mark.parametrize("item_type", ["string", "image", "pdf"])
def test_rest_sender(url, packages, server_process): 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))) assert all((r.status_code == 200 for r in sender(packages)))