浏览代码

Update test_provider.py

Chris Pietschmann 2 月之前
父节点
当前提交
4c352ff974
共有 1 个文件被更改,包括 10 次插入6 次删除
  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_blob_service_client.get_container_client.return_value = (
             mock_container_client
             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 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
         )
         )