UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-23 13:51:07 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Update test_provider.py

Browse Source
This commit is contained in:
Chris Pietschmann 2025-02-18 15:14:46 -05:00
parent 55bd7a1c65
commit 4c352ff974
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
backend/open_webui/test/apps/webui/storage/test_provider.py
View File

@ -293,19 +293,23 @@ class TestAzureStorageProvider:
mock_blob_service_client.get_container_client.return_value = ( mock_blob_service_client.get_container_client.return_value = (
mock_container_client mock_container_client
) )
mock_container_client.get_blob_client.return_value = ( mock_container_client.get_blob_client.return_value = mock_blob_client
mock_blob_client
)
# Monkeypatch the Azure classes to return our mocks # Monkeypatch the Azure classes to return our mocks
monkeypatch.setattr( monkeypatch.setattr(
azure.storage.blob, "BlobServiceClient", lambda *args, **kwargs: mock_blob_service_client azure.storage.blob,
"BlobServiceClient",
lambda *args, **kwargs: mock_blob_service_client
) )
monkeypatch.setattr( monkeypatch.setattr(
azure.storage.blob, "BlobContainerClient", lambda *args, **kwargs: mock_container_client azure.storage.blob,
"BlobContainerClient",
lambda *args, **kwargs: mock_container_client
) )
monkeypatch.setattr( monkeypatch.setattr(
azure.storage.blob, "BlobClient", lambda *args, **kwargs: mock_blob_client azure.storage.blob,
"BlobClient",
lambda *args, **kwargs: mock_blob_client
) )