Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 03:47:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2024-12-10 00:54:13 -08:00
parent f6bec8d9f3
commit d3d161f723
112 changed files with 1217 additions and 1165 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/test/apps/webui/routers/test_auths.py
View File

@@ -7,8 +7,8 @@ class TestAuths(AbstractPostgresTest):
def setup_class(cls):
super().setup_class()
from open_webui.apps.webui.models.auths import Auths
from open_webui.apps.webui.models.users import Users
from open_webui.models.auths import Auths
from open_webui.models.users import Users
cls.users = Users
cls.auths = Auths

4
backend/open_webui/test/apps/webui/routers/test_chats.py
View File

@@ -12,7 +12,7 @@ class TestChats(AbstractPostgresTest):
def setup_method(self):
super().setup_method()
from open_webui.apps.webui.models.chats import ChatForm, Chats
from open_webui.models.chats import ChatForm, Chats
self.chats = Chats
self.chats.insert_new_chat(
@@ -88,7 +88,7 @@ class TestChats(AbstractPostgresTest):
def test_get_user_archived_chats(self):
self.chats.archive_all_chats_by_user_id("2")
from open_webui.apps.webui.internal.db import Session
from open_webui.internal.db import Session
Session.commit()
with mock_webui_user(id="2"):

2
backend/open_webui/test/apps/webui/routers/test_models.py
View File

@@ -7,7 +7,7 @@ class TestModels(AbstractPostgresTest):
def setup_class(cls):
super().setup_class()
from open_webui.apps.webui.models.models import Model
from open_webui.models.models import Model
cls.models = Model

2
backend/open_webui/test/apps/webui/routers/test_users.py
View File

@@ -25,7 +25,7 @@ class TestUsers(AbstractPostgresTest):
def setup_class(cls):
super().setup_class()
from open_webui.apps.webui.models.users import Users
from open_webui.models.users import Users
cls.users = Users

4
backend/open_webui/test/util/abstract_integration_test.py
View File

@@ -115,7 +115,7 @@ class AbstractPostgresTest(AbstractIntegrationTest):
pytest.fail(f"Could not setup test environment: {ex}")
def _check_db_connection(self):
from open_webui.apps.webui.internal.db import Session
from open_webui.internal.db import Session
retries = 10
while retries > 0:
@@ -139,7 +139,7 @@ class AbstractPostgresTest(AbstractIntegrationTest):
cls.docker_client.containers.get(cls.DOCKER_CONTAINER_NAME).remove(force=True)
def teardown_method(self):
from open_webui.apps.webui.internal.db import Session
from open_webui.internal.db import Session
# rollback everything not yet committed
Session.commit()

4
backend/open_webui/test/util/mock_user.py
View File

@@ -5,7 +5,7 @@ from fastapi import FastAPI
@contextmanager
def mock_webui_user(**kwargs):
from open_webui.apps.webui.main import app
from backend.open_webui.routers.webui import app
with mock_user(app, **kwargs):
yield
@@ -19,7 +19,7 @@ def mock_user(app: FastAPI, **kwargs):
get_admin_user,
get_current_user_by_api_key,
)
from open_webui.apps.webui.models.users import User
from open_webui.models.users import User
def create_user():
user_parameters = {