github-starred/open-webui
2
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-05-05 03:38:57 -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

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

@@ -3,17 +3,17 @@ from test.util.mock_user import mock_webui_user
def _get_user_by_id(data, param):
return next((item for item in data if item["id"] == param), None)
return next((item for item in data if item['id'] == param), None)
def _assert_user(data, id, **kwargs):
user = _get_user_by_id(data, id)
assert user is not None
comparison_data = {
"name": f"user {id}",
"email": f"user{id}@openwebui.com",
"profile_image_url": f"/api/v1/users/{id}/profile/image",
"role": "user",
'name': f'user {id}',
'email': f'user{id}@openwebui.com',
'profile_image_url': f'/api/v1/users/{id}/profile/image',
'role': 'user',
**kwargs,
}
for key, value in comparison_data.items():
@@ -21,7 +21,7 @@ def _assert_user(data, id, **kwargs):
class TestUsers(AbstractPostgresTest):
BASE_PATH = "/api/v1/users"
BASE_PATH = '/api/v1/users'
def setup_class(cls):
super().setup_class()
@@ -32,136 +32,134 @@ class TestUsers(AbstractPostgresTest):
def setup_method(self):
super().setup_method()
self.users.insert_new_user(
id="1",
name="user 1",
email="user1@openwebui.com",
profile_image_url="/user1.png",
role="user",
id='1',
name='user 1',
email='user1@openwebui.com',
profile_image_url='/user1.png',
role='user',
)
self.users.insert_new_user(
id="2",
name="user 2",
email="user2@openwebui.com",
profile_image_url="/user2.png",
role="user",
id='2',
name='user 2',
email='user2@openwebui.com',
profile_image_url='/user2.png',
role='user',
)
def test_users(self):
# Get all users
with mock_webui_user(id="3"):
response = self.fast_api_client.get(self.create_url(""))
with mock_webui_user(id='3'):
response = self.fast_api_client.get(self.create_url(''))
assert response.status_code == 200
assert len(response.json()) == 2
data = response.json()
_assert_user(data, "1")
_assert_user(data, "2")
_assert_user(data, '1')
_assert_user(data, '2')
# update role
with mock_webui_user(id="3"):
response = self.fast_api_client.post(
self.create_url("/update/role"), json={"id": "2", "role": "admin"}
)
with mock_webui_user(id='3'):
response = self.fast_api_client.post(self.create_url('/update/role'), json={'id': '2', 'role': 'admin'})
assert response.status_code == 200
_assert_user([response.json()], "2", role="admin")
_assert_user([response.json()], '2', role='admin')
# Get all users
with mock_webui_user(id="3"):
response = self.fast_api_client.get(self.create_url(""))
with mock_webui_user(id='3'):
response = self.fast_api_client.get(self.create_url(''))
assert response.status_code == 200
assert len(response.json()) == 2
data = response.json()
_assert_user(data, "1")
_assert_user(data, "2", role="admin")
_assert_user(data, '1')
_assert_user(data, '2', role='admin')
# Get (empty) user settings
with mock_webui_user(id="2"):
response = self.fast_api_client.get(self.create_url("/user/settings"))
with mock_webui_user(id='2'):
response = self.fast_api_client.get(self.create_url('/user/settings'))
assert response.status_code == 200
assert response.json() is None
# Update user settings
with mock_webui_user(id="2"):
with mock_webui_user(id='2'):
response = self.fast_api_client.post(
self.create_url("/user/settings/update"),
self.create_url('/user/settings/update'),
json={
"ui": {"attr1": "value1", "attr2": "value2"},
"model_config": {"attr3": "value3", "attr4": "value4"},
'ui': {'attr1': 'value1', 'attr2': 'value2'},
'model_config': {'attr3': 'value3', 'attr4': 'value4'},
},
)
assert response.status_code == 200
# Get user settings
with mock_webui_user(id="2"):
response = self.fast_api_client.get(self.create_url("/user/settings"))
with mock_webui_user(id='2'):
response = self.fast_api_client.get(self.create_url('/user/settings'))
assert response.status_code == 200
assert response.json() == {
"ui": {"attr1": "value1", "attr2": "value2"},
"model_config": {"attr3": "value3", "attr4": "value4"},
'ui': {'attr1': 'value1', 'attr2': 'value2'},
'model_config': {'attr3': 'value3', 'attr4': 'value4'},
}
# Get (empty) user info
with mock_webui_user(id="1"):
response = self.fast_api_client.get(self.create_url("/user/info"))
with mock_webui_user(id='1'):
response = self.fast_api_client.get(self.create_url('/user/info'))
assert response.status_code == 200
assert response.json() is None
# Update user info
with mock_webui_user(id="1"):
with mock_webui_user(id='1'):
response = self.fast_api_client.post(
self.create_url("/user/info/update"),
json={"attr1": "value1", "attr2": "value2"},
self.create_url('/user/info/update'),
json={'attr1': 'value1', 'attr2': 'value2'},
)
assert response.status_code == 200
# Get user info
with mock_webui_user(id="1"):
response = self.fast_api_client.get(self.create_url("/user/info"))
with mock_webui_user(id='1'):
response = self.fast_api_client.get(self.create_url('/user/info'))
assert response.status_code == 200
assert response.json() == {"attr1": "value1", "attr2": "value2"}
assert response.json() == {'attr1': 'value1', 'attr2': 'value2'}
# Get user by id
with mock_webui_user(id="1"):
response = self.fast_api_client.get(self.create_url("/2"))
with mock_webui_user(id='1'):
response = self.fast_api_client.get(self.create_url('/2'))
assert response.status_code == 200
assert response.json() == {"name": "user 2", "profile_image_url": "/user2.png"}
assert response.json() == {'name': 'user 2', 'profile_image_url': '/user2.png'}
# Update user by id
with mock_webui_user(id="1"):
with mock_webui_user(id='1'):
response = self.fast_api_client.post(
self.create_url("/2/update"),
self.create_url('/2/update'),
json={
"name": "user 2 updated",
"email": "user2-updated@openwebui.com",
"profile_image_url": "/user2-updated.png",
'name': 'user 2 updated',
'email': 'user2-updated@openwebui.com',
'profile_image_url': '/user2-updated.png',
},
)
assert response.status_code == 200
# Get all users
with mock_webui_user(id="3"):
response = self.fast_api_client.get(self.create_url(""))
with mock_webui_user(id='3'):
response = self.fast_api_client.get(self.create_url(''))
assert response.status_code == 200
assert len(response.json()) == 2
data = response.json()
_assert_user(data, "1")
_assert_user(data, '1')
_assert_user(
data,
"2",
role="admin",
name="user 2 updated",
email="user2-updated@openwebui.com",
profile_image_url=f"/api/v1/users/2/profile/image",
'2',
role='admin',
name='user 2 updated',
email='user2-updated@openwebui.com',
profile_image_url=f'/api/v1/users/2/profile/image',
)
# Delete user by id
with mock_webui_user(id="1"):
response = self.fast_api_client.delete(self.create_url("/2"))
with mock_webui_user(id='1'):
response = self.fast_api_client.delete(self.create_url('/2'))
assert response.status_code == 200
# Get all users
with mock_webui_user(id="3"):
response = self.fast_api_client.get(self.create_url(""))
with mock_webui_user(id='3'):
response = self.fast_api_client.get(self.create_url(''))
assert response.status_code == 200
assert len(response.json()) == 1
data = response.json()
_assert_user(data, "1")
_assert_user(data, '1')