Jelajahi Sumber

Update test_provider.py

Chris Pietschmann 2 bulan lalu
induk
melakukan
4c352ff974
1 mengubah file dengan 10 tambahan dan 6 penghapusan
  1. 10 6
      backend/open_webui/test/apps/webui/storage/test_provider.py

+ 10 - 6
backend/open_webui/test/apps/webui/storage/test_provider.py

@@ -293,19 +293,23 @@ class TestAzureStorageProvider:
         mock_blob_service_client.get_container_client.return_value = (
             mock_container_client
         )
-        mock_container_client.get_blob_client.return_value = (
-            mock_blob_client
-        )
+        mock_container_client.get_blob_client.return_value = mock_blob_client
 
         # Monkeypatch the Azure classes to return our mocks
         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(
-            azure.storage.blob, "BlobContainerClient", lambda *args, **kwargs: mock_container_client
+            azure.storage.blob,
+            "BlobContainerClient",
+            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
         )