mirror of
https://github.com/haiwen/seahub.git
synced 2025-07-13 06:55:59 +00:00
7df1192cbf
* admin api role check * add check to new tabs and tests * fix work-weixin virus scan check
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
import os
|
|
import json
|
|
from mock import patch
|
|
from django.core.urlresolvers import reverse
|
|
|
|
from seahub.api2.endpoints.admin import license as license_api
|
|
from seahub.settings import LICENSE_PATH
|
|
from seahub.utils.error_msg import file_type_error_msg
|
|
from seahub.test_utils import BaseTestCase
|
|
from tests.common.utils import urljoin
|
|
from tests.common.common import BASE_URL
|
|
|
|
|
|
class AdminLicenseTest(BaseTestCase):
|
|
def setUp(self):
|
|
self.login_as(self.admin)
|
|
|
|
def test_post_admin_permission_denied(self):
|
|
self.logout()
|
|
self.login_as(self.admin_cannot_config_system)
|
|
resp = self.client.post(reverse('api-v2.1-admin-license'))
|
|
self.assertEqual(403, resp.status_code)
|
|
|
|
@patch.object(license_api, 'ccnet_api')
|
|
def test_update_license(self, mock_ccnet_api):
|
|
mock_ccnet_api.return_val = {}
|
|
|
|
url = reverse('api-v2.1-admin-license')
|
|
url = urljoin(BASE_URL, url)
|
|
with open(
|
|
os.path.join(os.getcwd(), 'tests/seahub/utils/seafile-license.txt'), 'rb') as f:
|
|
resp = self.client.post(url, {'license': f})
|
|
json_resp = json.loads(resp.content)
|
|
|
|
assert json_resp['license_expiration'] is not None
|
|
assert json_resp['license_mode'] is not None
|
|
assert json_resp['license_maxusers'] is not None
|
|
assert json_resp['license_to'] is not None
|
|
assert os.path.exists(LICENSE_PATH)
|
|
|
|
@patch.object(license_api, 'ccnet_api')
|
|
def test_update_license_with_invalid_type(self, mock_ccnet_api):
|
|
mock_ccnet_api.return_val = {}
|
|
|
|
url = reverse('api-v2.1-admin-license')
|
|
url = urljoin(BASE_URL, url)
|
|
with open('temp.notxt', 'w') as f:
|
|
f.write('1')
|
|
|
|
with open(
|
|
os.path.join(os.getcwd(), 'temp.notxt'), 'rb') as f:
|
|
resp = self.client.post(url, {'license': f})
|
|
json_resp = json.loads(resp.content)
|
|
assert 400 == resp.status_code
|
|
assert file_type_error_msg('notxt', 'txt') == json_resp['error_msg']
|
|
|