1
0
mirror of https://github.com/haiwen/seahub.git synced 2025-07-04 02:36:59 +00:00
seahub/tests/api/endpoints/test_invitations.py

72 lines
2.5 KiB
Python
Raw Normal View History

2016-05-13 07:34:49 +00:00
import json
from mock import patch
from post_office.models import Email
from seahub.base.accounts import UserPermissions
from seahub.invitations.models import Invitation
from seahub.test_utils import BaseTestCase
2016-07-20 07:30:39 +00:00
from seahub.api2.permissions import CanInviteGuest
2016-05-13 07:34:49 +00:00
class InvitationsTest(BaseTestCase):
def setUp(self):
self.login_as(self.user)
self.endpoint = '/api/v2.1/invitations/'
self.username = self.user.username
2016-07-20 07:30:39 +00:00
@patch.object(CanInviteGuest, 'has_permission')
2016-05-13 07:34:49 +00:00
@patch.object(UserPermissions, 'can_invite_guest')
2016-07-20 07:30:39 +00:00
def test_can_add(self, mock_can_invite_guest, mock_has_permission):
2016-05-13 07:34:49 +00:00
mock_can_invite_guest.return_val = True
2016-07-20 07:30:39 +00:00
mock_has_permission.return_val = True
2016-05-13 07:34:49 +00:00
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)
2016-07-12 08:34:20 +00:00
assert json_resp['inviter'] == self.username
assert json_resp['accepter'] == 'some_random_user@1.com'
assert json_resp['expire_time'] is not None
2016-05-13 07:34:49 +00:00
assert len(Invitation.objects.all()) == 1
2016-07-20 07:30:39 +00:00
@patch.object(CanInviteGuest, 'has_permission')
2016-05-13 07:34:49 +00:00
@patch.object(UserPermissions, 'can_invite_guest')
2016-07-20 07:30:39 +00:00
def test_can_send_mail(self, mock_can_invite_guest, mock_has_permission):
2016-05-13 07:34:49 +00:00
mock_can_invite_guest.return_val = True
2016-07-20 07:30:39 +00:00
mock_has_permission.return_val = True
2016-05-13 07:34:49 +00:00
self.assertEqual(len(Email.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)
self.assertEqual(len(Email.objects.all()), 1)
self.assertRegexpMatches(Email.objects.all()[0].html_message,
2016-07-12 08:34:20 +00:00
json_resp['token'])
2016-05-13 07:34:49 +00:00
2016-07-20 07:30:39 +00:00
@patch.object(CanInviteGuest, 'has_permission')
2016-05-13 07:34:49 +00:00
@patch.object(UserPermissions, 'can_invite_guest')
2016-07-20 07:30:39 +00:00
def test_can_list(self, mock_can_invite_guest, mock_has_permission):
2016-05-13 07:34:49 +00:00
mock_can_invite_guest.return_val = True
2016-07-20 07:30:39 +00:00
mock_has_permission.return_val = True
2016-05-13 07:34:49 +00:00
Invitation.objects.add(inviter=self.username, accepter='1@1.com')
Invitation.objects.add(inviter=self.username, accepter='1@2.com')
resp = self.client.get(self.endpoint)
self.assertEqual(200, resp.status_code)
json_resp = json.loads(resp.content)
2016-07-12 08:34:20 +00:00
assert len(json_resp) == 2