refactoring: rename
This commit is contained in:
parent
456fb1db06
commit
7a1db32c3b
@ -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)
|
||||
@ -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
|
||||
|
||||
@ -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)
|
||||
|
||||
|
||||
@ -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))))
|
||||
|
||||
@ -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)))
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user