updates to formatting

This commit is contained in:
Chris Pietschmann 2025-02-18 15:17:49 -05:00
parent 4c352ff974
commit a29f83c4e7
2 changed files with 7 additions and 8 deletions

View File

@ -228,7 +228,6 @@ class GCSStorageProvider(StorageProvider):
LocalStorageProvider.delete_all_files()
class AzureStorageProvider(StorageProvider):
def __init__(self):
self.endpoint = AZURE_STORAGE_ENDPOINT
@ -246,7 +245,9 @@ class AzureStorageProvider(StorageProvider):
self.blob_service_client = BlobServiceClient(
account_url=self.endpoint, credential=DefaultAzureCredential()
)
self.container_client = self.blob_service_client.get_container_client(self.container_name)
self.container_client = self.blob_service_client.get_container_client(
self.container_name
)
def upload_file(self, file: BinaryIO, filename: str) -> Tuple[bytes, str]:
"""Handles uploading of the file to Azure Blob Storage."""
@ -293,7 +294,7 @@ class AzureStorageProvider(StorageProvider):
# Always delete from local storage
LocalStorageProvider.delete_all_files()
def get_storage_provider(storage_provider: str):
if storage_provider == "local":

View File

@ -299,17 +299,15 @@ class TestAzureStorageProvider:
monkeypatch.setattr(
azure.storage.blob,
"BlobServiceClient",
lambda *args, **kwargs: mock_blob_service_client
lambda *args, **kwargs: mock_blob_service_client,
)
monkeypatch.setattr(
azure.storage.blob,
"BlobContainerClient",
lambda *args, **kwargs: mock_container_client
lambda *args, **kwargs: mock_container_client,
)
monkeypatch.setattr(
azure.storage.blob,
"BlobClient",
lambda *args, **kwargs: mock_blob_client
azure.storage.blob, "BlobClient", lambda *args, **kwargs: mock_blob_client
)