1
0
mirror of https://github.com/haiwen/seahub.git synced 2025-05-30 18:46:19 +00:00
seahub/tests/api/test_accounts.py
WJH 532fa5ef8c
Improve login and create user (#5458)
* add create oauth/ldap/saml user in UserManager

* improve admin add/import users

* improve dingtalk/weixin/work_weixin login/create user

* improve saml login/create user

* improve oauth login/create user

* login ldap user in seahub

* improve invite user

* fix code

* fix github test action

* fix test

* fix saml login

* optimize code

* specify the version of python-ldap

* fix code

* improve code

* add get_old_user

* optimize oauth login code

* optimize code

* remove LDAP_USER_UNIQUE_ID

* remove test_primary_id

* improve authenticate user

* improve saml login

* optimize code
2023-06-19 13:06:15 +08:00

94 lines
3.7 KiB
Python

import requests
from tests.common.utils import randstring
from tests.api.apitestbase import ApiTestBase
from tests.api.urls import PING_URL, AUTH_PING_URL
test_account_username = 'test_%s@test.com' % randstring(10)
test_account_password = randstring(20)
test_account_password2 = randstring(20)
class AccountsApiTest(ApiTestBase):
def test_update_account_passwd(self):
with self.get_tmp_user() as user:
data = {'password': 'new_password'}
self.admin_put(user.user_url, data=data, expected=200)
def test_set_account_to_staff(self):
with self.get_tmp_user() as user:
self.assertEqual(self.admin_get(user.user_url).json()['is_staff'],
False)
data = {'is_staff': 'true'}
self.admin_put(user.user_url, data=data, expected=200)
self.assertEqual(self.admin_get(user.user_url).json()['is_staff'],
True)
def test_set_account_inactive(self):
with self.get_tmp_user() as user:
self.assertEqual(self.admin_get(user.user_url).json()['is_active'],
True)
data = {'is_active': 'false'}
self.admin_put(user.user_url, data=data, expected=200)
self.assertEqual(self.admin_get(user.user_url).json()['is_active'],
False)
def test_set_account_inactive_with_wrong_arg(self):
with self.get_tmp_user() as user:
self.assertEqual(self.admin_get(user.user_url).json()['is_active'],
True)
data = {'is_active': 'fals'}
self.admin_put(user.user_url, data=data, expected=400)
def test_set_account_inactive_with_empty_arg(self):
with self.get_tmp_user() as user:
self.assertEqual(self.admin_get(user.user_url).json()['is_active'],
True)
data = {'is_active': ''}
self.admin_put(user.user_url, data=data, expected=400)
def test_update_account_nickname(self):
with self.get_tmp_user() as user:
data = {'name': 'new nick name'}
self.admin_put(user.user_url, data=data, expected=200)
# def test_update_account_nickname_with_slash(self):
# with self.get_tmp_user() as user:
# data = {'name': 'new /nick name'}
# self.admin_put(user.user_url, data=data, expected=400)
def test_update_account_intro(self):
with self.get_tmp_user() as user:
data = {'note': 'hello, my name is foo'}
self.admin_put(user.user_url, data=data, expected=200)
def test_update_account_storage_quota(self):
with self.get_tmp_user() as user:
data = {'storage': 1024} # 1 Mb
self.admin_put(user.user_url, data=data, expected=200)
self.assertEqual(self.admin_get(user.user_url).json()['total'],
1024000000)
# def test_update_account_sharing_quota(self):
# with self.get_tmp_user() as user:
# data = {'sharing': 1024} # 1KB
# self.admin_put(user.user_url, data=data, expected=200)
# self.assertEqual(self.admin_get(user.user_url).json()['sharing'],
# 1024)
def test_unset_trial_account(self):
with self.get_tmp_user() as user:
data = {'is_trial': 'false'}
self.admin_put(user.user_url, data=data, expected=200)
def test_auth_ping(self):
res = self.get(AUTH_PING_URL)
self.assertRegex(res.text, '"pong"')
res = requests.get(AUTH_PING_URL)
self.assertEqual(res.status_code, 403)
def test_ping(self):
res = requests.get(PING_URL)
self.assertRegex(res.text, '"pong"')
self.assertEqual(res.status_code, 200)