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:13:30 -05:00
parent 9c8c837ab9
commit 55bd7a1c65
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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_blob_service_client.get_container_client.return_value = (
mock_container_client.get_blob_client.return_value = mock_blob_client 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)