mirror of
https://github.com/haiwen/seahub.git
synced 2025-07-05 03:07:01 +00:00
77 lines
3.2 KiB
Python
77 lines
3.2 KiB
Python
import json
|
|
from mock import patch
|
|
from django.core.urlresolvers import reverse
|
|
from django.test import override_settings
|
|
from seahub.test_utils import BaseTestCase
|
|
from seahub.trusted_ip.models import TrustedIP
|
|
from seahub.api2.endpoints.admin.device_trusted_ip import cmp_ip
|
|
|
|
@override_settings(ENABLE_LIMIT_IPADDRESS=True)
|
|
class DeviceAccessibleIpSetting(BaseTestCase):
|
|
def setUp(self):
|
|
self.url = reverse('api-v2.1-admin-device-trusted-ip')
|
|
self.test_ip = '123.213.132.111'
|
|
TrustedIP.objects.get_or_create('127.0.0.1')
|
|
TrustedIP.objects.get_or_create('1.1.1.1')
|
|
self.login_as(self.admin)
|
|
|
|
@patch('seahub.api2.permissions.IsProVersion.has_permission')
|
|
def test_can_get(self, mock_IsProVersion):
|
|
mock_IsProVersion.return_value= True
|
|
resp = self.client.get(self.url)
|
|
assert resp.status_code == 200
|
|
json_resp = json.loads(resp.content)
|
|
assert '127.0.0.1' in [x['ip'] for x in json_resp]
|
|
|
|
@patch('seahub.api2.permissions.IsProVersion.has_permission')
|
|
def test_can_post(self, mock_IsProVersion):
|
|
mock_IsProVersion.return_value= True
|
|
resp = self.client.post(self.url, {'ipaddress': self.test_ip})
|
|
assert resp.status_code == 201
|
|
resp = self.client.get(self.url)
|
|
json_resp = json.loads(resp.content)
|
|
assert self.test_ip in [x['ip'] for x in json_resp]
|
|
|
|
@patch('seahub.api2.permissions.IsProVersion.has_permission')
|
|
def test_can_delete(self, mock_IsProVersion):
|
|
mock_IsProVersion.return_value= True
|
|
data = 'ipaddress=1.1.1.1'
|
|
resp = self.client.delete(self.url, data, 'application/x-www-form-urlencoded')
|
|
assert resp.status_code == 200
|
|
resp = self.client.get(self.url)
|
|
json_resp = json.loads(resp.content)
|
|
assert '1.1.1.1' not in [x['ip'] for x in json_resp]
|
|
|
|
def test_cmp_ip(self):
|
|
ip_list = [{'ip': '200.1.1.1'}, {'ip': '192.1.1.1'}, {'ip': '111.1.1.1'}]
|
|
new_ip_list = sorted(ip_list, cmp = cmp_ip)
|
|
assert new_ip_list == ip_list[::-1]
|
|
|
|
ip_list = [{'ip': '192.1.1.1'}, {'ip': '192.*.1.1'}]
|
|
new_ip_list = sorted(ip_list, cmp = cmp_ip)
|
|
assert new_ip_list == ip_list
|
|
|
|
ip_list = [{'ip': '192.*.1.1'}, {'ip': '192.1.1.1'}]
|
|
new_ip_list = sorted(ip_list, cmp = cmp_ip)
|
|
assert new_ip_list == ip_list[::-1]
|
|
|
|
ip_list = [{'ip': '111.1.1.1'}, {'ip': '111.8.1.1'}]
|
|
new_ip_list = sorted(ip_list, cmp = cmp_ip)
|
|
assert new_ip_list == ip_list
|
|
|
|
ip_list = [{'ip': '111.1.*.2'}, {'ip': '111.1.*.1'}]
|
|
new_ip_list = sorted(ip_list, cmp = cmp_ip)
|
|
assert new_ip_list == ip_list[::-1]
|
|
|
|
ip_list = [{'ip': '111.1.*.2'}, {'ip': '111.2.*.1'}, {'ip': '111.1.*.2'}]
|
|
new_ip_list = sorted(ip_list, cmp = cmp_ip)
|
|
assert new_ip_list == [ip_list[0], ip_list[2], ip_list[1]]
|
|
|
|
ip_list = [{'ip': '111.1.*.2'}, {'ip': '112.2.*.1'}, {'ip': '110.1.*.2'}]
|
|
new_ip_list = sorted(ip_list, cmp = cmp_ip)
|
|
assert new_ip_list == [ip_list[2], ip_list[0], ip_list[1]]
|
|
|
|
ip_list = [{'ip': '111.1.*.2'}, {'ip': '111.1.*.*'}, {'ip': '111.*.*.2'}]
|
|
new_ip_list = sorted(ip_list, cmp = cmp_ip)
|
|
assert new_ip_list == [ip_list[0], ip_list[1], ip_list[2]]
|