mirror of
https://github.com/haiwen/seahub.git
synced 2025-07-03 18:26:36 +00:00
532fa5ef8c
* 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
88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
import json
|
|
from seahub.test_utils import BaseTestCase
|
|
from seahub.notifications.models import UserNotification
|
|
from seahub.base.accounts import UserManager
|
|
|
|
class NotificationsTest(BaseTestCase):
|
|
def setUp(self):
|
|
self.endpoint = '/api/v2.1/notifications/'
|
|
self.username = self.user.username
|
|
|
|
def test_can_get_unseen_count(self):
|
|
|
|
UserNotification.objects.add_file_uploaded_msg(self.username, 'test')
|
|
|
|
self.login_as(self.user)
|
|
resp = self.client.get(self.endpoint)
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
json_resp = json.loads(resp.content)
|
|
assert json_resp['unseen_count'] == 1
|
|
|
|
def test_get_with_invalid_user_permission(self):
|
|
|
|
resp = self.client.get(self.endpoint)
|
|
self.assertEqual(403, resp.status_code)
|
|
|
|
def test_can_unseen_all_notifications(self):
|
|
|
|
UserNotification.objects.add_file_uploaded_msg(self.username, 'test')
|
|
assert UserNotification.objects.count_unseen_user_notifications(self.username) == 1
|
|
|
|
self.login_as(self.user)
|
|
resp = self.client.put(self.endpoint, {}, 'application/x-www-form-urlencoded')
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
assert UserNotification.objects.count_unseen_user_notifications(self.username) == 0
|
|
|
|
def test_unseen_notifications_with_invalid_user_permission(self):
|
|
|
|
UserNotification.objects.add_file_uploaded_msg(self.username, 'test')
|
|
assert UserNotification.objects.count_unseen_user_notifications(self.username) == 1
|
|
|
|
resp = self.client.put(self.endpoint, {}, 'application/x-www-form-urlencoded')
|
|
self.assertEqual(403, resp.status_code)
|
|
|
|
class NotificationTest(BaseTestCase):
|
|
def setUp(self):
|
|
self.endpoint = '/api/v2.1/notification/'
|
|
self.username = self.user.username
|
|
|
|
def test_can_unseen_notification_by_id(self):
|
|
|
|
notice = UserNotification.objects.add_file_uploaded_msg(self.username, 'test')
|
|
assert UserNotification.objects.count_unseen_user_notifications(self.username) == 1
|
|
|
|
self.login_as(self.user)
|
|
data = 'notice_id=%d' % notice.id
|
|
resp = self.client.put(self.endpoint, data, 'application/x-www-form-urlencoded')
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
assert UserNotification.objects.count_unseen_user_notifications(self.username) == 0
|
|
|
|
def test_argument_check_notice_id_invalid(self):
|
|
self.login_as(self.user)
|
|
data = 'notice_id=%s' % 'a'
|
|
|
|
resp = self.client.put(self.endpoint, data, 'application/x-www-form-urlencoded')
|
|
self.assertEqual(400, resp.status_code)
|
|
|
|
def test_resource_check_notification_not_found(self):
|
|
self.login_as(self.user)
|
|
notice1 = UserNotification.objects.add_user_message(self.username, 'test1')
|
|
notice2 = UserNotification.objects.add_user_message(self.username, 'test2')
|
|
data = 'notice_id=%s' % str(notice2.id + 1)
|
|
|
|
resp = self.client.put(self.endpoint, data, 'application/x-www-form-urlencoded')
|
|
self.assertEqual(404, resp.status_code)
|
|
|
|
def test_permission_check_permission_denied(self):
|
|
self.login_as(self.user)
|
|
new_user = self.create_user(email='new@new.com', password='root')
|
|
notice_to_new_user = UserNotification.objects.add_user_message(new_user.name, 'test for new user')
|
|
data = 'notice_id=%s' % notice_to_new_user.id
|
|
|
|
resp = self.client.put(self.endpoint, data, 'application/x-www-form-urlencoded')
|
|
self.assertEqual(403, resp.status_code)
|
|
|