Bläddra i källkod

Add test classes for the providers

Rodrigo Agundez 3 månader sedan
förälder
incheckning
3aa28de5f1
1 ändrade filer med 34 tillägg och 0 borttagningar
  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)
     with pytest.raises(RuntimeError):
         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
+
+)