1
0
mirror of https://github.com/haiwen/seahub.git synced 2025-09-17 07:41:26 +00:00

[api] Add guest invite feature

This commit is contained in:
zhengxie
2016-05-13 15:34:49 +08:00
committed by lian
parent f29855a3c1
commit 7452c0cfc2
29 changed files with 731 additions and 0 deletions

View File

@@ -0,0 +1,63 @@
import json
from mock import patch
from post_office.models import Email
from django.core.urlresolvers import reverse
from seahub.base.accounts import UserPermissions
from seahub.invitations.models import Invitation
from seahub.test_utils import BaseTestCase
class InvitationsTest(BaseTestCase):
def setUp(self):
self.login_as(self.user)
self.endpoint = '/api/v2.1/invitations/'
self.username = self.user.username
@patch.object(UserPermissions, 'can_invite_guest')
def test_can_add(self, mock_can_invite_guest):
mock_can_invite_guest.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['accepter_exists'] is False
assert json_resp['invitation']['inviter'] == self.username
assert json_resp['invitation']['accepter'] == 'some_random_user@1.com'
assert json_resp['invitation']['expire_time'] is not None
assert len(Invitation.objects.all()) == 1
@patch.object(UserPermissions, 'can_invite_guest')
def test_can_send_mail(self, mock_can_invite_guest):
mock_can_invite_guest.return_val = True
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,
json_resp['invitation']['token'])
@patch.object(UserPermissions, 'can_invite_guest')
def test_can_list(self, mock_can_invite_guest):
mock_can_invite_guest.return_val = True
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)
assert len(json_resp['invitations']) == 2