diff --git a/test/unit_tests/rest/sender_test.py b/test/unit_tests/rest/sender_test.py index bce95bf..7185c13 100644 --- a/test/unit_tests/rest/sender_test.py +++ b/test/unit_tests/rest/sender_test.py @@ -5,6 +5,7 @@ from pyinfra.server.dispatcher.dispatchers.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 = RestDispatcher(f"{url}/process") - assert all((r.status_code == 200 for r in sender(packages))) +def test_rest_sender(endpoint, packages, server_process): + sender = RestDispatcher(endpoint) + responses = sender(packages) + assert all((r.status_code == 200 for r in responses))