haiwen/seahub
1
0
Fork 0
mirror of https://github.com/haiwen/seahub.git synced 2025-09-06 17:33:18 +00:00
Code Issues Releases Wiki Activity

[invitation] Check duplicated accepter and add blocklist for accepter

Browse Source
This commit is contained in:
zhengxie
2016-12-19 16:12:47 +08:00
parent 4a99ee32a0
commit 11fdfe7299
5 changed files with 90 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
tests/api/endpoints/test_invitations.py
View File

@@ -1,6 +1,7 @@
import json
from mock import patch
from django.test import override_settings
from post_office.models import Email
from seahub.base.accounts import UserPermissions
@@ -35,6 +36,48 @@ class InvitationsTest(BaseTestCase):
assert len(Invitation.objects.all()) == 1
@patch.object(CanInviteGuest, 'has_permission')
@patch.object(UserPermissions, 'can_invite_guest')
def test_can_not_add_same_email(self, mock_can_invite_guest, mock_has_permission):
mock_can_invite_guest.return_val = True
mock_has_permission.return_val = True
assert len(Invitation.objects.all()) == 0
resp = self.client.post(self.endpoint, {
'type': 'guest',
'accepter': 'some_random_user@1.com',
})
self.assertEqual(201, resp.status_code)
json_resp = json.loads(resp.content)
assert json_resp['inviter'] == self.username
assert json_resp['accepter'] == 'some_random_user@1.com'
assert json_resp['expire_time'] is not None
resp = self.client.post(self.endpoint, {
'type': 'guest',
'accepter': 'some_random_user@1.com',
})
self.assertEqual(400, resp.status_code)
assert len(Invitation.objects.all()) == 1
@override_settings(INVITATION_ACCEPTER_BLACKLIST=["a@a.com", "*@a-a-a.com", r".*@(foo|bar).com"])
@patch.object(CanInviteGuest, 'has_permission')
@patch.object(UserPermissions, 'can_invite_guest')
def test_can_not_add_blocked_email(self, mock_can_invite_guest, mock_has_permission):
mock_can_invite_guest.return_val = True
mock_has_permission.return_val = True
assert len(Invitation.objects.all()) == 0
resp = self.client.post(self.endpoint, {
'type': 'guest',
'accepter': 'some_random_user@a-a-a.com',
})
self.assertEqual(400, resp.status_code)
assert len(Invitation.objects.all()) == 0
@patch.object(CanInviteGuest, 'has_permission')
@patch.object(UserPermissions, 'can_invite_guest')
def test_can_send_mail(self, mock_can_invite_guest, mock_has_permission):