haiwen/seahub
1
0
Fork 0
mirror of https://github.com/haiwen/seahub.git synced 2025-09-16 23:29:49 +00:00
Code Issues Releases Wiki Activity

[tests] Fix shared file tests

Browse Source
This commit is contained in:
zhengxie
2015-10-20 11:44:19 +08:00
parent 8fd9bebf45
commit df1ff9e0f6
1 changed files with 32 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
tests/api/test_shares.py
View File

@@ -1,40 +1,22 @@
#coding: UTF-8 #coding: UTF-8
from django.core.cache import cache
from constance import config
from seahub.share.models import FileShare, UploadLinkShare from seahub.share.models import FileShare, UploadLinkShare
from seahub.test_utils import BaseTestCase from seahub.test_utils import BaseTestCase
from tests.common.utils import urljoin, randstring from tests.common.utils import randstring
from tests.api.apitestbase import ApiTestBase
from tests.api.urls import SHARED_LINKS_URL
#class SharesApiTest(ApiTestBase):
# def test_create_file_shared_link(self):
# with self.get_tmp_repo() as repo:
# fname, _ = self.create_file(repo)
# fsurl = urljoin(repo.file_url, 'shared-link')
# data = {
# 'type': 'f',
# 'p': '/' + fname,
# }
# res = self.put(fsurl, data=data, expected=201)
# self.assertRegexpMatches(res.headers['Location'], \
# r'http(.*)/f/(\w{10,10})/')
#
# res = self.get(SHARED_LINKS_URL).json()
# self.assertNotEmpty(res)
# for fileshare in res['fileshares']:
# self.assertIsNotNone(fileshare['username'])
# self.assertIsNotNone(fileshare['repo_id'])
# #self.assertIsNotNone(fileshare['ctime'])
# self.assertIsNotNone(fileshare['s_type'])
# self.assertIsNotNone(fileshare['token'])
# self.assertIsNotNone(fileshare['view_cnt'])
# self.assertIsNotNone(fileshare['path'])
#
#
class FileSharedLinkApiTest(BaseTestCase): class FileSharedLinkApiTest(BaseTestCase):
def setUp(self):
cache.clear()
self.curr_passwd_len = config.SHARE_LINK_PASSWORD_MIN_LENGTH
config.SHARE_LINK_PASSWORD_MIN_LENGTH = 3
def tearDown(self): def tearDown(self):
self.remove_repo() self.remove_repo()
config.SHARE_LINK_PASSWORD_MIN_LENGTH = self.curr_passwd_len
def test_create_file_shared_link_with_invalid_path(self): def test_create_file_shared_link_with_invalid_path(self):
self.login_as(self.user) self.login_as(self.user)
@@ -70,8 +52,8 @@ class FileSharedLinkApiTest(BaseTestCase):
'application/x-www-form-urlencoded', 'application/x-www-form-urlencoded',
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.assertRegexpMatches(resp._headers['location'][1], \ self.assertRegexpMatches(resp._headers['location'][1],
r'http(.*)/f/(\w{10,10})/') r'http(.*)/f/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
self.assertIsNotNone(FileShare.objects.get(token=token)) self.assertIsNotNone(FileShare.objects.get(token=token))
@@ -86,8 +68,8 @@ class FileSharedLinkApiTest(BaseTestCase):
'application/x-www-form-urlencoded', 'application/x-www-form-urlencoded',
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.assertRegexpMatches(resp._headers['location'][1], \ self.assertRegexpMatches(resp._headers['location'][1],
r'http(.*)/f/(\w{10,10})/') r'http(.*)/f/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
fileshare = FileShare.objects.get(token=token) fileshare = FileShare.objects.get(token=token)
@@ -103,8 +85,8 @@ class FileSharedLinkApiTest(BaseTestCase):
'application/x-www-form-urlencoded', 'application/x-www-form-urlencoded',
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.assertRegexpMatches(resp._headers['location'][1], \ self.assertRegexpMatches(resp._headers['location'][1],
r'http(.*)/f/(\w{10,10})/') r'http(.*)/f/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
fileshare = FileShare.objects.get(token=token) fileshare = FileShare.objects.get(token=token)
@@ -120,8 +102,8 @@ class FileSharedLinkApiTest(BaseTestCase):
'application/x-www-form-urlencoded', 'application/x-www-form-urlencoded',
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.assertRegexpMatches(resp._headers['location'][1], \ self.assertRegexpMatches(resp._headers['location'][1],
r'http(.*)/f/(\w{10,10})/') r'http(.*)/f/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
fileshare = FileShare.objects.get(token=token) fileshare = FileShare.objects.get(token=token)
@@ -139,8 +121,8 @@ class FileSharedLinkApiTest(BaseTestCase):
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.dir_link_location = resp._headers['location'][1] self.dir_link_location = resp._headers['location'][1]
self.assertRegexpMatches(self.dir_link_location, \ self.assertRegexpMatches(self.dir_link_location,
r'http(.*)/d/(\w{10,10})/') r'http(.*)/d/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
self.assertIsNotNone(FileShare.objects.get(token=token)) self.assertIsNotNone(FileShare.objects.get(token=token))
@@ -156,8 +138,8 @@ class FileSharedLinkApiTest(BaseTestCase):
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.dir_link_location = resp._headers['location'][1] self.dir_link_location = resp._headers['location'][1]
self.assertRegexpMatches(self.dir_link_location, \ self.assertRegexpMatches(self.dir_link_location,
r'http(.*)/d/(\w{10,10})/') r'http(.*)/d/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
fileshare = FileShare.objects.get(token=token) fileshare = FileShare.objects.get(token=token)
@@ -174,8 +156,8 @@ class FileSharedLinkApiTest(BaseTestCase):
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.dir_link_location = resp._headers['location'][1] self.dir_link_location = resp._headers['location'][1]
self.assertRegexpMatches(self.dir_link_location, \ self.assertRegexpMatches(self.dir_link_location,
r'http(.*)/d/(\w{10,10})/') r'http(.*)/d/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
fileshare = FileShare.objects.get(token=token) fileshare = FileShare.objects.get(token=token)
@@ -192,8 +174,8 @@ class FileSharedLinkApiTest(BaseTestCase):
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.dir_link_location = resp._headers['location'][1] self.dir_link_location = resp._headers['location'][1]
self.assertRegexpMatches(self.dir_link_location, \ self.assertRegexpMatches(self.dir_link_location,
r'http(.*)/d/(\w{10,10})/') r'http(.*)/d/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
fileshare = FileShare.objects.get(token=token) fileshare = FileShare.objects.get(token=token)
@@ -211,8 +193,8 @@ class FileSharedLinkApiTest(BaseTestCase):
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.dir_link_location = resp._headers['location'][1] self.dir_link_location = resp._headers['location'][1]
self.assertRegexpMatches(self.dir_link_location, \ self.assertRegexpMatches(self.dir_link_location,
r'http(.*)/u/d/(\w{10,10})/') r'http(.*)/u/d/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
self.assertIsNotNone(UploadLinkShare.objects.get(token=token)) self.assertIsNotNone(UploadLinkShare.objects.get(token=token))
@@ -228,8 +210,8 @@ class FileSharedLinkApiTest(BaseTestCase):
) )
self.assertEqual(201, resp.status_code) self.assertEqual(201, resp.status_code)
self.dir_link_location = resp._headers['location'][1] self.dir_link_location = resp._headers['location'][1]
self.assertRegexpMatches(self.dir_link_location, \ self.assertRegexpMatches(self.dir_link_location,
r'http(.*)/u/d/(\w{10,10})/') r'http(.*)/u/d/(\w{10,10})/')
token = resp._headers['location'][1].split('/')[-2] token = resp._headers['location'][1].split('/')[-2]
uls = UploadLinkShare.objects.get(token=token) uls = UploadLinkShare.objects.get(token=token)
@@ -240,7 +222,8 @@ class SharedFileDetailApiTest(BaseTestCase):
def _add_file_shared_link_with_password(self): def _add_file_shared_link_with_password(self):
password = randstring(6) password = randstring(6)
uls = FileShare.objects.create_file_link(self.user.username, uls = FileShare.objects.create_file_link(self.user.username,
self.repo.id, self.file, password) self.repo.id, self.file,
password)
return (uls.token, password) return (uls.token, password)