Browse Source

Update test_provider.py

Chris Pietschmann 2 months ago
parent
commit
4a9a88b683
1 changed files with 1 additions and 0 deletions
  1. 1 0
      backend/open_webui/test/apps/webui/storage/test_provider.py

+ 1 - 0
backend/open_webui/test/apps/webui/storage/test_provider.py

@@ -305,6 +305,7 @@ class TestAzureStorageProvider:
         mock_container_client.get_blob_client.return_value = mock_blob_client
 
         # Mock `from_connection_string` and `BlobServiceClient` constructor
+        monkeypatch.setattr(provider, "BlobServiceClient", lambda *_: mock_blob_service_client)
         monkeypatch.setattr("azure.storage.blob.BlobServiceClient", lambda *_: mock_blob_service_client)
 
         # Apply to instance variables