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
|
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)
|
||||||
@ -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
|
||||||
|
|
||||||
@ -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)
|
||||||
|
|
||||||
|
|||||||
@ -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))))
|
||||||
|
|||||||
@ -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)))
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user