ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-15 05:45:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Update test_provider.py

Browse source
This commit is contained in:
Chris Pietschmann 2025-02-18 14:49:03 -05:00
parent f674e28263
commit 9a2e81f5f0
1 changed files with 12 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
backend/open_webui/test/apps/webui/storage/test_provider.py
View file

@ -283,13 +283,6 @@ class TestGCSStorageProvider:
class TestAzureStorageProvider: class TestAzureStorageProvider:
def __init__(self): def __init__(self):
self.Storage = provider.AzureStorageProvider()
self.Storage.endpoint = "https://myaccount.blob.core.windows.net"
self.Storage.container_name = "my-container"
self.file_content = b"test content"
self.filename = "test.txt"
self.filename_extra = "test_extra.txt"
self.file_bytesio_empty = io.BytesIO()
super().__init__() super().__init__()
@pytest.fixture(scope="class") @pytest.fixture(scope="class")
@ -305,12 +298,21 @@ class TestAzureStorageProvider:
mock_container_client.get_blob_client.return_value = mock_blob_client mock_container_client.get_blob_client.return_value = mock_blob_client
# Mock `from_connection_string` and `BlobServiceClient` constructor # 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) monkeypatch.setattr("azure.storage.blob.BlobServiceClient", lambda *_: mock_blob_service_client)
self.Storage = provider.AzureStorageProvider()
self.Storage.endpoint = "https://myaccount.blob.core.windows.net"
self.Storage.container_name = "my-container"
self.file_content = b"test content"
self.filename = "test.txt"
self.filename_extra = "test_extra.txt"
self.file_bytesio_empty = io.BytesIO()
# Apply to instance variables # Apply to instance variables
self.Storage.blob_service_client = mock_blob_service_client #self.Storage.blob_service_client = mock_blob_service_client
self.Storage.container_client = mock_container_client #self.Storage.container_client = mock_container_client
yield yield