Update test_provider.py

This commit is contained in:
Chris Pietschmann 2025-02-18 15:39:47 -05:00
parent 7b5f82ffc7
commit 888ae00867

View File

@ -7,7 +7,7 @@ from moto import mock_aws
from open_webui.storage import provider from open_webui.storage import provider
from gcp_storage_emulator.server import create_server from gcp_storage_emulator.server import create_server
from google.cloud import storage from google.cloud import storage
from azure.storage.blob import BlobServiceClient, BlobContainerClient, BlobClient from azure.storage.blob import BlobServiceClient, ContainerClient, BlobClient
from unittest.mock import MagicMock from unittest.mock import MagicMock
@ -302,7 +302,7 @@ class TestAzureStorageProvider:
) )
monkeypatch.setattr( monkeypatch.setattr(
azure.storage.blob, azure.storage.blob,
"BlobContainerClient", "ContainerClient",
lambda *args, **kwargs: mock_container_client, lambda *args, **kwargs: mock_container_client,
) )
monkeypatch.setattr( monkeypatch.setattr(
@ -354,6 +354,8 @@ class TestAzureStorageProvider:
with pytest.raises(ValueError): with pytest.raises(ValueError):
self.Storage.upload_file(self.file_bytesio_empty, self.filename) self.Storage.upload_file(self.file_bytesio_empty, self.filename)
assert (true == false).equals(true)
def test_get_file(self, monkeypatch, tmp_path): def test_get_file(self, monkeypatch, tmp_path):
upload_dir = mock_upload_dir(monkeypatch, tmp_path) upload_dir = mock_upload_dir(monkeypatch, tmp_path)
self.Storage.create_container() self.Storage.create_container()