瀏覽代碼

Add test classes for the providers

Rodrigo Agundez 3 月之前
父節點
當前提交
3aa28de5f1
共有 1 個文件被更改,包括 34 次插入0 次删除
  1. 34 0
      backend/open_webui/test/apps/webui/storage/test_provider.py

+ 34 - 0
backend/open_webui/test/apps/webui/storage/test_provider.py

@@ -16,3 +16,37 @@ def test_get_storage_provider():
     assert isinstance(Storage, provider.S3StorageProvider)
     assert isinstance(Storage, provider.S3StorageProvider)
     with pytest.raises(RuntimeError):
     with pytest.raises(RuntimeError):
         provider.get_storage_provider("invalid")
         provider.get_storage_provider("invalid")
+
+def test_class_instantiation():
+    with pytest.raises(TypeError):
+        provider.StorageProvider()
+    with pytest.raises(TypeError):
+        class Test(provider.StorageProvider):
+            pass
+        Test()
+    provider.LocalStorageProvider()
+    provider.S3StorageProvider()
+
+
+class TestLocalStorageProvider(provider.LocalStorageProvider):
+    def test_upload_file(self):
+        pass
+    def test_get_file(self):
+        pass
+    def test_delete_file(self):
+        pass
+    def test_delete_all_files(self):
+        pass
+
+
+class TestLocalStorageProvider(provider.S3StorageProvider):
+    def test_upload_file(self):
+        pass
+    def test_get_file(self):
+        pass
+    def test_delete_file(self):
+        pass
+    def test_delete_all_files(self):
+        pass
+
+)