Sfoglia il codice sorgente

Update test_provider.py

Chris Pietschmann 2 mesi fa
parent
commit
55bd7a1c65

+ 9 - 3
backend/open_webui/test/apps/webui/storage/test_provider.py

@@ -290,8 +290,12 @@ class TestAzureStorageProvider:
         mock_blob_client = MagicMock()
         mock_blob_client = MagicMock()
 
 
         # Set up return values for the mock
         # Set up return values for the mock
-        mock_blob_service_client.get_container_client.return_value = mock_container_client
-        mock_container_client.get_blob_client.return_value = mock_blob_client
+        mock_blob_service_client.get_container_client.return_value = (
+            mock_container_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(
@@ -321,7 +325,9 @@ class TestAzureStorageProvider:
         upload_dir = mock_upload_dir(monkeypatch, tmp_path)
         upload_dir = mock_upload_dir(monkeypatch, tmp_path)
 
 
         # Simulate an error when container does not exist
         # Simulate an error when container does not exist
-        self.Storage.container_client.get_blob_client.side_effect = Exception("Container does not exist")
+        self.Storage.container_client.get_blob_client.side_effect = Exception(
+            "Container does not exist"
+        )
         with pytest.raises(Exception):
         with pytest.raises(Exception):
             self.Storage.upload_file(io.BytesIO(self.file_content), self.filename)
             self.Storage.upload_file(io.BytesIO(self.file_content), self.filename)