/
Netrunners
/
Millenium_UI
Обзор
Документация
Войти
/
Netrunners
/
Millenium_UI
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
backend/open_webui/test/util/mock_user.py
45 строк
1 KB
Валерий Конюхов
first_commit
08 июл 2025, 22:20
08 июл 2025, 22:20
a4e8dc9
Код
Авторство
О чём код?
from contextlib import contextmanager from fastapi import FastAPI @contextmanager def mock_webui_user(**kwargs): from open_webui.routers.webui import app with mock_user(app, **kwargs): yield @contextmanager def mock_user(app: FastAPI, **kwargs): from open_webui.utils.auth import ( get_current_user, get_verified_user, get_admin_user, get_current_user_by_api_key, ) from open_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 = {}