Merge branch 'master' of ssh://git.iqser.com:2222/rr/pyinfra into sonarscan_refac

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

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
 

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)