mirror of
https://github.com/haiwen/seahub.git
synced 2025-07-12 14:38:58 +00:00
103 lines
2.8 KiB
Python
103 lines
2.8 KiB
Python
import json
|
|
from mock import patch
|
|
|
|
from seaserv import ccnet_api, seafile_api
|
|
from django.urls import reverse
|
|
from seahub.test_utils import BaseTestCase
|
|
from tests.common.utils import randstring
|
|
|
|
from seaserv import seafserv_threaded_rpc
|
|
|
|
try:
|
|
from seahub.settings import LOCAL_PRO_DEV_ENV
|
|
except ImportError:
|
|
LOCAL_PRO_DEV_ENV = False
|
|
|
|
|
|
def remove_org(org_id):
|
|
org_id = int(org_id)
|
|
org = ccnet_api.get_org_by_id(org_id)
|
|
if org:
|
|
users =ccnet_api.get_org_emailusers(org.url_prefix, -1, -1)
|
|
for u in users:
|
|
ccnet_api.remove_org_user(org_id, u.email)
|
|
|
|
groups = ccnet_api.get_org_groups(org.org_id, -1, -1)
|
|
for g in groups:
|
|
ccnet_api.remove_org_group(org_id, g.gid)
|
|
|
|
# remove org repos
|
|
seafserv_threaded_rpc.remove_org_repo_by_org_id(org_id)
|
|
|
|
# remove org
|
|
ccnet_api.remove_org(org_id)
|
|
|
|
|
|
class OrgGroupsTest(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
|
|
self.user_name = self.user.username
|
|
self.admin_name = self.admin.username
|
|
|
|
if LOCAL_PRO_DEV_ENV:
|
|
self.org_name = randstring(6)
|
|
self.org_url_prefix = randstring(6)
|
|
tmp_user = self.create_user(email='%s@%s.com' % (randstring(6), randstring(6)))
|
|
self.org_creator = tmp_user.username
|
|
|
|
self.org_id = ccnet_api.create_org(self.org_name,
|
|
self.org_url_prefix, self.org_creator)
|
|
self.org_groups_url = reverse('api-v2.1-admin-org-groups',
|
|
args=[self.org_id])
|
|
|
|
def tearDown(self):
|
|
self.remove_group()
|
|
self.remove_repo()
|
|
|
|
if LOCAL_PRO_DEV_ENV:
|
|
remove_org(self.org_id)
|
|
self.remove_user(self.org_creator)
|
|
|
|
def test_can_get_groups(self):
|
|
|
|
if not LOCAL_PRO_DEV_ENV:
|
|
return
|
|
|
|
self.login_as(self.admin)
|
|
resp = self.client.get(self.org_groups_url)
|
|
json_resp = json.loads(resp.content)
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
for repo in json_resp['group_list']:
|
|
assert repo['org_id'] == self.org_id
|
|
|
|
def test_no_permission(self):
|
|
|
|
if not LOCAL_PRO_DEV_ENV:
|
|
return
|
|
|
|
self.login_as(self.admin_no_other_permission)
|
|
resp = self.client.get(self.org_groups_url)
|
|
self.assertEqual(403, resp.status_code)
|
|
|
|
def test_can_not_get_repos_if_not_admin(self):
|
|
|
|
if not LOCAL_PRO_DEV_ENV:
|
|
return
|
|
|
|
self.login_as(self.user)
|
|
resp = self.client.get(self.org_groups_url)
|
|
self.assertEqual(403, resp.status_code)
|
|
|
|
def test_get_with_invalid_org_id(self):
|
|
|
|
if not LOCAL_PRO_DEV_ENV:
|
|
return
|
|
|
|
self.login_as(self.admin)
|
|
invalid_org_groups_url = reverse('api-v2.1-admin-org-groups', args=[0])
|
|
|
|
resp = self.client.get(invalid_org_groups_url)
|
|
self.assertEqual(400, resp.status_code)
|