mirror of
https://github.com/haiwen/seahub.git
synced 2025-07-02 01:42:41 +00:00
351 lines
13 KiB
Python
351 lines
13 KiB
Python
#coding: UTF-8
|
|
|
|
from seahub.share.models import FileShare, UploadLinkShare
|
|
from seahub.test_utils import BaseTestCase
|
|
from tests.common.utils import urljoin, 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):
|
|
|
|
def tearDown(self):
|
|
self.remove_repo()
|
|
|
|
def test_create_file_shared_link_with_invalid_type(self):
|
|
self.login_as(self.user)
|
|
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=sf" % (self.file),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(400, resp.status_code)
|
|
|
|
def test_create_file_shared_link_with_invalid_path(self):
|
|
self.login_as(self.user)
|
|
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"ps=%s&type=f" % (self.file),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(400, resp.status_code)
|
|
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"type=f",
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(400, resp.status_code)
|
|
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=f" % randstring(6),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(400, resp.status_code)
|
|
|
|
def test_can_create_file_download_link(self):
|
|
self.login_as(self.user)
|
|
|
|
# create file download share link
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=f" % (self.file),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.assertRegexpMatches(resp._headers['location'][1], \
|
|
r'http(.*)/f/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
self.assertIsNotNone(FileShare.objects.get(token=token))
|
|
|
|
def test_can_create_file_download_link_with_exipre(self):
|
|
self.login_as(self.user)
|
|
|
|
# create file download share link with expire
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=f&expire=5" % (self.file),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.assertRegexpMatches(resp._headers['location'][1], \
|
|
r'http(.*)/f/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
fileshare = FileShare.objects.get(token=token)
|
|
self.assertIsNotNone(fileshare.expire_date)
|
|
|
|
def test_can_create_file_download_link_with_password(self):
|
|
self.login_as(self.user)
|
|
|
|
# create file download share link with password
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=f&password=123" % (self.file),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.assertRegexpMatches(resp._headers['location'][1], \
|
|
r'http(.*)/f/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
fileshare = FileShare.objects.get(token=token)
|
|
self.assertIsNotNone(fileshare.password)
|
|
|
|
def test_can_create_file_download_link_with_password_exipre(self):
|
|
self.login_as(self.user)
|
|
|
|
# create file download share link with password and expire
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=f&password=123&expire=5" % (self.file),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.assertRegexpMatches(resp._headers['location'][1], \
|
|
r'http(.*)/f/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
fileshare = FileShare.objects.get(token=token)
|
|
self.assertIsNotNone(fileshare.expire_date)
|
|
self.assertIsNotNone(fileshare.password)
|
|
|
|
def test_can_create_dir_download_link(self):
|
|
self.login_as(self.user)
|
|
|
|
# create dir download share link
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=d" % (self.folder),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.dir_link_location = resp._headers['location'][1]
|
|
self.assertRegexpMatches(self.dir_link_location, \
|
|
r'http(.*)/d/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
self.assertIsNotNone(FileShare.objects.get(token=token))
|
|
|
|
def test_can_create_dir_download_link_with_exipre(self):
|
|
self.login_as(self.user)
|
|
|
|
# create dir download share link with expire
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=d&expire=5" % (self.folder),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.dir_link_location = resp._headers['location'][1]
|
|
self.assertRegexpMatches(self.dir_link_location, \
|
|
r'http(.*)/d/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
fileshare = FileShare.objects.get(token=token)
|
|
self.assertIsNotNone(fileshare.expire_date)
|
|
|
|
def test_can_create_dir_download_link_with_password(self):
|
|
self.login_as(self.user)
|
|
|
|
# create dir download share link with password
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=d&password=123" % (self.folder),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.dir_link_location = resp._headers['location'][1]
|
|
self.assertRegexpMatches(self.dir_link_location, \
|
|
r'http(.*)/d/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
fileshare = FileShare.objects.get(token=token)
|
|
self.assertIsNotNone(fileshare.password)
|
|
|
|
def test_can_create_dir_download_link_with_password_exipre(self):
|
|
self.login_as(self.user)
|
|
|
|
# create dir download share link with password and expire
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=d&password=123&expire=5" % (self.folder),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.dir_link_location = resp._headers['location'][1]
|
|
self.assertRegexpMatches(self.dir_link_location, \
|
|
r'http(.*)/d/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
fileshare = FileShare.objects.get(token=token)
|
|
self.assertIsNotNone(fileshare.expire_date)
|
|
self.assertIsNotNone(fileshare.password)
|
|
|
|
def test_can_create_dir_upload_link(self):
|
|
self.login_as(self.user)
|
|
|
|
# create dir download share link
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=d&share_type=upload" % (self.folder),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.dir_link_location = resp._headers['location'][1]
|
|
self.assertRegexpMatches(self.dir_link_location, \
|
|
r'http(.*)/u/d/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
self.assertIsNotNone(UploadLinkShare.objects.get(token=token))
|
|
|
|
def test_can_create_dir_upload_link_with_password(self):
|
|
self.login_as(self.user)
|
|
|
|
# create dir download share link with password
|
|
resp = self.client.put(
|
|
'/api2/repos/%s/file/shared-link/' % (self.repo.id),
|
|
"p=%s&type=d&share_type=upload&password=123" % (self.folder),
|
|
'application/x-www-form-urlencoded',
|
|
)
|
|
self.assertEqual(201, resp.status_code)
|
|
self.dir_link_location = resp._headers['location'][1]
|
|
self.assertRegexpMatches(self.dir_link_location, \
|
|
r'http(.*)/u/d/(\w{10,10})/')
|
|
|
|
token = resp._headers['location'][1].split('/')[-2]
|
|
uls = UploadLinkShare.objects.get(token=token)
|
|
self.assertIsNotNone(uls.password)
|
|
|
|
class SharedFileDetailApiTest(BaseTestCase):
|
|
|
|
def _add_file_shared_link_with_password(self):
|
|
password = randstring(6)
|
|
uls = FileShare.objects.create_file_link(self.user.username,
|
|
self.repo.id, self.file, password)
|
|
|
|
return (uls.token, password)
|
|
|
|
def tearDown(self):
|
|
self.remove_repo()
|
|
|
|
def test_get_file_share_detail_with_password(self):
|
|
self.login_as(self.user)
|
|
|
|
token, password = self._add_file_shared_link_with_password()
|
|
|
|
url = '/api2/f/%s/detail/' % (token)
|
|
resp = self.client.get(url + '?password=' + password)
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertIsNotNone(resp)
|
|
|
|
def test_get_file_share_detail_with_invalid_password(self):
|
|
self.login_as(self.user)
|
|
|
|
token, password = self._add_file_shared_link_with_password()
|
|
|
|
url = '/api2/f/%s/detail/' % (token)
|
|
resp = self.client.get(url + '?password=' + randstring(5))
|
|
self.assertEqual(403, resp.status_code)
|
|
|
|
resp = self.client.get('/api2/f/%s/detail/' % (token))
|
|
self.assertEqual(403, resp.status_code)
|
|
|
|
class SharedDirApiTest(BaseTestCase):
|
|
|
|
def tearDown(self):
|
|
self.remove_repo()
|
|
|
|
def _add_dir_download_link(self):
|
|
ls = FileShare.objects.create_dir_link(self.user.username,
|
|
self.repo.id, self.folder)
|
|
|
|
self.create_folder(repo_id=self.repo.id,
|
|
parent_dir=self.folder,
|
|
dirname='sub-folder',
|
|
username='test@test.com')
|
|
|
|
self.create_file(repo_id=self.repo.id,
|
|
parent_dir=self.folder,
|
|
filename='sub-test.txt',
|
|
username='test@test.com')
|
|
|
|
return ls.token
|
|
|
|
def test_can_get_direntry_in_dir_download_link(self):
|
|
self.login_as(self.user)
|
|
|
|
token = self._add_dir_download_link()
|
|
|
|
# get direntry in dir download share link
|
|
resp = self.client.get(
|
|
'/api2/d/%s/dir/' % (token),
|
|
)
|
|
self.assertEqual(200, resp.status_code)
|
|
self.assertIsNotNone(resp)
|
|
|
|
class SharedLinksApiTest(BaseTestCase):
|
|
|
|
def tearDown(self):
|
|
self.remove_repo()
|
|
|
|
def _add_file_shared_link(self):
|
|
ls = FileShare.objects.create_dir_link(self.user.username,
|
|
self.repo.id, self.folder)
|
|
|
|
return ls.token
|
|
|
|
def _add_upload_shared_link(self):
|
|
uls = UploadLinkShare.objects.create_upload_link_share(
|
|
self.user.username, self.repo.id, self.folder)
|
|
|
|
return uls.token
|
|
|
|
def test_can_delete_file_shared_link(self):
|
|
self.login_as(self.user)
|
|
|
|
token = self._add_file_shared_link()
|
|
|
|
resp = self.client.get(
|
|
'/api2/shared-links/?t=%s' % (token),
|
|
)
|
|
self.assertEqual(200, resp.status_code)
|
|
|
|
def test_can_delete_upload_shared_link(self):
|
|
self.login_as(self.user)
|
|
|
|
token = self._add_upload_shared_link()
|
|
|
|
resp = self.client.get(
|
|
'/api2/shared-links/?t=%s' % (token),
|
|
)
|
|
self.assertEqual(200, resp.status_code)
|