瀏覽代碼

Merge pull request #8712 from rragundez/improve-s3-provider-tests

Improve s3 provider tests
Timothy Jaeryang Baek 3 月之前
父節點
當前提交
ffd10e8b6f
共有 1 個文件被更改,包括 11 次插入7 次删除
  1. 11 7
      backend/open_webui/test/apps/webui/storage/test_provider.py

+ 11 - 7
backend/open_webui/test/apps/webui/storage/test_provider.py

@@ -87,13 +87,17 @@ class TestLocalStorageProvider:
 
 @mock_aws
 class TestS3StorageProvider:
-    Storage = provider.S3StorageProvider()
-    Storage.bucket_name = "my-bucket"
-    s3_client = boto3.resource("s3", region_name="us-east-1")
-    file_content = b"test content"
-    filename = "test.txt"
-    filename_extra = "test_exyta.txt"
-    file_bytesio_empty = io.BytesIO()
+
+    def __init__(self):
+        self.Storage = provider.S3StorageProvider()
+        self.Storage.bucket_name = "my-bucket"
+        self.s3_client = boto3.resource("s3", region_name="us-east-1")
+        self.file_content = b"test content"
+        self.filename = "test.txt"
+        self.filename_extra = "test_exyta.txt"
+        self.file_bytesio_empty = io.BytesIO()
+        super().__init__()
+
 
     def test_upload_file(self, monkeypatch, tmp_path):
         upload_dir = mock_upload_dir(monkeypatch, tmp_path)