diff --git a/pyinfra/queue/queue_manager/pika_queue_manager.py b/pyinfra/queue/queue_manager/pika_queue_manager.py index 4346658..8bc70d4 100644 --- a/pyinfra/queue/queue_manager/pika_queue_manager.py +++ b/pyinfra/queue/queue_manager/pika_queue_manager.py @@ -72,6 +72,7 @@ class PikaQueueManager(QueueManager): self.connection = pika.BlockingConnection(parameters=connection_params) self.channel = self.connection.channel() + self.channel.basic_qos(prefetch_count=1) if not dead_letter_queue: dead_letter_queue = CONFIG.rabbitmq.queues.dead_letter diff --git a/pyinfra/storage/adapters/azure.py b/pyinfra/storage/adapters/azure.py index aa28279..334dbaf 100644 --- a/pyinfra/storage/adapters/azure.py +++ b/pyinfra/storage/adapters/azure.py @@ -1,10 +1,8 @@ -from itertools import repeat import logging -from functools import wraps +from itertools import repeat from operator import attrgetter from azure.storage.blob import ContainerClient, BlobServiceClient -from retry import retry from pyinfra.storage.adapters.adapter import StorageAdapter diff --git a/pyinfra/test/unit_tests/storage_test.py b/pyinfra/test/unit_tests/storage_test.py index 6e8293f..efb7b62 100644 --- a/pyinfra/test/unit_tests/storage_test.py +++ b/pyinfra/test/unit_tests/storage_test.py @@ -2,9 +2,6 @@ import logging import pytest -from pyinfra.storage.storage import Storage -from pyinfra.storage.storages import get_azure_storage, get_s3_storage - logger = logging.getLogger(__name__) logger.setLevel(logging.DEBUG) @@ -46,11 +43,3 @@ class TestStorage: storage.put_object(bucket_name, "file2", b"content 2") full_names_received = storage.get_all_object_names(bucket_name) assert {(bucket_name, "file1"), (bucket_name, "file2")} == {*full_names_received} - - -def test_get_azure_storage_yields_storage(): - assert isinstance(get_azure_storage(), Storage) - - -def test_get_s3_storage_yields_storage(): - assert isinstance(get_s3_storage(), Storage)