mirror of
https://github.com/open-webui/open-webui
synced 2024-12-26 05:42:05 +00:00
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
from contextlib import contextmanager
|
||
|
|
|
||
|
|
from fastapi import FastAPI
|
||
|
|
|
||
|
|
|
||
|
|
@contextmanager
|
||
|
|
def mock_webui_user(**kwargs):
|
||
|
|
from apps.webui.main import app
|
||
|
|
|
||
|
|
with mock_user(app, **kwargs):
|
||
|
|
yield
|
||
|
|
|
||
|
|
|
||
|
|
@contextmanager
|
||
|
|
def mock_user(app: FastAPI, **kwargs):
|
||
|
|
from utils.utils import (
|
||
|
|
get_current_user,
|
||
|
|
get_verified_user,
|
||
|
|
get_admin_user,
|
||
|
|
get_current_user_by_api_key,
|
||
|
|
)
|
||
|
|
from apps.webui.models.users import User
|
||
|
|
|
||
|
|
def create_user():
|
||
|
|
user_parameters = {
|
||
|
|
"id": "1",
|
||
|
|
"name": "John Doe",
|
||
|
|
"email": "john.doe@openwebui.com",
|
||
|
|
"role": "user",
|
||
|
|
"profile_image_url": "/user.png",
|
||
|
|
"last_active_at": 1627351200,
|
||
|
|
"updated_at": 1627351200,
|
||
|
|
"created_at": 162735120,
|
||
|
|
**kwargs,
|
||
|
|
}
|
||
|
|
return User(**user_parameters)
|
||
|
|
|
||
|
|
app.dependency_overrides = {
|
||
|
|
get_current_user: create_user,
|
||
|
|
get_verified_user: create_user,
|
||
|
|
get_admin_user: create_user,
|
||
|
|
get_current_user_by_api_key: create_user,
|
||
|
|
}
|
||
|
|
yield
|
||
|
|
app.dependency_overrides = {}
|