github-starred/open-webui
2
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-05-03 18:59:38 -05:00
Code Issues 915 Packages Projects Releases 126 Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-03-17 17:58:01 -05:00
parent fcf7208352
commit de3317e26b
220 changed files with 17200 additions and 22836 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,7 +3,7 @@ from test.util.mock_user import mock_webui_user
class TestModels(AbstractPostgresTest):
BASE_PATH = "/api/v1/models"
BASE_PATH = '/api/v1/models'
def setup_class(cls):
super().setup_class()
@@ -12,50 +12,46 @@ class TestModels(AbstractPostgresTest):
cls.models = Model
def test_models(self):
with mock_webui_user(id="2"):
response = self.fast_api_client.get(self.create_url("/"))
with mock_webui_user(id='2'):
response = self.fast_api_client.get(self.create_url('/'))
assert response.status_code == 200
assert len(response.json()) == 0
with mock_webui_user(id="2"):
with mock_webui_user(id='2'):
response = self.fast_api_client.post(
self.create_url("/add"),
self.create_url('/add'),
json={
"id": "my-model",
"base_model_id": "base-model-id",
"name": "Hello World",
"meta": {
"profile_image_url": "/static/favicon.png",
"description": "description",
"capabilities": None,
"model_config": {},
'id': 'my-model',
'base_model_id': 'base-model-id',
'name': 'Hello World',
'meta': {
'profile_image_url': '/static/favicon.png',
'description': 'description',
'capabilities': None,
'model_config': {},
},
"params": {},
'params': {},
},
)
assert response.status_code == 200
with mock_webui_user(id="2"):
response = self.fast_api_client.get(self.create_url("/"))
with mock_webui_user(id='2'):
response = self.fast_api_client.get(self.create_url('/'))
assert response.status_code == 200
assert len(response.json()) == 1
with mock_webui_user(id="2"):
response = self.fast_api_client.get(
self.create_url(query_params={"id": "my-model"})
)
with mock_webui_user(id='2'):
response = self.fast_api_client.get(self.create_url(query_params={'id': 'my-model'}))
assert response.status_code == 200
data = response.json()[0]
assert data["id"] == "my-model"
assert data["name"] == "Hello World"
assert data['id'] == 'my-model'
assert data['name'] == 'Hello World'
with mock_webui_user(id="2"):
response = self.fast_api_client.delete(
self.create_url("/delete?id=my-model")
)
with mock_webui_user(id='2'):
response = self.fast_api_client.delete(self.create_url('/delete?id=my-model'))
assert response.status_code == 200
with mock_webui_user(id="2"):
response = self.fast_api_client.get(self.create_url("/"))
with mock_webui_user(id='2'):
response = self.fast_api_client.get(self.create_url('/'))
assert response.status_code == 200
assert len(response.json()) == 0