Browse Source

Start testing files for provider

Rodrigo Agundez 6 months ago
parent
commit
535e59af8d
1 changed files with 18 additions and 0 deletions
  1. 18 0
      backend/open_webui/test/apps/webui/storage/test_provider.py

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

@@ -0,0 +1,18 @@
+import pytest
+from open_webui.storage import provider
+
+
+def test_imports():
+    provider.StorageProvider
+    provider.LocalStorageProvider
+    provider.S3StorageProvider
+    provider.Storage
+
+
+def test_get_storage_provider():
+    Storage = provider.get_storage_provider("local")
+    assert isinstance(Storage, provider.LocalStorageProvider)
+    Storage = provider.get_storage_provider("s3")
+    assert isinstance(Storage, provider.S3StorageProvider)
+    with pytest.raises(RuntimeError):
+        provider.get_storage_provider("invalid")