OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions 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/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 = {